package config import "fmt" // type Config struct { CfgBase MySQLInfo MysqlDbInfo `toml:"mysql_info"` OutDir string `toml:"out_dir"` Simple bool `toml:"simple"` IsJsonTag bool `toml:"isJsonTag"` SingularTable bool `toml:"singular_table"` } //mysql 数据库信息 type MysqlDbInfo struct { Host string `validate:"required"` //地址 Port int `validate:"required"` //端口号 Username string `validate:"required"` //用户名 Password string `validate:"required"` //密码 Database string `validate:"required"` //数据库名 } //更新mysql配置信息 func SetMysqlDbInfo(info *MysqlDbInfo) { _map.MySQLInfo = *info } //获取mysql配置信息 func GetMysqlDbInfo() MysqlDbInfo { return _map.MySQLInfo } //获取mysql 连接字符串 func GetMysqlConStr() string { return fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=True&loc=Local", _map.MySQLInfo.Username, _map.MySQLInfo.Password, _map.MySQLInfo.Host, _map.MySQLInfo.Port, _map.MySQLInfo.Database, ) } //设置输出目录 func SetOutDir(outDir string) { _map.OutDir = outDir } //获取输出目录 func GetOutDir() string { return _map.OutDir } //设置禁用表名复数 func SetSingularTable(b bool) { _map.SingularTable = b } //获取禁用表名复数 func GetSingularTable() bool { return _map.SingularTable } //简单输出 func GetSimple() bool { return _map.Simple } //json标记 func GetIsJsonTag() bool { return _map.IsJsonTag }