add config with self_define_type_mysql_dic_map out_file_name web_tag_type
This commit is contained in:
@@ -24,6 +24,9 @@ type Config struct {
|
||||
IsTableName bool `yaml:"is_table_name"`
|
||||
IsNullToPoint bool `yaml:"is_null_to_point"` // null to porint
|
||||
TablePrefix string `yaml:"table_prefix"` // 表前缀
|
||||
SelfDefineTypeMysqlDicMap map[string]string `yaml:"self_define_type_mysql_dic_map"`
|
||||
OutFileName string `yaml:"out_file_name"`
|
||||
WebTagType int `yaml:"web_tag_type"`
|
||||
}
|
||||
|
||||
// DBInfo mysql database information. mysql 数据库信息
|
||||
@@ -221,3 +224,33 @@ func SetTablePrefix(t string) {
|
||||
func GetTablePrefix() string {
|
||||
return _map.TablePrefix
|
||||
}
|
||||
|
||||
// SetSelfDefineTypeMysqlDicMap 设置自定义字段映射
|
||||
func SetSelfDefineTypeMysqlDicMap(data map[string]string) {
|
||||
_map.SelfDefineTypeMysqlDicMap = data
|
||||
}
|
||||
|
||||
// GetSelfDefineTypeMysqlDicMap 获取自定义字段映射
|
||||
func GetSelfDefineTypeMysqlDicMap() map[string]string {
|
||||
return _map.SelfDefineTypeMysqlDicMap
|
||||
}
|
||||
|
||||
// SetOutFileName 设置输出文件名
|
||||
func SetOutFileName(s string) {
|
||||
_map.OutFileName = s
|
||||
}
|
||||
|
||||
// GetOutFileName 获取输出文件名
|
||||
func GetOutFileName() string {
|
||||
return _map.OutFileName
|
||||
}
|
||||
|
||||
// SetWebTagType 设置json tag类型
|
||||
func SetWebTagType(i int) {
|
||||
_map.WebTagType = i
|
||||
}
|
||||
|
||||
// GetWebTagType 获取json tag类型
|
||||
func GetWebTagType() int {
|
||||
return _map.WebTagType
|
||||
}
|
||||
|
||||
@@ -44,6 +44,9 @@ var _map = Config{
|
||||
IsOutFunc: true,
|
||||
IsGUI: false,
|
||||
TablePrefix: "",
|
||||
SelfDefineTypeMysqlDicMap: make(map[string]string),
|
||||
OutFileName: "",
|
||||
WebTagType: 0,
|
||||
}
|
||||
|
||||
var configPath string
|
||||
|
||||
@@ -67,6 +67,12 @@ func FilterKeywords(src string) string {
|
||||
|
||||
// getTypeName Type acquisition filtering.类型获取过滤
|
||||
func getTypeName(name string, isNull bool) string {
|
||||
// 优先匹配自定义类型
|
||||
selfDefineTypeMqlDicMap := config.GetSelfDefineTypeMysqlDicMap()
|
||||
if v, ok := selfDefineTypeMqlDicMap[name]; ok {
|
||||
return fixNullToPorint(v, isNull)
|
||||
}
|
||||
|
||||
// Precise matching first.先精确匹配
|
||||
if v, ok := cnf.TypeMysqlDicMp[name]; ok {
|
||||
return fixNullToPorint(v, isNull)
|
||||
|
||||
@@ -29,7 +29,11 @@ func Generate(info DBInfo) (out []GenOutInfo, m _Model) {
|
||||
// struct
|
||||
var stt GenOutInfo
|
||||
stt.FileCtx = m.generate()
|
||||
stt.FileName = info.DbName + ".go"
|
||||
if name := config.GetOutFileName(); name != "" {
|
||||
stt.FileName = name + ".go"
|
||||
} else {
|
||||
stt.FileName = info.DbName + ".go"
|
||||
}
|
||||
|
||||
out = append(out, stt)
|
||||
// ------end
|
||||
@@ -131,7 +135,11 @@ func (m *_Model) genTableElement(cols []ColumnsInfo) (el []genstruct.GenElement)
|
||||
if isPK && config.GetIsWebTagPkHidden() {
|
||||
tmp.AddTag(_tagJSON, "-")
|
||||
} else {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnSmallMarshal(mybigcamel.Marshal(v.Name)))
|
||||
if config.GetWebTagType() == 0 {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnMarshal(v.Name))
|
||||
} else {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnSmallMarshal(mybigcamel.Marshal(v.Name)))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -174,7 +182,12 @@ func (m *_Model) genForeignKey(col ColumnsInfo) (fklist []genstruct.GenElement)
|
||||
|
||||
// json tag
|
||||
if config.GetIsWEBTag() {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnSmallMarshal(mybigcamel.Marshal(v.TableName))+"List")
|
||||
if config.GetWebTagType() == 0 {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnMarshal(v.TableName)+"_list")
|
||||
} else {
|
||||
tmp.AddTag(_tagJSON, mybigcamel.UnSmallMarshal(mybigcamel.Marshal(v.TableName))+"List")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
fklist = append(fklist, tmp)
|
||||
|
||||
Reference in New Issue
Block a user