eafonyang
87bdad34fc
- 修改`.env.example`文件，更新数据库和Redis的默认主机地址为`192.168.9.127` - 在`docker-compose.yml`中移除不必要的卷挂载配置 - 在`go.mod`中添加`github.com/joho/godotenv`依赖以支持环境变量加载 - 更新`README.md`以反映新的数据库和Redis连接信息 - 在`main.go`中加载环境变量以支持动态配置 - 新增缓存存在性检查功能，避免重复预热缓存 - 优化CSV读取逻辑，支持从S3获取数据
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
type RedisConfig struct {
|
|
Host string
|
|
Port int
|
|
Password string
|
|
Database int
|
|
}
|
|
|
|
func loadRedis(env string) RedisConfig {
|
|
switch env {
|
|
case "production":
|
|
if os.Getenv("REDIS_HOST") != "" {
|
|
return redisFromEnv("16279")
|
|
}
|
|
return RedisConfig{
|
|
Host: "172.31.38.162",
|
|
Port: 16279,
|
|
Password: getEnv("REDIS_PASSWORD", "eafon123!"),
|
|
Database: 1,
|
|
}
|
|
default:
|
|
port, _ := strconv.Atoi(getEnv("REDIS_PORT", "6379"))
|
|
db, _ := strconv.Atoi(getEnv("REDIS_DATABASE", "1"))
|
|
return RedisConfig{
|
|
Host: getEnv("REDIS_HOST", "192.168.9.127"),
|
|
Port: port,
|
|
Password: getEnv("REDIS_PASSWORD", ""),
|
|
Database: db,
|
|
}
|
|
}
|
|
}
|
|
|
|
func redisFromEnv(defaultPort string) RedisConfig {
|
|
port, _ := strconv.Atoi(getEnv("REDIS_PORT", defaultPort))
|
|
db, _ := strconv.Atoi(getEnv("REDIS_DATABASE", "1"))
|
|
|
|
return RedisConfig{
|
|
Host: os.Getenv("REDIS_HOST"),
|
|
Port: port,
|
|
Password: os.Getenv("REDIS_PASSWORD"),
|
|
Database: db,
|
|
}
|
|
}
|
|
|
|
func (r RedisConfig) validate() error {
|
|
if r.Host == "" {
|
|
return fmt.Errorf("redis host is required")
|
|
}
|
|
return nil
|
|
}
|