You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
2.5 KiB
Go
127 lines
2.5 KiB
Go
package gogen
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/logrusorgru/aurora"
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
apiformat "github.com/tal-tech/go-zero/tools/goctl/api/format"
|
|
"github.com/tal-tech/go-zero/tools/goctl/api/parser"
|
|
apiutil "github.com/tal-tech/go-zero/tools/goctl/api/util"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
const tmpFile = "%s-%d"
|
|
|
|
var tmpDir = path.Join(os.TempDir(), "goctl")
|
|
|
|
func GoCommand(c *cli.Context) error {
|
|
apiFile := c.String("api")
|
|
dir := c.String("dir")
|
|
if len(apiFile) == 0 {
|
|
return errors.New("missing -api")
|
|
}
|
|
if len(dir) == 0 {
|
|
return errors.New("missing -dir")
|
|
}
|
|
|
|
return DoGenProject(apiFile, dir)
|
|
}
|
|
|
|
func DoGenProject(apiFile, dir string) error {
|
|
p, err := parser.NewParser(apiFile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
api, err := p.Parse()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
logx.Must(util.MkdirIfNotExist(dir))
|
|
logx.Must(genEtc(dir, api))
|
|
logx.Must(genConfig(dir, api))
|
|
logx.Must(genMain(dir, api))
|
|
logx.Must(genServiceContext(dir, api))
|
|
logx.Must(genTypes(dir, api))
|
|
logx.Must(genHandlers(dir, api))
|
|
logx.Must(genRoutes(dir, api))
|
|
logx.Must(genLogic(dir, api))
|
|
|
|
if err := backupAndSweep(apiFile); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := apiformat.ApiFormatByPath(apiFile); err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Println(aurora.Green("Done."))
|
|
return nil
|
|
}
|
|
|
|
func backupAndSweep(apiFile string) error {
|
|
var err error
|
|
var wg sync.WaitGroup
|
|
|
|
wg.Add(2)
|
|
_ = os.MkdirAll(tmpDir, os.ModePerm)
|
|
|
|
go func() {
|
|
_, fileName := filepath.Split(apiFile)
|
|
_, e := apiutil.Copy(apiFile, fmt.Sprintf(path.Join(tmpDir, tmpFile), fileName, time.Now().Unix()))
|
|
if e != nil {
|
|
err = e
|
|
}
|
|
wg.Done()
|
|
}()
|
|
go func() {
|
|
if e := sweep(); e != nil {
|
|
err = e
|
|
}
|
|
wg.Done()
|
|
}()
|
|
wg.Wait()
|
|
|
|
return err
|
|
}
|
|
|
|
func sweep() error {
|
|
keepTime := time.Now().AddDate(0, 0, -7)
|
|
return filepath.Walk(tmpDir, func(fpath string, info os.FileInfo, err error) error {
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
|
|
pos := strings.LastIndexByte(info.Name(), '-')
|
|
if pos > 0 {
|
|
timestamp := info.Name()[pos+1:]
|
|
seconds, err := strconv.ParseInt(timestamp, 10, 64)
|
|
if err != nil {
|
|
// print error and ignore
|
|
fmt.Println(aurora.Red(fmt.Sprintf("sweep ignored file: %s", fpath)))
|
|
return nil
|
|
}
|
|
|
|
tm := time.Unix(seconds, 0)
|
|
if tm.Before(keepTime) {
|
|
if err := os.Remove(fpath); err != nil {
|
|
fmt.Println(aurora.Red(fmt.Sprintf("failed to remove file: %s", fpath)))
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|