golang 整合antlr語法校驗解析

更新時間：2023年02月28日 10:53:55 作者：lj907722644

Antlr是一個語法分析器，本身是用java實現(xiàn)的，然是Runtime的庫也支持Golang、Java、Python等，本文給大家講解使用golang整合antlr進行語法解析，感興趣的朋友一起看看吧

1. 背景

在項目中我們可能會遇到表達式檢索的場景，例如，輸入以下表達式檢索，需要解析表達式并得到檢索結(jié)果。

ip="192.168.1.3" && (port="80" || protocol="http")

此時，我們需要對語法進行校驗、解析，應(yīng)當如何做呢？

下面給大家推薦一種使用語法校驗工具——Antlr

Antlr是一個語法分析器，本身是用java實現(xiàn)的，然是Runtime的庫也支持Golang、Java、Python等。

接下來給大家演示一下使用golang整合antlr進行語法解析。

2. goland安裝antlr插件

打開goland，F(xiàn)ile --> Settings --> Plugins, 搜索antlr，安裝 antlr4

在這里插入圖片描述

插件安裝完成后，可以看到ANTLR Preview窗口，一會我們可以在這個窗口進行簡單的語法校驗。

在這里插入圖片描述

3. 編寫語法校驗規(guī)則

1.創(chuàng)建工程，引入包

go get -u github.com/antlr/antlr4/runtime/Go/antlr/v4

2.在工程中新建一個antlr目錄，創(chuàng)建一個后綴名為 .g4 的文件，作為規(guī)則文件。此處我們創(chuàng)建Rule.g4

// 定義語法名稱，需要和文件名匹配
grammar Rule;

// DECIMAL, IDENTIFIER, COMMENTS, WS are set using regular expressions
// key 為表達式中可支持的檢索字段，可以是固定值(每個值中間用 | 隔開，是”或“的意思)，也可以是正則表達式
// value 使用正則表達式
KEY : 'ip' | 'port' | 'protocol';
//VALUE :'"' ( '""' | ~["\r\n] )* '"' ;
//KEY : ('A' .. 'Z' | 'a' .. 'z' |  '_') + ;
VALUE :'"' ( '\\"' | ~["] )* '"' ;

// COMMENT and WS are stripped from the output token stream by sending
// to a different channel 'skip'

COMMENT : '//' .+? ('\n'|EOF) -> skip ;

WS : [ \r\t\u000C\n]+ -> skip ;


/* Parser rules */
// 語法校驗的入口
start : logicalExpr* EOF;

// 語法支持的結(jié)構(gòu)
logicalExpr
    : comparisonExpr // 示例： key == value 表示支持 == 和 != 的表達式
    | logicalExpr operator logicalExpr // 示例： key1 == value1 && key2 != value2 表示支持 && 和 || 運算符連接表達式
    | lparen logicalExpr rparen // 示例： (key1 == value1 && key2 != value2) 表示支持 () 連接表達式
    ;

comparisonExpr
    : KEY compare VALUE
    ;
compare
    : '='
    | '!='
    ;
operator
    : '&&'
    | '||'
    ;
lparen
    :  '('
    ;
rparen
    :  ')'
    ;

3.初始化校驗語法

1.選中Rule.g4 文件，鼠標右鍵，選擇 Configure ANTLR Tool…

2.配置輸出路徑，和Rule.g4 同目錄；配置語言，使用Go

在這里插入圖片描述

3.選中Rule.g4 文件，鼠標右鍵，選擇 Generate ANTLR Recognizer，完成規(guī)則初始化

在這里插入圖片描述

4.樹狀圖校驗

在這里插入圖片描述

4. 語法校驗

1.自定義listener

package parser

import (
	"github.com/antlr/antlr4/runtime/Go/antlr/v4"
	"strings"
)

type MyRuleListener struct {
	*BaseRuleListener
	Queue    []interface{}
	QueueStr []string
}

// 注意：方法名必須是這個名字
func (s *MyRuleListener) EnterComparisonExpr(ctx *ComparisonExprContext) {
	key := ctx.GetChild(0).(antlr.ParseTree).GetText()
	operator := ctx.GetChild(1).(antlr.ParseTree).GetText()
	value := ctx.GetChild(2).(antlr.ParseTree).GetText()
	if strings.HasPrefix(value, "\"") {
		value = value[1:]
	}
	if strings.HasSuffix(value, "\"") {
		value = value[:len(value)-1]
	}
	keyValue := map[string]string{}
	keyValue["key"] = key
	keyValue["operator"] = operator
	keyValue["value"] = value

	s.PushStr(ctx.GetText())
	s.Push(keyValue)
}

// EnterKeyValue is called when production KeyValue is entered.
func (s *MyRuleListener) ExitOperator(ctx *OperatorContext) {
	s.Push(ctx.GetText())
	s.PushStr(ctx.GetText())
}

// EnterKeyValue is called when production KeyValue is entered.
func (s *MyRuleListener) ExitLparen(ctx *LparenContext) {
	s.Push(ctx.GetText())
	s.PushStr(ctx.GetText())
}

// EnterKeyValue is called when production KeyValue is entered.
func (s *MyRuleListener) ExitRparen(ctx *RparenContext) {
	s.Push(ctx.GetText())
	s.PushStr(ctx.GetText())
}

func (s *MyRuleListener) Push(i interface{}) {
	s.Queue = append(s.Queue, i)
}

func (s *MyRuleListener) PushStr(i string) {
	s.QueueStr = append(s.QueueStr, i)
}

2.獲取解析異常的錯誤信息

package parser

import "github.com/antlr/antlr4/runtime/Go/antlr/v4"

type RuleErrorListener struct {
	antlr.ErrorListener
	Msg string
}

func (l *RuleErrorListener) SyntaxError(recognizer antlr.Recognizer, offendingSymbol interface{}, line, column int, msg string, e antlr.RecognitionException) {
	l.Msg = msg
}

3.校驗

package main

import (
	parser "antlr-demo/antlr"
	"errors"
	"fmt"
	"github.com/antlr/antlr4/runtime/Go/antlr/v4"
)

func main() {
	expre := "ip=\"192.168.1.3\" && (port=\"80\" || protocol=\"http\")"
	err := checkExpre(expre)
	if err != nil {
		fmt.Println(err)
	}
}

func checkExpre(expre string) error {
	input := antlr.NewInputStream(expre)
	var lexerErr parser.RuleErrorListener
	lexer := parser.NewRuleLexer(input)
	lexer.AddErrorListener(&lexerErr)
	stream := antlr.NewCommonTokenStream(lexer, 0)
	ruleParser := parser.NewRuleParser(stream)
	ruleParser.BuildParseTrees = true
	var ruleErr parser.RuleErrorListener
	ruleParser.AddErrorListener(&ruleErr)
	tree := ruleParser.Start()
	listener := new(parser.MyRuleListener)
	antlr.ParseTreeWalkerDefault.Walk(listener, tree)

	if lexerErr.Msg != "" || ruleErr.Msg != "" {
		return errors.New("輸入的語法不正確")
	}
	expreList := listener.QueueStr
	fmt.Println("expreList--->", expreList)
	expreMap := listener.Queue
	fmt.Println("expreMap--->", expreMap)
	return nil
}

4.結(jié)果驗證

1.正確表達式

在這里插入圖片描述