02aead9615
2.将is_table_name中的column生成单独抽成配置，以免配置的耦合。 3.增加根据表名生成多个model的配置（一个表一个model） ------------------------------------------- 1. Simple is configured as true and does not output Gorm tag 2. Will be_ table_ The column in name generates a separate extraction configuration to avoid the coupling of configuration. 3. Add the configuration of generating multiple models according to the table name (one model for each table)
129 lines
3.0 KiB
Go
129 lines
3.0 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"path"
|
|
|
|
"github.com/xxjwxc/public/mylog"
|
|
|
|
"github.com/xxjwxc/public/dev"
|
|
"github.com/xxjwxc/public/tools"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
// CfgBase base config struct
|
|
type CfgBase struct {
|
|
// SerialNumber string `json:"serial_number" yaml:"serial_number"` // version.版本号
|
|
// ServiceName string `json:"service_name" yaml:"service_name"` // service name .service名字
|
|
// ServiceDisplayname string `json:"service_displayname" yaml:"service_displayname"` // display name .显示名
|
|
// SerciceDesc string `json:"sercice_desc" yaml:"sercice_desc"` // sercice desc .service描述
|
|
IsDev bool `json:"is_dev" yaml:"is_dev"` // Is it a development version?是否是开发版本
|
|
}
|
|
|
|
var _map = Config{
|
|
CfgBase: CfgBase{
|
|
IsDev: false,
|
|
},
|
|
DBInfo: DBInfo{
|
|
Host: "127.0.0.1",
|
|
Port: 3306,
|
|
Username: "root",
|
|
Password: "root",
|
|
Database: "test",
|
|
},
|
|
OutDir: "./model",
|
|
URLTag: "json",
|
|
Language: "中 文",
|
|
DbTag: "gorm",
|
|
Simple: false,
|
|
IsWEBTag: false,
|
|
// SingularTable: true,
|
|
IsForeignKey: true,
|
|
IsOutSQL: false,
|
|
IsOutFunc: true,
|
|
IsGUI: false,
|
|
TablePrefix: "",
|
|
SelfTypeDef: make(map[string]string),
|
|
WebTagType: 0,
|
|
TableNames: "",
|
|
IsColumnName: true,
|
|
IsOutFileByTableName: false,
|
|
}
|
|
|
|
var configPath string
|
|
|
|
func init() {
|
|
configPath = path.Join(tools.GetCurrentDirectory(), "config.yml") // 先找本程序文件夹
|
|
if !tools.CheckFileIsExist(configPath) { // dont find it
|
|
configPath = path.Join(tools.GetModelPath(), "config.yml")
|
|
if !tools.CheckFileIsExist(configPath) {
|
|
mylog.ErrorString("config.yml not exit. using default config")
|
|
}
|
|
}
|
|
|
|
onInit()
|
|
dev.OnSetDev(_map.IsDev)
|
|
}
|
|
|
|
func onInit() {
|
|
err := InitFile(configPath)
|
|
if err != nil {
|
|
fmt.Println("Load config file error: ", err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
// InitFile default value from file .
|
|
func InitFile(filename string) error {
|
|
// if _, e := os.Stat(filename); e != nil {
|
|
// fmt.Println("init default config file: ", filename)
|
|
// if err := SaveToFile(); err == nil {
|
|
// InitFile(filename)
|
|
// return nil
|
|
// } else {
|
|
// fmt.Println("shit,fail", err)
|
|
// }
|
|
// // os.Exit(0)
|
|
// }
|
|
bs, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := yaml.Unmarshal(bs, &_map); err != nil {
|
|
fmt.Println("read config file error: ", err.Error())
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// GetServiceConfig Get service configuration information
|
|
// func GetServiceConfig() (name, displayName, desc string) {
|
|
// name = _map.ServiceName
|
|
// displayName = _map.ServiceDisplayname
|
|
// desc = _map.SerciceDesc
|
|
// return
|
|
// }
|
|
|
|
// GetIsDev is is dev
|
|
func GetIsDev() bool {
|
|
return _map.IsDev
|
|
}
|
|
|
|
// SetIsDev is is dev
|
|
func SetIsDev(b bool) {
|
|
_map.IsDev = b
|
|
}
|
|
|
|
// SaveToFile save config info to file
|
|
func SaveToFile() error {
|
|
d, err := yaml.Marshal(_map)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tools.WriteFile(configPath, []string{
|
|
string(d),
|
|
}, true)
|
|
return nil
|
|
}
|