zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimize code (#417)

Browse Source 
* optimize code

* optimize code

* optimize code

* optimize code
This commit is contained in:
kingxt
2021-01-26 17:37:22 +08:00
committed by GitHub
parent a784982030
commit 2b07f22672
5 changed files with 70 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
tools/goctl/api/dartgen/gendata.go
View File

@@ -4,7 +4,6 @@ import (
"os"
"text/template"
"github.com/tal-tech/go-zero/core/logx"
"github.com/tal-tech/go-zero/tools/goctl/api/spec"
)
@@ -33,30 +32,27 @@ class {{.Name}}{
`
func genData(dir string, api *spec.ApiSpec) error {
e := os.MkdirAll(dir, 0755)
if e != nil {
logx.Error(e)
return e
}
e = genTokens(dir)
if e != nil {
logx.Error(e)
return e
err := os.MkdirAll(dir, 0755)
if err != nil {
return err
}
file, e := os.OpenFile(dir+api.Info.Title+".dart", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
if e != nil {
logx.Error(e)
return e
err = genTokens(dir)
if err != nil {
return err
}
file, err := os.OpenFile(dir+api.Service.Name+".dart", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
return err
}
defer file.Close()
t := template.New("dataTemplate")
t = t.Funcs(funcMap)
t, e = t.Parse(dataTemplate)
if e != nil {
logx.Error(e)
return e
t, err = t.Parse(dataTemplate)
if err != nil {
return err
}
return t.Execute(file, api)
@@ -67,12 +63,13 @@ func genTokens(dir string) error {
if fileExists(path) {
return nil
}
tokensFile, e := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
if e != nil {
logx.Error(e)
return e
tokensFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
if err != nil {
return err
}
defer tokensFile.Close()
tokensFile.WriteString(tokensFileContent)
return nil
_, err = tokensFile.WriteString(tokensFileContent)
return err
}