94 lines
2.9 KiB
Go
94 lines
2.9 KiB
Go
|
// Package config 负责加载和管理应用程序配置
|
|||
|
|
// 从 config.yaml 文件中读取配置，并提供全局访问接口
|
|||
|
|
package config
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"os"
|
|||
|
|
|
|||
|
|
"gopkg.in/yaml.v2"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Config 应用程序全局配置结构体
|
|||
|
|
type Config struct {
|
|||
|
|
Server ServerConfig `yaml:"server"` // 服务器配置
|
|||
|
|
Database DatabaseConfig `yaml:"database"` // 数据库配置
|
|||
|
|
Storage StorageConfig `yaml:"storage"` // 文件存储配置
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ServerConfig 服务器配置
|
|||
|
|
type ServerConfig struct {
|
|||
|
|
Port int `yaml:"port"` // API服务监听端口
|
|||
|
|
WsPort int `yaml:"ws_port"` // WebSocket服务监听端口
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DatabaseConfig 数据库连接配置
|
|||
|
|
type DatabaseConfig struct {
|
|||
|
|
Host string `yaml:"host"` // 数据库主机地址
|
|||
|
|
Port int `yaml:"port"` // 数据库端口
|
|||
|
|
Username string `yaml:"username"` // 数据库用户名
|
|||
|
|
Password string `yaml:"password"` // 数据库密码
|
|||
|
|
DBName string `yaml:"dbname"` // 数据库名称
|
|||
|
|
Charset string `yaml:"charset"` // 字符集
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// StorageConfig 文件存储配置
|
|||
|
|
type StorageConfig struct {
|
|||
|
|
Type string `yaml:"type"` // 存储类型：local/qiniu/tencent/aliyun
|
|||
|
|
Local LocalConfig `yaml:"local"` // 本地存储配置
|
|||
|
|
Qiniu CloudConfig `yaml:"qiniu"` // 七牛云配置
|
|||
|
|
Tencent CloudConfig `yaml:"tencent"` // 腾讯云配置
|
|||
|
|
Aliyun CloudConfig `yaml:"aliyun"` // 阿里云配置
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// LocalConfig 本地存储配置
|
|||
|
|
type LocalConfig struct {
|
|||
|
|
Path string `yaml:"path"` // 本地文件存储目录
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CloudConfig 云存储通用配置
|
|||
|
|
type CloudConfig struct {
|
|||
|
|
AccessKey string `yaml:"access_key"` // 访问密钥ID
|
|||
|
|
SecretKey string `yaml:"secret_key"` // 访问密钥Secret
|
|||
|
|
Bucket string `yaml:"bucket"` // 存储桶名称
|
|||
|
|
Domain string `yaml:"domain"` // 域名（七牛云）
|
|||
|
|
Region string `yaml:"region"` // 区域（腾讯云）
|
|||
|
|
SecretID string `yaml:"secret_id"` // SecretID（腾讯云）
|
|||
|
|
AccessKeyID string `yaml:"access_key_id"` // AccessKeyID（阿里云）
|
|||
|
|
AccessKeySecret string `yaml:"access_key_secret"` // AccessKeySecret（阿里云）
|
|||
|
|
Endpoint string `yaml:"endpoint"` // Endpoint（阿里云）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// App 全局配置实例
|
|||
|
|
var App *Config
|
|||
|
|
|
|||
|
|
// Init 从 config.yaml 文件加载配置到全局变量 App
|
|||
|
|
// 如果文件不存在或读取失败，返回错误
|
|||
|
|
func Init() error {
|
|||
|
|
App = &Config{}
|
|||
|
|
data, err := os.ReadFile("config.yaml")
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
err = yaml.Unmarshal(data, App)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
// 设置默认值
|
|||
|
|
if App.Server.Port == 0 {
|
|||
|
|
App.Server.Port = 16702
|
|||
|
|
}
|
|||
|
|
if App.Server.WsPort == 0 {
|
|||
|
|
App.Server.WsPort = 16703
|
|||
|
|
}
|
|||
|
|
if App.Database.Charset == "" {
|
|||
|
|
App.Database.Charset = "utf8mb4"
|
|||
|
|
}
|
|||
|
|
if App.Storage.Type == "" {
|
|||
|
|
App.Storage.Type = "local"
|
|||
|
|
}
|
|||
|
|
if App.Storage.Local.Path == "" {
|
|||
|
|
App.Storage.Local.Path = "./uploads"
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|