118 lines
2.8 KiB
Go
118 lines
2.8 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
type Config struct {
|
|
Server ServerConfig
|
|
Database DatabaseConfig
|
|
JWT JWTConfig
|
|
Storage StorageConfig
|
|
}
|
|
|
|
type ServerConfig struct {
|
|
Port string
|
|
Mode string
|
|
}
|
|
|
|
type DatabaseConfig struct {
|
|
Host string
|
|
Port string
|
|
User string
|
|
Password string
|
|
Name string
|
|
SSLMode string
|
|
}
|
|
|
|
type JWTConfig struct {
|
|
Secret string
|
|
ExpireTime time.Duration
|
|
}
|
|
|
|
// StorageConfig S3 兼容对象存储（MinIO / OSS / COS / S3 共用字段）
|
|
type StorageConfig struct {
|
|
Endpoint string // 如 localhost:9100（开发 MinIO）或 oss-cn-hangzhou.aliyuncs.com
|
|
AccessKey string
|
|
SecretKey string
|
|
Bucket string
|
|
UseSSL bool
|
|
Region string
|
|
PublicBase string // 浏览器可访问前缀，如 http://localhost:9100/kefu
|
|
MaxUploadMB int
|
|
MaxImageEdge int // 最长边像素
|
|
WebPQuality float32
|
|
}
|
|
|
|
func Load() *Config {
|
|
return &Config{
|
|
Server: ServerConfig{
|
|
Port: getEnv("SERVER_PORT", "8080"),
|
|
Mode: getEnv("GIN_MODE", "debug"),
|
|
},
|
|
Database: DatabaseConfig{
|
|
Host: getEnv("DB_HOST", "localhost"),
|
|
Port: getEnv("DB_PORT", "5433"),
|
|
User: getEnv("DB_USER", "postgres"),
|
|
Password: getEnv("DB_PASSWORD", "postgres"),
|
|
Name: getEnv("DB_NAME", "kefu_cloud"),
|
|
SSLMode: getEnv("DB_SSLMODE", "disable"),
|
|
},
|
|
JWT: JWTConfig{
|
|
Secret: getEnv("JWT_SECRET", "kefu-cloud-secret-key"),
|
|
ExpireTime: 24 * time.Hour,
|
|
},
|
|
Storage: StorageConfig{
|
|
// 默认映射宿主机 9100→容器 9000，避免与其它项目 MinIO 冲突
|
|
Endpoint: getEnv("STORAGE_ENDPOINT", "localhost:9100"),
|
|
AccessKey: getEnv("STORAGE_ACCESS_KEY", "minioadmin"),
|
|
SecretKey: getEnv("STORAGE_SECRET_KEY", "minioadmin"),
|
|
Bucket: getEnv("STORAGE_BUCKET", "kefu"),
|
|
UseSSL: getEnvBool("STORAGE_USE_SSL", false),
|
|
Region: getEnv("STORAGE_REGION", "us-east-1"),
|
|
PublicBase: getEnv("STORAGE_PUBLIC_BASE_URL", "http://localhost:9100/kefu"),
|
|
MaxUploadMB: getEnvInt("STORAGE_MAX_UPLOAD_MB", 10),
|
|
MaxImageEdge: getEnvInt("STORAGE_MAX_IMAGE_EDGE", 1920),
|
|
WebPQuality: float32(getEnvInt("STORAGE_WEBP_QUALITY", 80)),
|
|
},
|
|
}
|
|
}
|
|
|
|
func (d *DatabaseConfig) DSN() string {
|
|
return "host=" + d.Host + " port=" + d.Port + " user=" + d.User +
|
|
" password=" + d.Password + " dbname=" + d.Name + " sslmode=" + d.SSLMode
|
|
}
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if v := os.Getenv(key); v != "" {
|
|
return v
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func getEnvBool(key string, fallback bool) bool {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
b, err := strconv.ParseBool(v)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return b
|
|
}
|
|
|
|
func getEnvInt(key string, fallback int) int {
|
|
v := os.Getenv(key)
|
|
if v == "" {
|
|
return fallback
|
|
}
|
|
n, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
return fallback
|
|
}
|
|
return n
|
|
}
|