未验证 提交 28995015 编写于 作者: P Phodal Huang

fix: fix interface data issue

上级 a442df51
...@@ -38,7 +38,7 @@ var currentExtends = "" ...@@ -38,7 +38,7 @@ var currentExtends = ""
var currentImplements = "" var currentImplements = ""
var depInjectMap map[string]string var depInjectMap map[string]string
func NewJavaApiListener(ident map[string]models2.JIdentifier, diMap map[string]string) *JavaApiListener { func NewJavaApiListener(jIdentMap map[string]models2.JIdentifier, diMap map[string]string) *JavaApiListener {
isSpringRestController = false isSpringRestController = false
currentClz = "" currentClz = ""
currentPkg = "" currentPkg = ""
...@@ -47,7 +47,7 @@ func NewJavaApiListener(ident map[string]models2.JIdentifier, diMap map[string]s ...@@ -47,7 +47,7 @@ func NewJavaApiListener(ident map[string]models2.JIdentifier, diMap map[string]s
imports = nil imports = nil
identMap = ident identMap = jIdentMap
depInjectMap = diMap depInjectMap = diMap
params := make(map[string]string) params := make(map[string]string)
......
...@@ -21,7 +21,7 @@ func (j *JavaIdentifierApp) AnalysisPath(codeDir string) []models.JIdentifier { ...@@ -21,7 +21,7 @@ func (j *JavaIdentifierApp) AnalysisPath(codeDir string) []models.JIdentifier {
context := parser.CompilationUnit() context := parser.CompilationUnit()
clzInfo := models.NewJIdentifier() clzInfo := models.NewJIdentifier()
listener := new(JavaIdentifierListener) listener := NewJavaIdentifierListener()
listener.InitNode(clzInfo) listener.InitNode(clzInfo)
antlr.NewParseTreeWalker().Walk(listener, context) antlr.NewParseTreeWalker().Walk(listener, context)
......
...@@ -14,6 +14,11 @@ var hasEnterClass = false ...@@ -14,6 +14,11 @@ var hasEnterClass = false
var hasEnterMethod = false var hasEnterMethod = false
var imports []string var imports []string
func NewJavaIdentifierListener() *JavaIdentifierListener {
hasEnterClass = false
return &JavaIdentifierListener{}
}
type JavaIdentifierListener struct { type JavaIdentifierListener struct {
parser.BaseJavaParserListener parser.BaseJavaParserListener
} }
...@@ -52,7 +57,7 @@ func (s *JavaIdentifierListener) EnterClassDeclaration(ctx *parser.ClassDeclarat ...@@ -52,7 +57,7 @@ func (s *JavaIdentifierListener) EnterClassDeclaration(ctx *parser.ClassDeclarat
} }
} }
func (s *JavaIdentifierListener) ExitClassDeclaration(ctx *parser.ClassDeclarationContext) { func (s *JavaIdentifierListener) ExitClassBody(ctx *parser.ClassBodyContext) {
hasEnterClass = false hasEnterClass = false
} }
...@@ -100,6 +105,7 @@ func (s *JavaIdentifierListener) ExitInterfaceMethodDeclaration(ctx *parser.Inte ...@@ -100,6 +105,7 @@ func (s *JavaIdentifierListener) ExitInterfaceMethodDeclaration(ctx *parser.Inte
var isOverrideMethod = false var isOverrideMethod = false
func (s *JavaIdentifierListener) EnterMethodDeclaration(ctx *parser.MethodDeclarationContext) { func (s *JavaIdentifierListener) EnterMethodDeclaration(ctx *parser.MethodDeclarationContext) {
hasEnterClass = true
hasEnterMethod = true hasEnterMethod = true
startLine := ctx.GetStart().GetLine() startLine := ctx.GetStart().GetLine()
...@@ -169,6 +175,7 @@ func buildAnnotation(ctx *parser.AnnotationContext) models.Annotation { ...@@ -169,6 +175,7 @@ func buildAnnotation(ctx *parser.AnnotationContext) models.Annotation {
} }
func (s *JavaIdentifierListener) EnterInterfaceDeclaration(ctx *parser.InterfaceDeclarationContext) { func (s *JavaIdentifierListener) EnterInterfaceDeclaration(ctx *parser.InterfaceDeclarationContext) {
hasEnterClass = true
node.Type = "Interface" node.Type = "Interface"
node.ClassName = ctx.IDENTIFIER().GetText() node.ClassName = ctx.IDENTIFIER().GetText()
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册