GOLang AST库使用

基础结构说明

  • 普通Node,不是特定语法结构,属于某个语法结构的一部分.
    • Comment 表示一行注释 // 或者 / /
    • CommentGroup 表示多行注释
    • Field 表示结构体中的一个定义或者变量,或者函数签名当中的参数或者返回值
    • FieldList 表示以”{}”或者”()”包围的Filed列表
  • Expression & Types (都划分成Expr接口)
    • BadExpr 用来表示错误表达式的占位符
    • Ident 比如报名,函数名,变量名
    • Ellipsis 省略号表达式,比如参数列表的最后一个可以写成arg…
    • BasicLit 基本字面值,数字或者字符串
    • FuncLit 函数定义
    • CompositeLit 构造类型,比如{1,2,3,4}
    • ParenExpr 括号表达式,被括号包裹的表达式
    • SelectorExpr 选择结构,类似于a.b的结构
    • IndexExpr 下标结构,类似这样的结构 expr[expr]
    • SliceExpr 切片表达式,类似这样 expr[low:mid:high]
    • TypeAssertExpr 类型断言类似于 X.(type)
    • CallExpr 调用类型,类似于 expr()
    • StarExpr 指针表达式,类似于 *X
    • UnaryExpr 一元表达式
    • BinaryExpr 二元表达式
    • KeyValueExp 键值表达式 key:value
    • ArrayType 数组类型
    • StructType 结构体类型
    • FuncType 函数类型
    • InterfaceType 接口类型
    • MapType map类型
    • ChanType 管道类型
  • Statements语句
    • BadStmt 错误的语句
    • DeclStmt 在语句列表里的申明
    • EmptyStmt 空语句
    • LabeledStmt 标签语句类似于 indent:stmt
    • ExprStmt 包含单独的表达式语句
    • SendStmt chan发送语句
    • IncDecStmt 自增或者自减语句
    • AssignStmt 赋值语句
    • GoStmt Go语句
    • DeferStmt 延迟语句
    • ReturnStmt return 语句
    • BranchStmt 分支语句 例如break continue
    • BlockStmt 块语句 {} 包裹
    • IfStmt If 语句
    • CaseClause case 语句
    • SwitchStmt switch 语句
    • TypeSwitchStmt 类型switch 语句 switch x:=y.(type)
    • CommClause 发送或者接受的case语句,类似于 case x <-:
    • SelectStmt select 语句
    • ForStmt for 语句
    • RangeStmt range 语句
  • Declarations声明
    • Spec type
    • Import Spec
    • Value Spec
    • Type Spec
    • BadDecl 错误申明
    • GenDecl 一般申明(和Spec相关,比如 import “a”,var a,type a)
  • FuncDecl 函数申明
    • Files and Packages
    • File 代表一个源文件节点,包含了顶级元素.
    • Package 代表一个包,包含了很多文件.

以上内容转载自某片大神文章..具体地址忘了,知道的告知一下加上连接

定义函数

&ast.FuncDecl{
        Name: id("NewEndpoints"), //函数名称
        Body: //函数内语句包含在这里
        Type: &ast.FuncType{ //这里定义函数参数和返回结果
            Params: //函数参数
            Results: &ast.FieldList{
                List: []*ast.Field{
                    &ast.Field{
                        Type: &ast.StarExpr{ //返回指针类型
                            X: ast.NewIdent("Endpoints"),
                            //.e.g *Endpoints
                        },
                    },
                    &ast.Field{
                        Type: &ast.SelectorExpr{
                            X:   ast.NewIdent("a"),
                            Sel: ast.NewIdent("b"),
                        },
                        //返回包对象.e.g : a.b
                    },
                },
            },
        },
    }

例子:

   // i.method
   //=== Add
   //=== Update
   //=== UpdateName
   //=== UpdateParent
   //=== UpdateEnableState
   //=== QueryByCode
   //=== QueryByLikeName
   //=== QueryList
   //=== AllTree
   //=== AllChildrenCode
   elts := make([]ast.Expr, len(i.methods))
   for index, v := range i.methods {
       elts[index] = &ast.KeyValueExpr{
           Key: ast.NewIdent(v.name.Name),
           Value: &ast.CallExpr{
               Fun: ast.NewIdent("Make" + i.prefix + v.name.Name + "Endpoint"),
               Args: []ast.Expr{
                   &ast.BasicLit{
                       Kind:  token.TYPE,
                       Value: "s",
                   },
               },
           },
       }
   }
   return &ast.FuncDecl{
       Name: ast.NewIdent("New" + i.prefix + "Endpoints"),
       Body: &ast.BlockStmt{
           List: []ast.Stmt{
               &ast.AssignStmt{
                   Lhs: []ast.Expr{
                       ast.NewIdent("s"),
                   },
                   Tok: token.DEFINE,
                   Rhs: []ast.Expr{
                       &ast.CompositeLit{
                           Type: ast.NewIdent("service." + i.prefix + "Service"),
                       },
                   },
               },
               &ast.ReturnStmt{
                   Results: []ast.Expr{
                       &ast.UnaryExpr{
                           Op: token.AND,
                           X: &ast.CompositeLit{
                               Type: ast.NewIdent(i.prefix + "Endpoints"),
                               Elts: elts,
                           },
                       },
                   },
               },
           },
       },
       Type: &ast.FuncType{
           Results: &ast.FieldList{
               List: []*ast.Field{
                   {
                       Type: &ast.StarExpr{
                           X: ast.NewIdent(i.prefix + "Endpoints"),
                       },
                   },
               },
           },
       },
   }
结果:
   func NewDeptEndpoints() *DeptEndpoints {
   s := service.DeptService{}
   return &DeptEndpoints{
       Add:               MakeDeptAddEndpoint(s),
       Update:            MakeDeptUpdateEndpoint(s),
       UpdateName:        MakeDeptUpdateNameEndpoint(s),
       UpdateParent:      MakeDeptUpdateParentEndpoint(s),
       UpdateEnableState: MakeDeptUpdateEnableStateEndpoint(s),
       QueryByCode:       MakeDeptQueryByCodeEndpoint(s),
       QueryByLikeName:   MakeDeptQueryByLikeNameEndpoint(s),
       QueryList:         MakeDeptQueryListEndpoint(s),
       AllTree:           MakeDeptAllTreeEndpoint(s),
       AllChildrenCode:   MakeDeptAllChildrenCodeEndpoint(s)}
}

语句

dbCall := &ast.CallExpr{ //表达式语句
        Fun: &ast.Ident{
            Name: "db.TagConnect", //表达式内容
        },
        Args: []ast.Expr{ //表达式参数集合
            &ast.BasicLit{
                Kind:  token.TYPE,
                Value: "dbtag",
            },
        },
        //结果就是: db.TagConnect(dbtag)
    }
    db := ast.AssignStmt{ //赋值语句
        Lhs: []ast.Expr{
            &ast.Ident{ //变量名称
                Name: "q",
            },
        },
        Tok: token.DEFINE, //注意这个值一定要有这里表示":="
        Rhs: []ast.Expr{dbCall}, //后面的表达式
    }
    //整体执行结果: q := db.TagConnect(dbtag)

结构体注释

&ast.GenDecl{
    Doc://注释内容
    Tok:   token.TYPE,
    Specs: []ast.Spec{&ast.TypeSpec{
        Name: name,
        Type: &ast.StructType{
            Fields: fields,
        },
    }},
}

- 结果:
//注释内容
type name struct {
    fields...
}
    原文作者:heldiam
    原文地址: https://www.jianshu.com/p/e38786de29c7
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞