zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-07 15:10:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(swagger): allow users to specify the generated swagger file name (#4809)

Browse Source
This commit is contained in:
shaouai
2025-04-27 23:34:52 +08:00
committed by GitHub
parent eafd11d949
commit 98bebbc74f
3 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
tools/goctl/api/swagger/command.go
View File

@@ -8,9 +8,10 @@ import (
"strings"
"github.com/spf13/cobra"
"gopkg.in/yaml.v2"
"github.com/zeromicro/go-zero/tools/goctl/pkg/parser/api/parser"
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
"gopkg.in/yaml.v2"
)
var (
@@ -20,6 +21,9 @@ var (
// VarStringDir specifies the directory to generate swagger file.
VarStringDir string
// VarStringFilename specifies the generated swagger file name without the extension.
VarStringFilename string
// VarBoolYaml specifies whether to generate a YAML file.
VarBoolYaml bool
)
@@ -57,9 +61,14 @@ func Command(_ *cobra.Command, _ []string) error {
return err
}
base := filepath.Base(VarStringAPI)
filename := VarStringFilename
if filename == "" {
base := filepath.Base(VarStringAPI)
filename = strings.TrimSuffix(base, filepath.Ext(base))
}
if VarBoolYaml {
filename := filepath.Join(VarStringDir, strings.TrimSuffix(base, filepath.Ext(base))+".yaml")
filePath := filepath.Join(VarStringDir, filename+".yaml")
var jsonObj interface{}
if err := yaml.Unmarshal(data, &jsonObj); err != nil {
@@ -70,10 +79,10 @@ func Command(_ *cobra.Command, _ []string) error {
if err != nil {
return err
}
return os.WriteFile(filename, data, 0644)
return os.WriteFile(filePath, data, 0644)
}
// generate json swagger file
filename := filepath.Join(VarStringDir, strings.TrimSuffix(base, filepath.Ext(base))+".json")
return os.WriteFile(filename, data, 0644)
// generate json swagger file
filePath := filepath.Join(VarStringDir, filename+".json")
return os.WriteFile(filePath, data, 0644)
}