61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"gopkg.in/yaml.v2"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
Address string `yaml:"addr"`
|
||
|
|
Auth string `yaml:"auth"`
|
||
|
|
Name string `yaml:"name"`
|
||
|
|
ShowLog bool `yaml:"showLog"`
|
||
|
|
MaxLifeTime int `yaml:"maxLifeTime"`
|
||
|
|
MaxIdleTime int `yaml:"maxIdleTime"`
|
||
|
|
MaxOpenConnections int `yaml:"maxOpenConnections"`
|
||
|
|
MaxIdleConnections int `yaml:"maxIdleConnections"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Image struct {
|
||
|
|
Dir string `yaml:"dir"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Geo struct {
|
||
|
|
Key string `yaml:"key"`
|
||
|
|
Secret string `yaml:"secret"`
|
||
|
|
URL string `yaml:"url"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Config struct {
|
||
|
|
Database DB `yaml:"database"`
|
||
|
|
Image Image `yaml:"image"`
|
||
|
|
Geo Geo `yaml:"geo"`
|
||
|
|
Listen string `yaml:"listen"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *Config) Save(f string) error {
|
||
|
|
data, err := yaml.Marshal(c)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
err = os.WriteFile(f, data, os.ModePerm)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewConfig(file string) (*Config, error) {
|
||
|
|
data, err := os.ReadFile(file)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
conf := &Config{}
|
||
|
|
err = yaml.Unmarshal(data, conf)
|
||
|
|
return conf, err
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewStrConfig(confStr string) (*Config, error) {
|
||
|
|
cfg := &Config{}
|
||
|
|
err := yaml.Unmarshal([]byte(confStr), cfg)
|
||
|
|
return cfg, err
|
||
|
|
}
|