yumi-cron/internal/config/config.go

package config

import (
	"os"
	"strconv"
	"strings"
	"time"
)

type Config struct {
	MySQLDSN                           string
	RedisAddr                          string
	RedisPassword                      string
	RedisDB                            int
	JavaOtherBaseURL                   string
	JavaWalletBaseURL                  string
	WeekStarDefaultSysOrigin           string
	WeekStarTimezone                   string
	WeekStarSettleInterval             time.Duration
	RoomTurnoverDefaultSysOrigin       string
	RoomTurnoverTimezone               string
	RoomTurnoverSettleInterval         time.Duration
	RechargeRewardDefaultSysOrigin     string
	RechargeRewardTimezone             string
	RechargeRewardStorageTimezone      string
	RechargeRewardStartDate            string
	RechargeRewardSettleInterval       time.Duration
	RechargeRewardCoinSellerGoldPerUSD int64
	WeekStarRetryStreamKey             string
	WeekStarRetryConsumerGroup         string
	WeekStarRetryConsumerName          string
	WeekStarRetryBatchSize             int64
	WeekStarRetryBlock                 time.Duration
	WeekStarRetryMinIdle               time.Duration
	Timeout                            time.Duration
	ShutdownTimeout                    time.Duration
	AutoMigrate                        bool
}

func Load() Config {
	return Config{
		MySQLDSN: getEnvAny(
			[]string{"CHATAPP_STORE_MYSQL_DSN", "CRON_MYSQL_DSN", "INVITE_MYSQL_DSN"},
			"root:root@tcp(127.0.0.1:3306)/chatapp3?charset=utf8mb4&parseTime=True&loc=Local",
		),
		RedisAddr: getEnvAny(
			[]string{"CHATAPP_STORE_REDIS_ADDR", "CRON_REDIS_ADDR", "INVITE_REDIS_ADDR"},
			"127.0.0.1:6379",
		),
		RedisPassword: getEnvAny(
			[]string{"CHATAPP_STORE_REDIS_PASSWORD", "CRON_REDIS_PASSWORD", "INVITE_REDIS_PASSWORD"},
			"",
		),
		RedisDB: getEnvIntAny(
			[]string{"CHATAPP_STORE_REDIS_DB", "CRON_REDIS_DB", "INVITE_REDIS_DB"},
			0,
		),
		JavaOtherBaseURL: getEnvAny(
			[]string{"CHATAPP_JAVA_OTHER_BASE_URL", "CRON_JAVA_OTHER_BASE_URL", "GAME_JAVA_OTHER_BASE_URL", "INVITE_JAVA_APP_BASE_URL"},
			"http://127.0.0.1:2400",
		),
		JavaWalletBaseURL: getEnvAny(
			[]string{"CHATAPP_JAVA_WALLET_BASE_URL", "CRON_JAVA_WALLET_BASE_URL", "GAME_JAVA_WALLET_BASE_URL"},
			"http://127.0.0.1:2000",
		),
		WeekStarDefaultSysOrigin: strings.ToUpper(
			getEnvAny([]string{"CHATAPP_WEEK_STAR_DEFAULT_SYS_ORIGIN", "WEEK_STAR_DEFAULT_SYS_ORIGIN"}, "LIKEI"),
		),
		WeekStarTimezone:       getEnvAny([]string{"CHATAPP_WEEK_STAR_TIMEZONE", "WEEK_STAR_TIMEZONE"}, "Asia/Riyadh"),
		WeekStarSettleInterval: time.Duration(getEnvIntAny([]string{"CHATAPP_WEEK_STAR_SETTLE_INTERVAL_SECONDS", "WEEK_STAR_SETTLE_INTERVAL_SECONDS"}, 60)) * time.Second,
		RoomTurnoverDefaultSysOrigin: strings.ToUpper(getEnvAny(
			[]string{"CHATAPP_ROOM_TURNOVER_REWARD_DEFAULT_SYS_ORIGIN", "ROOM_TURNOVER_REWARD_DEFAULT_SYS_ORIGIN", "CHATAPP_WEEK_STAR_DEFAULT_SYS_ORIGIN", "WEEK_STAR_DEFAULT_SYS_ORIGIN"},
			"LIKEI",
		)),
		RoomTurnoverTimezone: getEnvAny(
			[]string{"CHATAPP_ROOM_TURNOVER_REWARD_TIMEZONE", "ROOM_TURNOVER_REWARD_TIMEZONE", "CHATAPP_WEEK_STAR_TIMEZONE", "WEEK_STAR_TIMEZONE"},
			"Asia/Riyadh",
		),
		RoomTurnoverSettleInterval: time.Duration(getEnvIntAny(
			[]string{"CHATAPP_ROOM_TURNOVER_REWARD_SETTLE_INTERVAL_SECONDS", "ROOM_TURNOVER_REWARD_SETTLE_INTERVAL_SECONDS"},
			60,
		)) * time.Second,
		RechargeRewardDefaultSysOrigin: strings.ToUpper(getEnvAny(
			[]string{"CHATAPP_RECHARGE_REWARD_DEFAULT_SYS_ORIGIN", "RECHARGE_REWARD_DEFAULT_SYS_ORIGIN", "CHATAPP_WEEK_STAR_DEFAULT_SYS_ORIGIN", "WEEK_STAR_DEFAULT_SYS_ORIGIN"},
			"LIKEI",
		)),
		RechargeRewardTimezone: getEnvAny(
			[]string{"CHATAPP_RECHARGE_REWARD_TIMEZONE", "RECHARGE_REWARD_TIMEZONE", "CHATAPP_WEEK_STAR_TIMEZONE", "WEEK_STAR_TIMEZONE"},
			"Asia/Riyadh",
		),
		RechargeRewardStorageTimezone: getEnvAny(
			[]string{"CHATAPP_RECHARGE_REWARD_STORAGE_TIMEZONE", "RECHARGE_REWARD_STORAGE_TIMEZONE"},
			"Asia/Shanghai",
		),
		RechargeRewardStartDate: getEnvAny(
			[]string{"CHATAPP_RECHARGE_REWARD_START_DATE", "RECHARGE_REWARD_START_DATE"},
			"2026-05-21",
		),
		RechargeRewardSettleInterval: time.Duration(getEnvIntAny(
			[]string{"CHATAPP_RECHARGE_REWARD_SETTLE_INTERVAL_SECONDS", "RECHARGE_REWARD_SETTLE_INTERVAL_SECONDS"},
			60,
		)) * time.Second,
		RechargeRewardCoinSellerGoldPerUSD: int64(getEnvIntAny(
			[]string{"CHATAPP_RECHARGE_REWARD_COIN_SELLER_GOLD_PER_USD", "RECHARGE_REWARD_COIN_SELLER_GOLD_PER_USD"},
			100000,
		)),
		WeekStarRetryStreamKey:     getEnvAny([]string{"CHATAPP_WEEK_STAR_RETRY_STREAM_KEY", "WEEK_STAR_RETRY_STREAM_KEY"}, "week-star:reward-retry"),
		WeekStarRetryConsumerGroup: getEnvAny([]string{"CHATAPP_WEEK_STAR_RETRY_CONSUMER_GROUP", "WEEK_STAR_RETRY_CONSUMER_GROUP"}, "week-star-retry"),
		WeekStarRetryConsumerName:  getEnvAny([]string{"CHATAPP_WEEK_STAR_RETRY_CONSUMER_NAME", "WEEK_STAR_RETRY_CONSUMER_NAME"}, defaultConsumerName("week-star-retry")),
		WeekStarRetryBatchSize:     int64(getEnvIntAny([]string{"CHATAPP_WEEK_STAR_RETRY_BATCH_SIZE", "WEEK_STAR_RETRY_BATCH_SIZE"}, 10)),
		WeekStarRetryBlock:         time.Duration(getEnvIntAny([]string{"CHATAPP_WEEK_STAR_RETRY_BLOCK_MILLIS", "WEEK_STAR_RETRY_BLOCK_MILLIS"}, 5000)) * time.Millisecond,
		WeekStarRetryMinIdle:       time.Duration(getEnvIntAny([]string{"CHATAPP_WEEK_STAR_RETRY_MIN_IDLE_MILLIS", "WEEK_STAR_RETRY_MIN_IDLE_MILLIS"}, 60000)) * time.Millisecond,
		Timeout:                    time.Duration(getEnvIntAny([]string{"CHATAPP_HTTP_TIMEOUT_SECONDS", "CRON_HTTP_TIMEOUT_SECONDS", "INVITE_HTTP_TIMEOUT_SECONDS"}, 8)) * time.Second,
		ShutdownTimeout:            time.Duration(getEnvIntAny([]string{"CHATAPP_HTTP_SHUTDOWN_TIMEOUT_SECONDS", "CRON_SHUTDOWN_TIMEOUT_SECONDS", "INVITE_HTTP_SHUTDOWN_TIMEOUT_SECONDS"}, 30)) * time.Second,
		AutoMigrate:                getEnvBoolAny([]string{"CHATAPP_AUTO_MIGRATE", "CRON_AUTO_MIGRATE", "INVITE_AUTO_MIGRATE"}, false),
	}
}

func getEnvAny(keys []string, fallback string) string {
	for _, key := range keys {
		key = strings.TrimSpace(key)
		if key == "" {
			continue
		}
		if value := strings.TrimSpace(os.Getenv(key)); value != "" {
			return value
		}
	}
	return fallback
}

func getEnvIntAny(keys []string, fallback int) int {
	value := getEnvAny(keys, "")
	if value == "" {
		return fallback
	}
	parsed, err := strconv.Atoi(value)
	if err != nil {
		return fallback
	}
	return parsed
}

func getEnvBoolAny(keys []string, fallback bool) bool {
	value := getEnvAny(keys, "")
	if value == "" {
		return fallback
	}
	parsed, err := strconv.ParseBool(value)
	if err != nil {
		return fallback
	}
	return parsed
}

func defaultConsumerName(prefix string) string {
	host, err := os.Hostname()
	if err != nil {
		return prefix
	}
	host = strings.TrimSpace(host)
	if host == "" {
		return prefix
	}
	return prefix + "-" + host
}