235 lines
7.6 KiB
Go
235 lines
7.6 KiB
Go
// Package config 负责 telesrv 运行配置的加载与校验。
|
||
package config
|
||
|
||
import (
|
||
"net"
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
// Config 是 telesrv 的运行配置。
|
||
type Config struct {
|
||
// ListenAddr 是 MTProto TCP 监听地址。
|
||
// 需与 TDesktop patch 指向的自建 DC 地址/端口一致（记录于 docs/tdesktop-patch-notes.md）。
|
||
ListenAddr string
|
||
// AdvertiseIP 是写入 help.getConfig DCOptions 的对外可达 IP（客户端据此连接本 DC）。
|
||
AdvertiseIP string
|
||
// RSAKeyPath 是 server RSA 私钥的 PEM 路径；不存在时自动生成。
|
||
RSAKeyPath string
|
||
// DC 是本 server 的 DC ID。
|
||
DC int
|
||
|
||
// PostgresDSN 是业务数据（auth_key / user / authorization 等）持久化的 PostgreSQL 连接串。
|
||
// 依赖由 deploy/docker-compose.yml 启动；职责划分见 docs/persistence-layer.md。
|
||
PostgresDSN string
|
||
// PostgresMaxConns 是 pgxpool 最大连接数。<=0 用 pgx 默认（max(4, NumCPU)，生产偏小）。
|
||
// 需覆盖发送事务 + outbox worker 并发 + RPC 读，过小会在高并发下排队（表现为尾延迟突刺）。
|
||
PostgresMaxConns int
|
||
// PostgresMinConns 是启动时预热的 pgxpool 连接数，降低 TDesktop 冷启动并发 RPC 的建连等待。
|
||
PostgresMinConns int
|
||
// RedisAddr 是高频易失态（验证码、限流计数、update 队列）的 Redis 地址。
|
||
RedisAddr string
|
||
// RedisPassword 是 Redis 密码；开发默认空。
|
||
RedisPassword string
|
||
// RedisDB 是 Redis 逻辑库编号。
|
||
RedisDB int
|
||
|
||
// DevAuthCode 是开发固定验证码；生产短信/风控不在当前范围内。
|
||
DevAuthCode string
|
||
// LangPackSeedDir 是 TDesktop 语言包 .strings 种子目录。
|
||
LangPackSeedDir string
|
||
// BlobDir 是本地磁盘 blob backend 根目录（媒体文件字节内容）。
|
||
BlobDir string
|
||
// StickerSeedDir 是 reaction / sticker 资源种子目录（导入到 documents/sticker_sets + blob）。
|
||
StickerSeedDir string
|
||
// StickerSeedMaxSets 限制导入的常规贴纸集数量（避免启动时导入过多包），<=0 表示不限。
|
||
StickerSeedMaxSets int
|
||
|
||
// OutboxWorkers 是并发 claim 的 outbox worker 数。默认 1，保证同一用户 pts update
|
||
// 在线投递顺序与持久化顺序一致；后续需要吞吐时应改成按 target_user_id 分片的串行 worker。
|
||
OutboxWorkers int
|
||
// OutboxBatch 是 transactional outbox worker 每次 claim 的最大条数。
|
||
// 调大提升吞吐、增大单批 PG/推送压力；调小降低延迟抖动。配套压测见 docs/message-module.md。
|
||
OutboxBatch int
|
||
// OutboxInterval 是 outbox worker 两次 claim 之间的轮询间隔。
|
||
OutboxInterval time.Duration
|
||
// OutboxLeaseTimeout 是 'dispatching' 行被判定为租约过期、允许其它 worker 重新 claim 的时长。
|
||
// 取值需大于单批投递耗时，否则会重复推送；过大则 worker 崩溃后积压恢复变慢。
|
||
OutboxLeaseTimeout time.Duration
|
||
// OutboundPushTimeout 是 best-effort updates 推送等待 outbound 队列接受的最长时间。
|
||
OutboundPushTimeout time.Duration
|
||
// UpdateEventRetention 是 durable update log 保留期；只清理已被水位/state 覆盖的事件。
|
||
UpdateEventRetention time.Duration
|
||
// RetentionInterval 是 retention worker 的运行间隔。
|
||
RetentionInterval time.Duration
|
||
// RetentionBatch 是单次 retention 最多删除的行数。
|
||
RetentionBatch int
|
||
}
|
||
|
||
// Load 从环境变量读取配置并填充默认值。第一阶段不做严格校验。
|
||
func Load() (Config, error) {
|
||
advertiseIP := os.Getenv("TELESRV_ADVERTISE_IP")
|
||
if advertiseIP == "" {
|
||
advertiseIP = defaultAdvertiseIP()
|
||
}
|
||
cfg := Config{
|
||
ListenAddr: envOr("TELESRV_LISTEN", "0.0.0.0:2398"),
|
||
AdvertiseIP: advertiseIP,
|
||
RSAKeyPath: envOr("TELESRV_RSA_KEY", "data/server_rsa.pem"),
|
||
DC: envIntOr("TELESRV_DC", 2),
|
||
|
||
PostgresDSN: envOr("TELESRV_POSTGRES_DSN", "postgres://telesrv:telesrv@localhost:5432/telesrv?sslmode=disable"),
|
||
PostgresMaxConns: envIntOr("TELESRV_POSTGRES_MAX_CONNS", 50),
|
||
PostgresMinConns: envIntOr("TELESRV_POSTGRES_MIN_CONNS", 16),
|
||
RedisAddr: envOr("TELESRV_REDIS_ADDR", "localhost:6399"),
|
||
RedisPassword: envOr("TELESRV_REDIS_PASSWORD", ""),
|
||
RedisDB: envIntOr("TELESRV_REDIS_DB", 0),
|
||
|
||
DevAuthCode: envOr("TELESRV_DEV_AUTH_CODE", "12345"),
|
||
LangPackSeedDir: envOr("TELESRV_LANGPACK_SEED_DIR", "data/langpack"),
|
||
BlobDir: envOr("TELESRV_BLOB_DIR", "data/blobs"),
|
||
StickerSeedDir: envOr("TELESRV_STICKER_SEED_DIR", "data/sticker-seed"),
|
||
StickerSeedMaxSets: envIntOr("TELESRV_STICKER_SEED_MAX_SETS", 40),
|
||
|
||
OutboxWorkers: envIntOr("TELESRV_OUTBOX_WORKERS", 1),
|
||
OutboxBatch: envIntOr("TELESRV_OUTBOX_BATCH", 100),
|
||
OutboxInterval: envDurationOr("TELESRV_OUTBOX_INTERVAL", 200*time.Millisecond),
|
||
OutboxLeaseTimeout: envDurationOr("TELESRV_OUTBOX_LEASE_TIMEOUT", 30*time.Second),
|
||
OutboundPushTimeout: envDurationOr("TELESRV_OUTBOUND_PUSH_TIMEOUT", 200*time.Millisecond),
|
||
UpdateEventRetention: envDurationOr("TELESRV_UPDATE_EVENT_RETENTION", 168*time.Hour),
|
||
RetentionInterval: envDurationOr("TELESRV_RETENTION_INTERVAL", time.Hour),
|
||
RetentionBatch: envIntOr("TELESRV_RETENTION_BATCH", 10000),
|
||
}
|
||
return cfg, nil
|
||
}
|
||
|
||
func defaultAdvertiseIP() string {
|
||
if ip := detectAdvertiseIP(); ip != "" {
|
||
return ip
|
||
}
|
||
return "127.0.0.1"
|
||
}
|
||
|
||
func detectAdvertiseIP() string {
|
||
ifaces, err := net.Interfaces()
|
||
if err != nil {
|
||
return ""
|
||
}
|
||
var physical []net.IP
|
||
var virtual []net.IP
|
||
for _, iface := range ifaces {
|
||
if iface.Flags&net.FlagUp == 0 || iface.Flags&net.FlagLoopback != 0 {
|
||
continue
|
||
}
|
||
addrs, err := iface.Addrs()
|
||
if err != nil {
|
||
continue
|
||
}
|
||
for _, addr := range addrs {
|
||
ip := ipFromAddr(addr)
|
||
if ip == nil || !isPrivateIPv4(ip) {
|
||
continue
|
||
}
|
||
if likelyVirtualInterface(iface.Name) {
|
||
virtual = append(virtual, ip)
|
||
continue
|
||
}
|
||
physical = append(physical, ip)
|
||
}
|
||
}
|
||
if ip := preferredAdvertiseIP(physical); ip != "" {
|
||
return ip
|
||
}
|
||
return preferredAdvertiseIP(virtual)
|
||
}
|
||
|
||
func ipFromAddr(addr net.Addr) net.IP {
|
||
switch v := addr.(type) {
|
||
case *net.IPNet:
|
||
return v.IP.To4()
|
||
case *net.IPAddr:
|
||
return v.IP.To4()
|
||
default:
|
||
return nil
|
||
}
|
||
}
|
||
|
||
func preferredAdvertiseIP(ips []net.IP) string {
|
||
for _, ip := range ips {
|
||
if v4 := ip.To4(); v4 != nil && v4[0] == 10 {
|
||
return v4.String()
|
||
}
|
||
}
|
||
for _, ip := range ips {
|
||
if v4 := ip.To4(); v4 != nil && v4[0] == 192 && v4[1] == 168 {
|
||
return v4.String()
|
||
}
|
||
}
|
||
for _, ip := range ips {
|
||
if v4 := ip.To4(); v4 != nil && v4[0] == 172 && v4[1] >= 16 && v4[1] <= 31 {
|
||
return v4.String()
|
||
}
|
||
}
|
||
if len(ips) == 0 {
|
||
return ""
|
||
}
|
||
return ips[0].String()
|
||
}
|
||
|
||
func isPrivateIPv4(ip net.IP) bool {
|
||
v4 := ip.To4()
|
||
if v4 == nil {
|
||
return false
|
||
}
|
||
return v4[0] == 10 ||
|
||
(v4[0] == 172 && v4[1] >= 16 && v4[1] <= 31) ||
|
||
(v4[0] == 192 && v4[1] == 168)
|
||
}
|
||
|
||
func likelyVirtualInterface(name string) bool {
|
||
name = strings.ToLower(name)
|
||
virtualMarkers := []string{
|
||
"docker",
|
||
"hyper-v",
|
||
"tailscale",
|
||
"virtual",
|
||
"virtualbox",
|
||
"vethernet",
|
||
"vmware",
|
||
"wsl",
|
||
}
|
||
for _, marker := range virtualMarkers {
|
||
if strings.Contains(name, marker) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
func envOr(key, def string) string {
|
||
if v := os.Getenv(key); v != "" {
|
||
return v
|
||
}
|
||
return def
|
||
}
|
||
|
||
func envIntOr(key string, def int) int {
|
||
if v := os.Getenv(key); v != "" {
|
||
if n, err := strconv.Atoi(v); err == nil {
|
||
return n
|
||
}
|
||
}
|
||
return def
|
||
}
|
||
|
||
// envDurationOr 读取 time.ParseDuration 格式（如 "200ms"、"30s"）的时长配置；解析失败回退默认值。
|
||
func envDurationOr(key string, def time.Duration) time.Duration {
|
||
if v := os.Getenv(key); v != "" {
|
||
if d, err := time.ParseDuration(v); err == nil {
|
||
return d
|
||
}
|
||
}
|
||
return def
|
||
}
|