-
Notifications
You must be signed in to change notification settings - Fork 1
/
args.go
109 lines (95 loc) · 3.85 KB
/
args.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
99
100
101
102
103
104
105
106
107
108
109
package main
import (
"github.com/jasonlabz/gentol/configx"
"github.com/pborman/getopt/v2"
"os"
"strings"
"sync"
)
var once sync.Once
type Gentol struct {
CurrentPkg string
}
var (
dbType = getopt.StringLong("db_type", 0, "postgres", "database type such as [mysql, sqlserver, postgres, oracle, greenplum etc. ]")
dsn = getopt.StringLong("dsn", 0, "", "option database connection string, or provide host and port ...")
host = getopt.StringLong("host", 'h', "", "db host, if there is a dsn, ignore it")
port = getopt.IntLong("port", 'p', 0, "db port, if there is a dsn, ignore it")
username = getopt.StringLong("username", 'U', "", "db username, if there is a dsn, ignore it")
password = getopt.StringLong("password", 'P', "", "db password, if there is a dsn, ignore it")
database = getopt.StringLong("database", 'd', "", "database to for db table")
module = getopt.StringLong("module", 'm', "", "module name for go project")
schema = getopt.StringLong("schema", 's', "", "schema to for db table")
table = getopt.StringLong("table", 't', "", "table name to build struct from")
templateDir = getopt.StringLong("template_dir", 0, "./template", "Template Dir")
modelPath = getopt.StringLong("model", 0, "dal/db/model", "name to set for model package")
daoPath = getopt.StringLong("dao", 0, "dal/db/dao", "name to set for dao package")
servicePath = getopt.StringLong("service", 0, "server/service", "name to set for service package")
grpcPath = getopt.StringLong("grpc", 0, "./grpc", "name to set for grpc package")
outDir = getopt.StringLong("out", 0, ".", "output dir")
jsonNameFormat = getopt.StringLong("json_format", 0, "snake", "json name format [snake | upper_camel | lower_camel]")
protoNameFormat = getopt.StringLong("protobuf_format", 0, "snake", "proto name format [snake | upper_camel | lower_camel]")
gogoProtoImport = getopt.StringLong("gogoproto", 0, "", "location of gogo import ")
onlyModel = getopt.BoolLong("only_model", 0, "overwrite existing files (default)", "disable overwriting files")
useHook = getopt.BoolLong("gen_hook", 0, "disable gorm hook file (default)", "gorm hook file")
useSQLNullable = getopt.BoolLong("use_sql_nullable", 0, "use sql.Null if use_sql_nullable true, default use guregu")
addProtobufAnnotation = getopt.BoolLong("proto", 0, "add protobuf annotations (tags)", "")
runGoFmt = getopt.BoolLong("rungofmt", 0, "run gofmt on output dir", "")
DefaultDBName = "_default_db_"
)
func init() {
exist := IsExist("./conf/table.yaml")
if !exist {
exist = IsExist("./table.yaml")
}
if exist {
configx.Init()
} else {
// check args
getopt.Lookup("db_type").SetGroup("check")
getopt.RequiredGroup("check")
getopt.ParseV2()
// fill args
configx.TableConfigs.AddProtobufAnnotation = *addProtobufAnnotation
configx.TableConfigs.RunGoFmt = *runGoFmt
configx.TableConfigs.JsonFormat = *jsonNameFormat
configx.TableConfigs.ProtobufFormat = *protoNameFormat
configx.TableConfigs.GoModule = *module
databaseConfig := &configx.DBTableInfo{
DBName: DefaultDBName,
DBType: *dbType,
DSN: *dsn,
OnlyModel: *onlyModel,
GenHook: *useHook,
ModelPath: *modelPath,
DaoPath: *daoPath,
ServicePath: *servicePath,
Host: *host,
Port: *port,
User: *username,
Password: *password,
UseSQLNullable: *useSQLNullable,
Tables: []*configx.TableInfo{
{
SchemaName: *schema,
TableList: func() []string {
if *table != "" {
return strings.Split(*table, ",")
}
return []string{}
}(),
},
},
}
databaseConfig.GenDSN()
configx.TableConfigs.Configs = []*configx.DBTableInfo{
databaseConfig,
}
}
}
func argHandler() {
// gentol new
secondStr := os.Args[1]
if secondStr == "new" {
}
}