memobus_relay_server/config/config.go

package config

import (
	"github.com/google/uuid"
	"github.com/spf13/viper"
	"log"
	"strings"
)

type Config struct {
	Server ServerConfig `mapstructure:"server"`
	Auth   AuthConfig   `mapstructure:"auth"`
	Redis  RedisConfig  `mapstructure:"redis"`
}
type ServerConfig struct {
	AppListenPort    string          `mapstructure:"app_listen_port"`
	DeviceListenPort string          `mapstructure:"device_listen_port"`
	PublicAppAddr    string          `mapstructure:"public_app_addr"`
	PublicDeviceAddr string          `mapstructure:"public_device_addr"`
	InstanceID       string          `mapstructure:"instance_id"`
	RateLimit        RateLimitConfig `mapstructure:"rate_limit"` // 新增
}
type AuthConfig struct {
	Enabled           bool   `mapstructure:"enabled"`
	AppAccessSecret   string `mapstructure:"app_access_secret"`
	DeviceRelaySecret string `mapstructure:"device_relay_secret"`
}

type RedisConfig struct {
	Enabled  bool   `mapstructure:"enabled"`
	Addr     string `mapstructure:"addr"`
	Password string `mapstructure:"password"`
	DB       int    `mapstructure:"db"`

	InstanceRegistryKey      string `mapstructure:"instance_registry_key"`
	DeviceRelayMappingKey    string `mapstructure:"device_relay_mapping_key"`
	HeartbeatIntervalSeconds int    `mapstructure:"heartbeat_interval_seconds"`
	InstanceTTLSeconds       int    `mapstructure:"instance_ttl_seconds"`
}

// 速率限制配置
type RateLimitConfig struct {
	Enabled      bool    `mapstructure:"enabled"`
	DownloadMBps float64 `mapstructure:"download_mbps"`
	UploadMBps   float64 `mapstructure:"upload_mbps"`
	BurstKB      int     `mapstructure:"burst_kb"`
}

var Cfg *Config

func LoadConfig() {
	viper.SetConfigName("config")
	viper.SetConfigType("yml")
	viper.AddConfigPath(".")
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_"))
	viper.AutomaticEnv()

	// 读取配置文件
	if err := viper.ReadInConfig(); err != nil {
		// 如果配置文件没找到，也没关系，可能完全通过环境变量配置
		if _, ok := err.(viper.ConfigFileNotFoundError); !ok {
			log.Fatalf("Fatal error reading config file: %v", err)
		}
	}

	// 将读取到的配置反序列化到 Cfg 结构体中
	if err := viper.Unmarshal(&Cfg); err != nil {
		log.Fatalf("Unable to decode config into struct: %v", err)
	}

	if Cfg.Server.InstanceID == "" {
		Cfg.Server.InstanceID = uuid.New().String()
	}
	log.Printf("Configuration loaded. Instance ID: %s", Cfg.Server.InstanceID)
}