实现自定义函数

Author: rockis

README.md

@@ -1,11 +1,67 @@
-# TExpr 
-Go语言布尔及数值运算表达式解析器
+# /Users/zhanglei/Work/Projects/texprTExpr 
+
 
 [![Build Status](https://travis-ci.org/aliyun/texpr.svg?branch=master)](https://travis-ci.org/aliyun/texpr)
 [![Coverage status](https://img.shields.io/codecov/c/github/aliyun/texpr/master.svg)](https://codecov.io/github/aliyun/texpr)
 
+Go语言布尔及数值运算表达式解析器
+
 [Java语言版本](https://github.com/tauris-io/expression)
 
+布尔表达式：
+
+```
+$country in ['cn', 'us', 'jp'] && $length > 100
+$host == 'cs.qa.com' && 'Webkit' not in $ua
+```
+
+数值运算表达式:
+
+```
+( 1 ^ 5) + (2 << 3)
+4 | 3 + 3
+```
+
+
+
+# Example
+
+```
+package main
+
+import (
+"fmt"
+"log"
+
+"github.com/aliyun/texpr"
+)
+
+type context map[string]interface{}
+
+func (vg context) Get(name string) interface{} {
+v, ok := vg[name]
+if ok {
+return v
+}
+return nil
+}
+
+func main() {
+m := context {
+"$value" : 15.0,
+}
+expr := texpr.MustCompile("$value - 10")
+v, err := expr.Eval(m)
+if err != nil {
+log.Fatalf("expression err, %s", err)
+}
+fmt.Println(v) // 5
+}
+
+```
+
+
+
 ## Quick Start
 
 ```bash
@@ -19,7 +75,7 @@ go run test/simple/main.go "33 > 22 && 22 > 11"
 go run test/simple/main.go "'world' =~ /.+orl.+/"
 ```
 
-
+# 表达式语法
 
 ## 数值计算
 
@@ -40,7 +96,7 @@ go run test/simple/main.go "'world' =~ /.+orl.+/"
 
 | 操作符 | 描述 | 例子 |
 | ------ | ------------------------------------------------------------ | ------------ |
-| =  | 两值相等 | A=B  |
+| == | 两值相等 | A==B |
 | > | 左值大于右值 | A>B |
 | < | 右值大于左值 | A<B |
 | >= | 左值大于等于右值 | A>=B |

TExpr.g4

@@ -85,18 +85,15 @@ scientific
  ;
 
 function
- : funcname LPAREN expression RPAREN
+ : funcname LPAREN parameters RPAREN
  ;
 
 funcname
- : COS
- | TAN
- | SIN
- | ACOS
- | ATAN
- | ASIN
- | LOG
- | LN
+ : IDENTIFIER
+ ;
+
+parameters
+ : expression (',' expression)*
  ;
 
 number

antlr.sh

@@ -2,4 +2,4 @@
 
 export PATH=/usr/local/bin:$PATH
 
-antlr -o ./ -no-listener -visitor -Dlanguage=Go -package ast TExpr.g4
+antlr -o ./ast -no-listener -visitor -Dlanguage=Go -package ast TExpr.g4

ast/texpr_base_visitor.go

@@ -1,6 +1,7 @@
 package texpr
 
 import (
+"errors"
 "fmt"
 "math"
 "reflect"
@@ -28,7 +29,7 @@ func Compile(expr string) (Expression, error) {
 func MustCompile(expr string) Expression {
 exp, err := Compile(expr)
 if err != nil {
-return nil
+panic(`texpr: Compile("` + expr + `"): ` + err.Error())
 }
 return exp
 }
@@ -668,13 +669,33 @@ func (v *ExprVisitor) VisitScientific(ctx *ast.ScientificContext) interface{} {
 }
 
 func (v *ExprVisitor) VisitFunction(ctx *ast.FunctionContext) interface{} {
-return v.VisitChildren(ctx)
+params := ctx.Parameters().Accept(v).([]interface{})
+fname := ctx.Funcname().Accept(v).(string)
+f, b := funcMap[fname]
+if !b {
+return errors.New("unknown function " + fname)
+}
+exprs := make([]Expression, len(params))
+for i, _ := range params {
+exprs[i] = params[i].(Expression)
+}
+return NewFuncExpr(fname, f, exprs)
 }
 
 func (v *ExprVisitor) VisitFuncname(ctx *ast.FuncnameContext) interface{} {
-return v.VisitChildren(ctx)
+return ctx.IDENTIFIER().GetText()
 }
 
+func (v *ExprVisitor) VisitParameters(ctx *ast.ParametersContext) interface{} {
+var params []interface{}
+allExprs := ctx.AllExpression()
+for i, _ := range allExprs {
+params = append(params, allExprs[i].Accept(v))
+}
+return params
+}
+
+
 func (v *ExprVisitor) VisitNumber(ctx *ast.NumberContext) interface{} {
 return v.VisitChildren(ctx)
 }