-
Notifications
You must be signed in to change notification settings - Fork 249
/
main.go
98 lines (90 loc) · 2.22 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
package codegen
import (
"bufio"
"github.com/sirupsen/logrus"
"github.com/unionj-cloud/go-doudou/cmd/internal/astutils"
"github.com/unionj-cloud/go-doudou/version"
"os"
"path/filepath"
"strings"
"text/template"
)
var mainTmpl = `/**
* Generated by go-doudou {{.Version}}.
* You can edit it as your need.
*/
package main
import (
ddhttp "github.com/unionj-cloud/go-doudou/framework/http"
{{.ServiceAlias}} "{{.ServicePackage}}"
"{{.ConfigPackage}}"
"{{.HttpPackage}}"
)
func main() {
conf := config.LoadFromEnv()
svc := {{.ServiceAlias}}.New{{.SvcName}}(conf)
handler := httpsrv.New{{.SvcName}}Handler(svc)
srv := ddhttp.NewDefaultHttpSrv()
srv.AddRoute(httpsrv.Routes(handler)...)
srv.Run()
}
`
// GenMain generates main function
func GenMain(dir string, ic astutils.InterfaceCollector) {
var (
err error
modfile string
modName string
mainfile string
firstLine string
f *os.File
tpl *template.Template
cmdDir string
svcName string
alias string
)
cmdDir = filepath.Join(dir, "cmd")
if err = MkdirAll(cmdDir, os.ModePerm); err != nil {
panic(err)
}
svcName = ic.Interfaces[0].Name
alias = ic.Package.Name
mainfile = filepath.Join(cmdDir, "main.go")
if _, err = Stat(mainfile); os.IsNotExist(err) {
modfile = filepath.Join(dir, "go.mod")
if f, err = Open(modfile); err != nil {
panic(err)
}
reader := bufio.NewReader(f)
firstLine, _ = reader.ReadString('\n')
modName = strings.TrimSpace(strings.TrimPrefix(firstLine, "module"))
if f, err = Create(mainfile); err != nil {
panic(err)
}
defer f.Close()
if tpl, err = template.New("main.go.tmpl").Parse(mainTmpl); err != nil {
panic(err)
}
if err = tpl.Execute(f, struct {
ServicePackage string
ConfigPackage string
DbPackage string
HttpPackage string
SvcName string
ServiceAlias string
Version string
}{
ServicePackage: modName,
ConfigPackage: modName + "/config",
DbPackage: modName + "/db",
HttpPackage: modName + "/transport/httpsrv",
SvcName: svcName,
ServiceAlias: alias,
Version: version.Release,
}); err != nil {
panic(err)
}
} else {
logrus.Warnf("file %s already exists", mainfile)
}
}