This commit is contained in:
xxj
2021-03-20 13:27:31 +08:00
parent 0153517cda
commit 8250592b73
6 changed files with 43 additions and 41 deletions

View File

@@ -9,24 +9,24 @@ import (
// Config custom config struct
type Config struct {
CfgBase `yaml:"base"`
DBInfo DBInfo `yaml:"db_info"`
OutDir string `yaml:"out_dir"`
URLTag string `yaml:"url_tag"` // url tag
Language string `yaml:"language"` // language
DbTag string `yaml:"db_tag"` // 数据库标签gormt,db
Simple bool `yaml:"simple"`
IsWEBTag bool `yaml:"is_web_tag"`
IsWebTagPkHidden bool `yaml:"is_web_tag_pk_hidden"` // web标记是否隐藏主键
IsForeignKey bool `yaml:"is_foreign_key"`
IsOutSQL bool `yaml:"is_out_sql"`
IsOutFunc bool `yaml:"is_out_func"`
IsGUI bool `yaml:"is_gui"` //
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 DBInfo `yaml:"db_info"`
OutDir string `yaml:"out_dir"`
URLTag string `yaml:"url_tag"` // url tag
Language string `yaml:"language"` // language
DbTag string `yaml:"db_tag"` // 数据库标签gormt,db
Simple bool `yaml:"simple"`
IsWEBTag bool `yaml:"is_web_tag"`
IsWebTagPkHidden bool `yaml:"is_web_tag_pk_hidden"` // web标记是否隐藏主键
IsForeignKey bool `yaml:"is_foreign_key"`
IsOutSQL bool `yaml:"is_out_sql"`
IsOutFunc bool `yaml:"is_out_func"`
IsGUI bool `yaml:"is_gui"` //
IsTableName bool `yaml:"is_table_name"`
IsNullToPoint bool `yaml:"is_null_to_point"` // null to porint
TablePrefix string `yaml:"table_prefix"` // 表前缀
SelfTypeDef map[string]string `yaml:"self_type_define"`
OutFileName string `yaml:"out_file_name"`
WebTagType int `yaml:"web_tag_type"` // 默认小驼峰
}
// DBInfo mysql database information. mysql 数据库信息
@@ -225,18 +225,18 @@ func GetTablePrefix() string {
return _map.TablePrefix
}
// SetSelfDefineTypeMysqlDicMap 设置自定义字段映射
func SetSelfDefineTypeMysqlDicMap(data map[string]string) {
_map.SelfDefineTypeMysqlDicMap = data
// SetSelfTypeDefine 设置自定义字段映射
func SetSelfTypeDefine(data map[string]string) {
_map.SelfTypeDef = data
}
// GetSelfDefineTypeMysqlDicMap 获取自定义字段映射
func GetSelfDefineTypeMysqlDicMap() map[string]string {
return _map.SelfDefineTypeMysqlDicMap
// GetSelfTypeDefine 获取自定义字段映射
func GetSelfTypeDefine() map[string]string {
return _map.SelfTypeDef
}
// SetOutFileName 设置输出文件名
func SetOutFileName(s string) {
func SetOutFileName(s string) {
_map.OutFileName = s
}

View File

@@ -44,9 +44,8 @@ var _map = Config{
IsOutFunc: true,
IsGUI: false,
TablePrefix: "",
SelfDefineTypeMysqlDicMap: make(map[string]string),
OutFileName: "",
WebTagType: 0,
SelfTypeDef: make(map[string]string),
WebTagType: 0,
}
var configPath string