背景
在做一些自动生成的代码工作时,有时需要知道方法以及对应的参数名
如果仅是方法,利用反射机制就可以解决
而参数名,程序编译后,已经丢失
可以通过 AST 事先获取方法的参数名
有了方法、参数名,加上反射,那么就可以方便生成胶水代码,自动集成进 HTTP 、 gRPC 等
获取方法、参数名的例子
下面的例子,从特定包 flagInput ,特定的结构体 flagStructName ,获取该结构体所有的方法,以及对应的参数名:
conf := &packages.Config{Mode: packages.LoadAllSyntax,Tests: false,}packages, err := packages.Load(conf, flagInput)if err != nil {fmt.Println("Error loading packages:", err)return}allPackages = packagesfor _, pkg := range packages {if pkg.Name != flagPackageName {continue}for _, info := range pkg.TypesInfo.Defs {if info == nil {continue}structType, ok := info.Type().(*types.Named)if !ok {continue}if structType.Obj().Name() != flagStructName {continue}for i := 0; i < structType.NumMethods(); i++ {method := structType.Method(i)methodName := method.Name()if _, ok := excludeMethods[method.Name()]; ok {continue}sig := method.Type().(*types.Signature)params := make([]string, sig.Params().Len())for i := 0; i < sig.Params().Len(); i++ {params[i] = sig.Params().At(i).Name()}allMethods[methodName] = paramssortMethods = append(sortMethods, methodName)}}}
可以把这些信息写入 map 声明,这样代码里就有类似反射参数的功能了。类似:
var methods = map[string][]string{"Method1": { "userid", "playerid", "otherparam1", "otherparam2" },"Method2": { "userid", "playerid" },
}
集成 HTTP 服务的例子
如有以下类似模板 register.tmpl:
func RegisterMethod(engine *gin.Engine) {{{ range .Methods -}}engine.Any(HttpPrefix+"{{.}}", func(context *gin.Context) { HandleGin(context, "{{.}}") }){{ end -}}
}
通过类似以下代码:
tmpl, err := template.ParseFiles(flagTmpl)if err != nil {panic(err)}var buf bytes.Buffererr = tmpl.Execute(&buf, map[string]interface{}{"Methods": sortMethods})if err != nil {panic(err)}
就可以自动生成到 HTTP 的胶水代码:
func RegisterMethod(engine *gin.Engine) {engine.Any(HttpPrefix+"Method1", func(context *gin.Context) { HandleGin(context, "Method1") })engine.Any(HttpPrefix+"Method2", func(context *gin.Context) { HandleGin(context, "Method2") })
}
例子中的 HandleGin 函数实现,则可以:
- 利用方法及对应的参数信息,获取实际参数值
- 利用反射,实现对应结构体函数的调用
其他例子
如生成函数的接口调用说明:
模板类似:
{{ range .Methods -}}
{{.Index}}.{{.Comment}} http://api/{{.MethodName}}?{{ range $index, $param := .Params }}{{ if $index }}&{{ end }}{{ $param }}=0{{ end }}
{{ end -}}
还是基于上面的获取的方法、参数名的 map 实例信息,就可以展开了