mirror of
https://github.com/DullJZ/s3-balance.git
synced 2026-07-01 16:11:40 +08:00
228 lines
7.1 KiB
Go
228 lines
7.1 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"time"
|
||
|
||
"gopkg.in/yaml.v3"
|
||
)
|
||
|
||
// Config 全局配置结构
|
||
type Config struct {
|
||
Server ServerConfig `yaml:"server"`
|
||
Database DatabaseConfig `yaml:"database"`
|
||
Buckets []BucketConfig `yaml:"buckets"`
|
||
Balancer BalancerConfig `yaml:"balancer"`
|
||
Metrics MetricsConfig `yaml:"metrics"`
|
||
S3API S3APIConfig `yaml:"s3api"`
|
||
API APIConfig `yaml:"api"`
|
||
}
|
||
|
||
// ServerConfig 服务器配置
|
||
type ServerConfig struct {
|
||
Host string `yaml:"host"`
|
||
Port int `yaml:"port"`
|
||
ReadTimeout time.Duration `yaml:"read_timeout"`
|
||
WriteTimeout time.Duration `yaml:"write_timeout"`
|
||
IdleTimeout time.Duration `yaml:"idle_timeout"`
|
||
}
|
||
|
||
// BucketConfig S3存储桶配置
|
||
type BucketConfig struct {
|
||
Name string `yaml:"name"` // 桶名称
|
||
Endpoint string `yaml:"endpoint"` // S3端点
|
||
Region string `yaml:"region"` // 区域
|
||
AccessKeyID string `yaml:"access_key_id"` // 访问密钥ID
|
||
SecretAccessKey string `yaml:"secret_access_key"` // 访问密钥
|
||
MaxSize string `yaml:"max_size"` // 最大容量 (例如: "10GB")
|
||
MaxSizeBytes int64 `yaml:"-"` // 内部使用，字节为单位
|
||
Weight int `yaml:"weight"` // 权重 (用于负载均衡)
|
||
Enabled bool `yaml:"enabled"` // 是否启用
|
||
PathStyle bool `yaml:"path_style"` // 是否使用路径风格访问
|
||
Virtual bool `yaml:"virtual"` // 是否为虚拟存储桶（仅S3 API中可见）
|
||
OperationLimits OperationLimitConfig `yaml:"operation_limits"`
|
||
}
|
||
|
||
// OperationLimitConfig 后端操作次数限制配置
|
||
type OperationLimitConfig struct {
|
||
TypeA int `yaml:"type_a"` // 类型A操作上限（0表示不限制）
|
||
TypeB int `yaml:"type_b"` // 类型B操作上限（0表示不限制）
|
||
}
|
||
|
||
// BalancerConfig 负载均衡配置
|
||
type BalancerConfig struct {
|
||
Strategy string `yaml:"strategy"` // 负载均衡策略: "round-robin", "least-space", "weighted"
|
||
HealthCheckPeriod time.Duration `yaml:"health_check_period"` // 健康检查周期
|
||
UpdateStatsPeriod time.Duration `yaml:"update_stats_period"` // 统计更新周期
|
||
RetryAttempts int `yaml:"retry_attempts"` // 重试次数
|
||
RetryDelay time.Duration `yaml:"retry_delay"` // 重试延迟
|
||
}
|
||
|
||
// MetricsConfig 监控指标配置
|
||
type MetricsConfig struct {
|
||
Enabled bool `yaml:"enabled"`
|
||
Path string `yaml:"path"`
|
||
Token string `yaml:"token"` // 可选Token，保护Prometheus端点
|
||
// Port int `yaml:"port"` // 目前未使用，与主服务共享端口
|
||
}
|
||
|
||
// S3APIConfig S3兼容API配置
|
||
type S3APIConfig struct {
|
||
AccessKey string `yaml:"access_key"` // S3访问密钥ID
|
||
SecretKey string `yaml:"secret_key"` // S3秘密访问密钥
|
||
VirtualHost bool `yaml:"virtual_host"` // 是否使用虚拟主机模式
|
||
ProxyMode bool `yaml:"proxy_mode"` // 是否使用代理模式（而非重定向）
|
||
AuthRequired bool `yaml:"auth_required"` // 是否需要认证
|
||
Host string `yaml:"host"` // 用于签名验证的Host（为空则使用请求的Host）
|
||
}
|
||
|
||
// APIConfig 管理API配置
|
||
type APIConfig struct {
|
||
Enabled bool `yaml:"enabled"` // 是否启用管理API
|
||
Token string `yaml:"token"` // API访问令牌
|
||
}
|
||
|
||
// DatabaseConfig 数据库配置
|
||
type DatabaseConfig struct {
|
||
Type string `yaml:"type"` // 数据库类型: sqlite, mysql, postgres
|
||
DSN string `yaml:"dsn"` // 数据源名称
|
||
MaxOpenConns int `yaml:"max_open_conns"` // 最大打开连接数
|
||
MaxIdleConns int `yaml:"max_idle_conns"` // 最大空闲连接数
|
||
ConnMaxLifetime int `yaml:"conn_max_lifetime"` // 连接最大生命周期（秒）
|
||
LogLevel string `yaml:"log_level"` // 日志级别: silent, error, warn, info
|
||
AutoMigrate bool `yaml:"auto_migrate"` // 是否自动迁移
|
||
}
|
||
|
||
// Load 从文件加载配置
|
||
func Load(configPath string) (*Config, error) {
|
||
file, err := os.Open(configPath)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to open config file: %w", err)
|
||
}
|
||
defer file.Close()
|
||
|
||
var config Config
|
||
decoder := yaml.NewDecoder(file)
|
||
if err := decoder.Decode(&config); err != nil {
|
||
return nil, fmt.Errorf("failed to decode config: %w", err)
|
||
}
|
||
|
||
// 解析容量大小
|
||
for i := range config.Buckets {
|
||
if err := config.Buckets[i].ParseMaxSize(); err != nil {
|
||
return nil, fmt.Errorf("failed to parse max size for bucket %s: %w",
|
||
config.Buckets[i].Name, err)
|
||
}
|
||
}
|
||
|
||
// 设置默认值
|
||
config.SetDefaults()
|
||
|
||
return &config, nil
|
||
}
|
||
|
||
// SetDefaults 设置默认配置值
|
||
func (c *Config) SetDefaults() {
|
||
if c.Server.Host == "" {
|
||
c.Server.Host = "0.0.0.0"
|
||
}
|
||
if c.Server.Port == 0 {
|
||
c.Server.Port = 8080
|
||
}
|
||
if c.Server.ReadTimeout == 0 {
|
||
c.Server.ReadTimeout = 30 * time.Second
|
||
}
|
||
if c.Server.WriteTimeout == 0 {
|
||
c.Server.WriteTimeout = 30 * time.Second
|
||
}
|
||
if c.Server.IdleTimeout == 0 {
|
||
c.Server.IdleTimeout = 60 * time.Second
|
||
}
|
||
|
||
if c.Balancer.Strategy == "" {
|
||
c.Balancer.Strategy = "least-space"
|
||
}
|
||
if c.Balancer.HealthCheckPeriod == 0 {
|
||
c.Balancer.HealthCheckPeriod = 30 * time.Second
|
||
}
|
||
if c.Balancer.UpdateStatsPeriod == 0 {
|
||
c.Balancer.UpdateStatsPeriod = 60 * time.Second
|
||
}
|
||
if c.Balancer.RetryAttempts == 0 {
|
||
c.Balancer.RetryAttempts = 3
|
||
}
|
||
if c.Balancer.RetryDelay == 0 {
|
||
c.Balancer.RetryDelay = time.Second
|
||
}
|
||
|
||
if c.Metrics.Path == "" {
|
||
c.Metrics.Path = "/metrics"
|
||
}
|
||
|
||
// 数据库默认值
|
||
if c.Database.Type == "" {
|
||
c.Database.Type = "sqlite"
|
||
}
|
||
if c.Database.Type == "sqlite" && c.Database.DSN == "" {
|
||
c.Database.DSN = "data/s3-balance.db"
|
||
}
|
||
if c.Database.MaxOpenConns == 0 {
|
||
c.Database.MaxOpenConns = 25
|
||
}
|
||
if c.Database.MaxIdleConns == 0 {
|
||
c.Database.MaxIdleConns = 5
|
||
}
|
||
if c.Database.ConnMaxLifetime == 0 {
|
||
c.Database.ConnMaxLifetime = 300
|
||
}
|
||
if c.Database.LogLevel == "" {
|
||
c.Database.LogLevel = "warn"
|
||
}
|
||
|
||
// S3 API默认值
|
||
if c.S3API.AccessKey == "" {
|
||
c.S3API.AccessKey = "AKIAIOSFODNN7EXAMPLE"
|
||
}
|
||
if c.S3API.SecretKey == "" {
|
||
c.S3API.SecretKey = "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY"
|
||
}
|
||
|
||
// 管理API默认值
|
||
if c.API.Token == "" {
|
||
c.API.Token = "your-secure-api-token-here"
|
||
}
|
||
}
|
||
|
||
// ParseMaxSize 解析最大容量字符串为字节
|
||
func (bc *BucketConfig) ParseMaxSize() error {
|
||
if bc.MaxSize == "" {
|
||
bc.MaxSizeBytes = 0 // 无限制
|
||
return nil
|
||
}
|
||
|
||
var size int64
|
||
var unit string
|
||
_, err := fmt.Sscanf(bc.MaxSize, "%d%s", &size, &unit)
|
||
if err != nil {
|
||
return fmt.Errorf("invalid size format: %s", bc.MaxSize)
|
||
}
|
||
|
||
switch unit {
|
||
case "B", "b":
|
||
bc.MaxSizeBytes = size
|
||
case "KB", "kb", "K", "k":
|
||
bc.MaxSizeBytes = size * 1024
|
||
case "MB", "mb", "M", "m":
|
||
bc.MaxSizeBytes = size * 1024 * 1024
|
||
case "GB", "gb", "G", "g":
|
||
bc.MaxSizeBytes = size * 1024 * 1024 * 1024
|
||
case "TB", "tb", "T", "t":
|
||
bc.MaxSizeBytes = size * 1024 * 1024 * 1024 * 1024
|
||
default:
|
||
return fmt.Errorf("unsupported unit: %s", unit)
|
||
}
|
||
|
||
return nil
|
||
}
|