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...
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,616评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,020评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,078评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,040评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,154评论 6 385
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,265评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,298评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,072评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,491评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,795评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,970评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,654评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,272评论 3 318
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,985评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,223评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,815评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,852评论 2 351

推荐阅读更多精彩内容