zishuo/public
1
0
Fork 0
mirror of https://github.com/xxjwxc/public.git synced 2025-09-26 20:01:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update myast.go

Browse Source
This commit is contained in:
xxj
2022-09-15 11:01:04 +08:00
parent 34b58cc3c1
commit 0db60dd3db
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
myast/myast.go
View File

@@ -59,7 +59,7 @@ func (a *structAnalys) ParserStruct(astPkg *ast.Package, structName string) (inf
} }
info.Name = structName info.Name = structName
info.Items = a.structFieldInfo(astPkg, st) info.Items = a.structFieldInfo(astPkg, st, info.Name)
return return
} }
} }
@@ -71,7 +71,7 @@ func (a *structAnalys) ParserStruct(astPkg *ast.Package, structName string) (inf
return nil return nil
} }
func (a *structAnalys) structFieldInfo(astPkg *ast.Package, sinfo *ast.StructType) (items []mydoc.ElementInfo) { func (a *structAnalys) structFieldInfo(astPkg *ast.Package, sinfo *ast.StructType, structName string) (items []mydoc.ElementInfo) {
if sinfo == nil || sinfo.Fields == nil { if sinfo == nil || sinfo.Fields == nil {
return return
} }
@@ -116,13 +116,13 @@ func (a *structAnalys) structFieldInfo(astPkg *ast.Package, sinfo *ast.StructTyp
case *ast.SelectorExpr: // 非本文件包 case *ast.SelectorExpr: // 非本文件包
a.dealSelectorExpr(x, &info, importMP) a.dealSelectorExpr(x, &info, importMP)
case *ast.Ident: case *ast.Ident:
a.dealIdent(astPkg, x, &info) a.dealIdent(astPkg, x, &info, structName)
case *ast.StarExpr: case *ast.StarExpr:
switch x1 := x.X.(type) { switch x1 := x.X.(type) {
case *ast.SelectorExpr: // 非本文件包 case *ast.SelectorExpr: // 非本文件包
a.dealSelectorExpr(x1, &info, importMP) a.dealSelectorExpr(x1, &info, importMP)
case *ast.Ident: case *ast.Ident:
a.dealIdent(astPkg, x1, &info) a.dealIdent(astPkg, x1, &info, structName)
} }
case *ast.InterfaceType: case *ast.InterfaceType:
info.Type = "Interface" info.Type = "Interface"
@@ -134,10 +134,10 @@ func (a *structAnalys) structFieldInfo(astPkg *ast.Package, sinfo *ast.StructTyp
case *ast.SelectorExpr: // 非本文件包 case *ast.SelectorExpr: // 非本文件包
a.dealSelectorExpr(x, &info, importMP) a.dealSelectorExpr(x, &info, importMP)
case *ast.Ident: case *ast.Ident:
a.dealIdent(astPkg, x, &info) a.dealIdent(astPkg, x, &info, structName)
} }
case *ast.Ident: // 本文件 case *ast.Ident: // 本文件
a.dealIdent(astPkg, exp, &info) a.dealIdent(astPkg, exp, &info, structName)
case *ast.MapType: // map case *ast.MapType: // map
key := "" key := ""
value := "" value := ""
@@ -196,9 +196,11 @@ func (a *structAnalys) dealSelectorExpr(exp *ast.SelectorExpr, info *mydoc.Eleme
} }
} }
func (a *structAnalys) dealIdent(astPkg *ast.Package, exp *ast.Ident, info *mydoc.ElementInfo) { // 本文件 func (a *structAnalys) dealIdent(astPkg *ast.Package, exp *ast.Ident, info *mydoc.ElementInfo, structName string) { // 本文件
info.Type = exp.Name info.Type = exp.Name
if !tools.IsInternalType(info.Type) { // 非基础类型 if !tools.IsInternalType(info.Type) { // 非基础类型
if structName != info.Type {
info.TypeRef = a.ParserStruct(astPkg, info.Type) info.TypeRef = a.ParserStruct(astPkg, info.Type)
} }
}
} }