39 lines
741 B
Go
39 lines
741 B
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type AppConfig struct {
|
|
Port int `yaml:"port"`
|
|
Database DatabaseConfig `yaml:"database"`
|
|
WorkerID int64 `yaml:"workerID"`
|
|
DatacenterID int64 `yaml:"datacenterID"`
|
|
}
|
|
|
|
type DatabaseConfig struct {
|
|
Username string `yaml:"username"`
|
|
Password string `yaml:"password"`
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
Database string `yaml:"database"`
|
|
}
|
|
|
|
func LoadConfig() (*AppConfig, error) {
|
|
// 读取文件内容
|
|
data, err := os.ReadFile("config.yaml")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 解析 YAML
|
|
var cfg AppConfig
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &cfg, nil
|
|
}
|