94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package config
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
// DatabaseConfig 数据库配置结构
|
|
type DatabaseConfig struct {
|
|
Host string
|
|
Port int
|
|
User string
|
|
Password string
|
|
DBName string
|
|
SSLMode string
|
|
}
|
|
|
|
// AIConfig AI服务配置结构
|
|
type AIConfig struct {
|
|
BaseURL string
|
|
APIKey string
|
|
Model string
|
|
BaiduAppID string // 百度云AppBuilder应用ID
|
|
}
|
|
|
|
// GetDatabaseConfig 获取数据库配置
|
|
// 优先使用环境变量，如果没有设置则使用默认值
|
|
func GetDatabaseConfig() *DatabaseConfig {
|
|
// 从环境变量获取配置，如果未设置则使用默认值
|
|
host := getEnv("DB_HOST", "pgsql.yuchat.top")
|
|
port := getEnvAsInt("DB_PORT", 5432)
|
|
user := getEnv("DB_USER", "postgres")
|
|
password := getEnv("DB_PASSWORD", "longqi@1314")
|
|
dbname := getEnv("DB_NAME", "ankao")
|
|
sslmode := getEnv("DB_SSLMODE", "disable")
|
|
|
|
return &DatabaseConfig{
|
|
Host: host,
|
|
Port: port,
|
|
User: user,
|
|
Password: password,
|
|
DBName: dbname,
|
|
SSLMode: sslmode,
|
|
}
|
|
}
|
|
|
|
// getEnv 获取环境变量，如果不存在则返回默认值
|
|
func getEnv(key, defaultValue string) string {
|
|
if value := os.Getenv(key); value != "" {
|
|
return value
|
|
}
|
|
return defaultValue
|
|
}
|
|
|
|
// getEnvAsInt 获取整型环境变量，如果不存在或转换失败则返回默认值
|
|
func getEnvAsInt(key string, defaultValue int) int {
|
|
if value := os.Getenv(key); value != "" {
|
|
if intValue, err := strconv.Atoi(value); err == nil {
|
|
return intValue
|
|
}
|
|
}
|
|
return defaultValue
|
|
}
|
|
|
|
// GetDSN 返回数据库连接字符串
|
|
func (c *DatabaseConfig) GetDSN() string {
|
|
return fmt.Sprintf(
|
|
"host=%s user=%s password=%s dbname=%s port=%d sslmode=%s",
|
|
c.Host,
|
|
c.User,
|
|
c.Password,
|
|
c.DBName,
|
|
c.Port,
|
|
c.SSLMode,
|
|
)
|
|
}
|
|
|
|
// GetAIConfig 获取AI服务配置
|
|
// 优先使用环境变量，如果没有设置则使用默认值
|
|
func GetAIConfig() *AIConfig {
|
|
baseURL := getEnv("AI_BASE_URL", "http://172.20.0.117")
|
|
apiKey := getEnv("AI_API_KEY", "bce-v3/ALTAK-TgZ1YSBmbwNXo3BIuzNZ2/768b777896453e820a2c46f38614c8e9bf43f845")
|
|
model := getEnv("AI_MODEL", "deepseek-v3")
|
|
baiduAppID := getEnv("BAIDU_APP_ID", "7b336aaf-f448-46d6-9e5f-bb9e38a1167c")
|
|
|
|
return &AIConfig{
|
|
BaseURL: baseURL,
|
|
APIKey: apiKey,
|
|
Model: model,
|
|
BaiduAppID: baiduAppID,
|
|
}
|
|
}
|