Function TestGenerateXlog

core/cmd/server/docs/swagger_test.go:16–77

(t *testing.T)

)

func TestGenerateXlog(t *testing.T) {
	workDir := panelWorkDir(t)
	fset := token.NewFileSet()

	apiDirs := []string{
		filepath.Join(workDir, "agent/app/api/v2"),
		filepath.Join(workDir, "core/app/api/v2"),
		filepath.Join(workDir, "agent/xpack/app/api/v2"),
		filepath.Join(workDir, "core/xpack/app/api/v2"),
		filepath.Join(workDir, "agent/enterprise/app/api/v2"),
		filepath.Join(workDir, "core/enterprise/app/api/v2"),
	}

	xlogMap := make(map[string]operationJson)
	for _, dir := range apiDirs {
		entries, _ := os.ReadDir(dir)
		for _, info := range entries {
			if info.IsDir() {
				continue
			}
			fileItem, err := parser.ParseFile(fset, filepath.Join(dir, info.Name()), nil, parser.ParseComments)
			if err != nil {
				continue
			}
			for _, decl := range fileItem.Decls {
				switch d := decl.(type) {
				case *ast.FuncDecl:
					if d.Doc != nil {
						routerContent := ""
						logContent := ""
						for _, comment := range d.Doc.List {
							if strings.HasPrefix(comment.Text, "// @Router") {
								routerContent = replaceStr(comment.Text, "// @Router", "[post]", "[get]")
							}
							if strings.HasPrefix(comment.Text, "// @x-panel-log") {
								logContent = replaceStr(comment.Text, "// @x-panel-log")
							}
						}
						if len(routerContent) != 0 && len(logContent) != 0 {
							var item operationJson
							if err := json.Unmarshal([]byte(logContent), &item); err != nil {
								panic(fmt.Sprintf("json unmarshal failed, err: %v", err))
							}
							xlogMap[routerContent] = item
						}
					}
				}
			}
		}
	}

	newJson, err := json.MarshalIndent(xlogMap, "", "\t")
	if err != nil {
		panic(fmt.Sprintf("json marshal for new file failed, err: %v", err))
	}
	if err := os.WriteFile("x-log.json", newJson, 0640); err != nil {
		panic(fmt.Sprintf("write core x-log.json failed, err: %v", err))
	}