添加redis,doris配置

config/config.go

@@ -11,9 +11,11 @@ import (
 
 // Config 应用配置
 type Config struct {
-	Database DBConfig `yaml:"database"`
-	Auth     Auth     `yaml:"auth"`
-	Service  Service  `yaml:"service"`
+	Database DBConfig    `yaml:"database"`
+	Redis    RedisConfig `yaml:"redis"`
+	Doris    DorisConfig `yaml:"doris"`
+	Auth     Auth        `yaml:"auth"`
+	Service  Service     `yaml:"service"`
 }
 
 // DBConfig 数据库配置
@@ -29,6 +31,44 @@ type DBConfig struct {
 	ConnMaxLifetime int    `yaml:"conn_max_lifetime"` // 单位：秒
 }
 
+// RedisConfig Redis配置
+type RedisConfig struct {
+	Host     string `yaml:"host"`
+	Port     int    `yaml:"port"`
+	Password string `yaml:"password"`
+	DB       int    `yaml:"db"`
+	PoolSize int    `yaml:"pool_size"`
+	// 连接超时和读写超时（秒）
+	DialTimeout  int `yaml:"dial_timeout"`
+	ReadTimeout  int `yaml:"read_timeout"`
+	WriteTimeout int `yaml:"write_timeout"`
+	// 连接池配置
+	IdleTimeout int `yaml:"idle_timeout"` // 空闲连接超时（秒）
+	MaxConnAge  int `yaml:"max_conn_age"` // 连接最大存活时间（秒）
+}
+
+// DorisConfig Doris配置
+type DorisConfig struct {
+	// FE节点配置
+	FEHost     string `yaml:"fe_host"`
+	FEPort     int    `yaml:"fe_port"`
+	FEUsername string `yaml:"fe_username"`
+	FEPassword string `yaml:"fe_password"`
+
+	// MySQL协议配置
+	MySQLHost string `yaml:"mysql_host"`
+	MySQLPort int    `yaml:"mysql_port"`
+
+	// 连接池配置
+	MaxOpenConns    int `yaml:"max_open_conns"`
+	MaxIdleConns    int `yaml:"max_idle_conns"`
+	ConnMaxLifetime int `yaml:"conn_max_lifetime"` // 单位：秒
+
+	// Stream Load配置
+	StreamLoadTimeout int `yaml:"stream_load_timeout"` // 秒
+	BatchSize         int `yaml:"batch_size"`          // 批量大小
+}
+
 // Auth 认证配置
 type Auth struct {
 	Token string `yaml:"token"`
@@ -36,6 +76,7 @@ type Auth struct {
 
 // Service 微服务配置
 type Service struct {
+	ServiceName    string `yaml:"service_name"`
 	Port           int    `yaml:"port"`
 	ReadTimeout    int    `yaml:"read_timeout"`
 	WriteTimeout   int    `yaml:"write_timeout"`
@@ -46,9 +87,13 @@ type Service struct {
 // 导出接口，防止外部修改
 type IConfig interface {
 	GetDatabase() DBConfig
+	GetRedis() RedisConfig
+	GetDoris() DorisConfig
 	GetAuth() Auth
 	GetService() Service
 	IsDatabaseConfigured() bool
+	IsRedisConfigured() bool
+	IsDorisConfigured() bool
 }
 
 // 实现接口的具体类型
@@ -60,6 +105,14 @@ func (cw *configWrapper) GetDatabase() DBConfig {
 	return cw.config.Database
 }
 
+func (cw *configWrapper) GetRedis() RedisConfig {
+	return cw.config.Redis
+}
+
+func (cw *configWrapper) GetDoris() DorisConfig {
+	return cw.config.Doris
+}
+
 func (cw *configWrapper) GetAuth() Auth {
 	return cw.config.Auth
 }
@@ -69,11 +122,23 @@ func (cw *configWrapper) GetService() Service {
 }
 
 func (cw *configWrapper) IsDatabaseConfigured() bool {
-	return cw.config.Database.Type != "" &&
-		cw.config.Database.Host != "" &&
-		cw.config.Database.Port > 0 &&
-		cw.config.Database.Username != "" &&
-		cw.config.Database.Database != ""
+	db := cw.config.Database
+	return db.Type != "" &&
+		db.Host != "" &&
+		db.Port > 0 &&
+		db.Username != "" &&
+		db.Database != ""
+}
+
+func (cw *configWrapper) IsRedisConfigured() bool {
+	redis := cw.config.Redis
+	return redis.Host != "" && redis.Port > 0
+}
+
+func (cw *configWrapper) IsDorisConfigured() bool {
+	doris := cw.config.Doris
+	// 检查FE配置（HTTP API方式）
+	return doris.FEHost != "" && doris.FEPort > 0
 }
 
 func (cw *configWrapper) IsAuthConfigured() bool {
@@ -119,12 +184,32 @@ func loadConfig() (*Config, error) {
 	// 先创建带默认值的配置
 	config := &Config{
 		Service: Service{
+			ServiceName:    "myService",
 			Port:           8080,
 			ReadTimeout:    30,
 			WriteTimeout:   30,
 			IdleTimeout:    60,
 			TrustedProxies: "",
 		},
+		Redis: RedisConfig{
+			Port:         6379,
+			DB:           0,
+			PoolSize:     10,
+			DialTimeout:  5,
+			ReadTimeout:  3,
+			WriteTimeout: 3,
+			IdleTimeout:  300,
+			MaxConnAge:   0, // 0表示不限制
+		},
+		Doris: DorisConfig{
+			FEPort:            8030,
+			MySQLPort:         9030,
+			MaxOpenConns:      20,
+			MaxIdleConns:      10,
+			ConnMaxLifetime:   3600,
+			StreamLoadTimeout: 30,
+			BatchSize:         1000,
+		},
 	}
 	// 读取配置文件
 	data, err := os.ReadFile(configFile)