feat: add lost api parent

cmd/api.go

@@ -35,14 +35,8 @@ var apiCmd *cobra.Command = &cobra.Command{
 			}
 			_ = json.Unmarshal(file, &parsedDeps)
 
-			var apiCallers []string
-			for _, api := range restApis {
-				apiCaller := api.PackageName + "." + api.ClassName + "." + api.MethodName
-				apiCallers = append(apiCallers, apiCaller)
-			}
-
 			analyser := domain.NewCallGraph()
-			dotContent := analyser.AnalysisByFiles(apiCallers, parsedDeps)
+			dotContent := analyser.AnalysisByFiles(restApis, parsedDeps)
 
 			if remove != "" {
 				dotContent = strings.ReplaceAll(dotContent, remove, "")

src/domain/call_graph.go

 package domain
 
 import (
+	"coca/src/adapter/api"
 	"coca/src/adapter/models"
-	"fmt"
 )
 
 type CallGraph struct {
@@ -43,14 +43,16 @@ func BuildCallChain(funcName string, methodMap map[string][]string) string {
 	return "\n"
 }
 
-func (c CallGraph) AnalysisByFiles(callers []string, deps []models.JClassNode) string {
+func (c CallGraph) AnalysisByFiles(restApis []api.RestApi, deps []models.JClassNode) string {
 	methodMap := c.BuildMethodMap(deps)
 
 	results := "digraph G { \n"
-	for _, caller := range callers {
-		fmt.Println(caller)
-		chain := BuildCallChain(caller, methodMap)
-		fmt.Println(chain)
+
+	for _, restApi := range restApis {
+		caller := restApi.PackageName + "." + restApi.ClassName + "." + restApi.MethodName
+
+		chain := "\"" + restApi.HttpMethod + " " + restApi.Uri + "\" -> \"" + caller + "\";\n"
+		chain = chain + BuildCallChain(caller, methodMap)
 		results = results + "\n" + chain
 	}
 	return results + "}\n"