Files
pangolin/server/internal/config/config.go
T
wangjia f3471ae139 feat(server/db): 数据层多库支持(1/4)— 连接分派 + 双方言迁移管线
- config 增 DB_DRIVER(mysql 默认 | sqlite);DSN 对 sqlite 为文件路径
- db.OpenDriver 按驱动分派:sqlite 用 modernc(纯 Go 免 CGO)+ WAL/
  busy_timeout/foreign_keys/_txlock=immediate;mysql 路径不变
- store.Open 分派;mysql 保留 UTC/collation 断言,sqlite 跳过
- 迁移拆 migrations/{mysql,sqlite}/ 双套,embed 双 FS,migrate 按驱动选源
  与 golang-migrate 驱动;修复 m.Close() 误关调用方 *sql.DB 的坑
- cmd/migrate 串入 DB_DRIVER;集成测试 MigrateUp 签名更新
- 新增 SQLite 时间往返 smoke 测试与端到端迁移测试(免 docker)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-18 00:01:03 +08:00

137 lines
4.6 KiB
Go

// Package config holds application configuration loaded from environment variables.
package config
import (
"fmt"
"os"
"strings"
"time"
)
// Config holds all application-level configuration.
type Config struct {
// Driver selects the database engine: "mysql" (default) or "sqlite".
// Read from DB_DRIVER. Switching engines is a single env var change.
Driver string
// DSN is the database connection string.
// mysql: user:password@tcp(host:port)/dbname?parseTime=true&loc=UTC&time_zone=%%27UTC%%27
// sqlite: a file path (e.g. /var/lib/pangolin/pangolin.db) or :memory:
DSN string
// RedisAddr is the Redis server address (host:port).
RedisAddr string
RedisPassword string
RedisDB int
// WebhookSecret is the HMAC-SHA256 shared secret for the card-store webhook.
// Must be set; no default.
WebhookSecret string
// RedeemFailMax is the number of consecutive redeem failures before a 1-hour lock.
// Default: 5
RedeemFailMax int
// RedeemLockDuration is how long the lock lasts after hitting RedeemFailMax.
// Default: 1 hour
RedeemLockDuration time.Duration
// WebhookTimestampTolerance is the ±window for webhook timestamp validation.
// Default: 5 minutes
WebhookTimestampTolerance time.Duration
// WebhookNonceTTL is how long a webhook nonce is kept in Redis to prevent replay.
// Should be > 2 * WebhookTimestampTolerance. Default: 15 minutes.
WebhookNonceTTL time.Duration
// ── Auth / JWT (RS256) ────────────────────────────────────────────────
// JWTPrivateKeyPath is the PEM file holding the active RS256 signing key.
JWTPrivateKeyPath string
// JWTKeyID is the `kid` written into the JWT header (identifies the signing key).
JWTKeyID string
// JWTPublicKeys maps kid -> PEM public-key file path. It must include the
// active key's kid and may carry previous keys still accepted during
// rotation. Parsed from JWT_PUBLIC_KEYS="kid1:/path1,kid2:/path2".
JWTPublicKeys map[string]string
// ─── gRPC / agent server ──────────────────────────────────────────────
// GRPCAddr is the listen address for the mTLS gRPC agent server (host:port).
// Optional: if empty, the gRPC server is not started.
// Example: ":9443"
GRPCAddr string
// CAKeyPath and CACertPath are file paths for the Pangolin Node CA.
// Required when GRPCAddr is set.
CAKeyPath string
CACertPath string
// GRPCCertPath and GRPCKeyPath are the TLS certificate/key used by the gRPC
// listener for its own transport credential (distinct from the node CA).
// Required when GRPCAddr is set.
GRPCCertPath string
GRPCKeyPath string
}
// FromEnv reads configuration from environment variables.
// Returns an error if required variables are missing.
func FromEnv() (*Config, error) {
c := &Config{
Driver: os.Getenv("DB_DRIVER"),
DSN: os.Getenv("DB_DSN"),
RedisAddr: getEnvDefault("REDIS_ADDR", "127.0.0.1:6379"),
RedisPassword: os.Getenv("REDIS_PASSWORD"),
RedisDB: 0,
WebhookSecret: os.Getenv("WEBHOOK_SECRET"),
RedeemFailMax: 5,
RedeemLockDuration: time.Hour,
WebhookTimestampTolerance: 5 * time.Minute,
WebhookNonceTTL: 15 * time.Minute,
JWTPrivateKeyPath: os.Getenv("JWT_PRIVATE_KEY_PATH"),
JWTKeyID: os.Getenv("JWT_KEY_ID"),
JWTPublicKeys: parseKeyMap(os.Getenv("JWT_PUBLIC_KEYS")),
// gRPC server (optional)
GRPCAddr: os.Getenv("GRPC_ADDR"),
CAKeyPath: os.Getenv("CA_KEY_PATH"),
CACertPath: os.Getenv("CA_CERT_PATH"),
GRPCCertPath: os.Getenv("GRPC_CERT_PATH"),
GRPCKeyPath: os.Getenv("GRPC_KEY_PATH"),
}
if c.DSN == "" {
return nil, fmt.Errorf("config: DB_DSN is required")
}
if c.WebhookSecret == "" {
return nil, fmt.Errorf("config: WEBHOOK_SECRET is required")
}
return c, nil
}
func getEnvDefault(key, def string) string {
if v := os.Getenv(key); v != "" {
return v
}
return def
}
// parseKeyMap parses a "kid1:/path1,kid2:/path2" string into a map. Empty input
// yields a nil map. Malformed entries (missing ':') are skipped.
func parseKeyMap(raw string) map[string]string {
if raw == "" {
return nil
}
m := map[string]string{}
for _, pair := range strings.Split(raw, ",") {
pair = strings.TrimSpace(pair)
if pair == "" {
continue
}
i := strings.IndexByte(pair, ':')
if i <= 0 || i == len(pair)-1 {
continue
}
m[strings.TrimSpace(pair[:i])] = strings.TrimSpace(pair[i+1:])
}
return m
}