DESKTOP-72TV0V4\caoxiaozhu
4a7199de93
- 优化 AI-Core 客户端调用 - 添加更多知识库配置选项 - 完善文档解析逻辑 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
121 lines
3.9 KiB
Go
121 lines
3.9 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"log"
|
||
|
||
"gorm.io/driver/mysql"
|
||
"gorm.io/gorm"
|
||
"gorm.io/gorm/logger"
|
||
|
||
"github.com/spf13/viper"
|
||
)
|
||
|
||
type Config struct {
|
||
Port string
|
||
JWTSecret string
|
||
DatabaseHost string
|
||
DatabasePort string
|
||
DatabaseUser string
|
||
DatabasePassword string
|
||
DatabaseName string
|
||
DatabaseURL string // 拼接后的完整连接字符串
|
||
PythonServiceURL string
|
||
AICoreServiceAddr string // AI-Core gRPC 服务地址，如 "localhost:50051"
|
||
// 文件上传配置
|
||
UploadMode string // "local" 或 "minio"
|
||
UploadLocalPath string // 本地存储路径，如 "resource/files"
|
||
ServerBaseURL string // 服务器基础 URL，用于生成本地文件 URL
|
||
MarkdownLocalPath string // Markdown 文件存储路径，如 "resource/markdown"
|
||
// MinIO 配置
|
||
MinIOEndpoint string
|
||
MinIOAccessKey string
|
||
MinIOSecretKey string
|
||
MinIOBucket string
|
||
MinIOUseSSL bool
|
||
}
|
||
|
||
func Load() *Config {
|
||
viper.SetConfigName("config")
|
||
viper.SetConfigType("yaml")
|
||
viper.AddConfigPath("./config")
|
||
viper.AddConfigPath("../config")
|
||
viper.AddConfigPath("../../config")
|
||
|
||
// 默认值
|
||
viper.SetDefault("port", "8080")
|
||
viper.SetDefault("jwt_secret", "your-secret-key-change-in-production")
|
||
viper.SetDefault("python_service_url", "http://localhost:8081")
|
||
viper.SetDefault("ai_core_service_addr", "localhost:50051")
|
||
// 数据库默认配置
|
||
viper.SetDefault("database_host", "localhost")
|
||
viper.SetDefault("database_port", "3306")
|
||
viper.SetDefault("database_user", "root")
|
||
viper.SetDefault("database_password", "root")
|
||
viper.SetDefault("database_name", "x_agents")
|
||
// 文件上传默认配置
|
||
viper.SetDefault("upload_mode", "local")
|
||
viper.SetDefault("upload_local_path", "resource/files")
|
||
viper.SetDefault("server_base_url", "http://localhost:8080")
|
||
viper.SetDefault("markdown_local_path", "resource/markdown")
|
||
viper.SetDefault("minio_endpoint", "localhost:9000")
|
||
viper.SetDefault("minio_access_key", "")
|
||
viper.SetDefault("minio_secret_key", "")
|
||
viper.SetDefault("minio_bucket", "x-agents")
|
||
viper.SetDefault("minio_use_ssl", false)
|
||
|
||
if err := viper.ReadInConfig(); err != nil {
|
||
log.Printf("Using default config: %v", err)
|
||
}
|
||
|
||
// 拼接数据库连接字符串
|
||
dbHost := viper.GetString("database_host")
|
||
dbPort := viper.GetString("database_port")
|
||
dbUser := viper.GetString("database_user")
|
||
dbPassword := viper.GetString("database_password")
|
||
dbName := viper.GetString("database_name")
|
||
databaseURL := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
|
||
dbUser, dbPassword, dbHost, dbPort, dbName)
|
||
|
||
return &Config{
|
||
Port: viper.GetString("port"),
|
||
JWTSecret: viper.GetString("jwt_secret"),
|
||
DatabaseURL: databaseURL,
|
||
DatabaseHost: dbHost,
|
||
DatabasePort: dbPort,
|
||
DatabaseUser: dbUser,
|
||
DatabasePassword: dbPassword,
|
||
DatabaseName: dbName,
|
||
PythonServiceURL: viper.GetString("python_service_url"),
|
||
AICoreServiceAddr: viper.GetString("ai_core_service_addr"),
|
||
// 文件上传配置
|
||
UploadMode: viper.GetString("upload_mode"),
|
||
UploadLocalPath: viper.GetString("upload_local_path"),
|
||
ServerBaseURL: viper.GetString("server_base_url"),
|
||
MarkdownLocalPath: viper.GetString("markdown_local_path"),
|
||
// MinIO 配置
|
||
MinIOEndpoint: viper.GetString("minio_endpoint"),
|
||
MinIOAccessKey: viper.GetString("minio_access_key"),
|
||
MinIOSecretKey: viper.GetString("minio_secret_key"),
|
||
MinIOBucket: viper.GetString("minio_bucket"),
|
||
MinIOUseSSL: viper.GetBool("minio_use_ssl"),
|
||
}
|
||
}
|
||
|
||
func InitDB(cfg *Config) (*gorm.DB, error) {
|
||
dsn := cfg.DatabaseURL
|
||
if dsn == "" {
|
||
return nil, fmt.Errorf("database URL is empty")
|
||
}
|
||
|
||
db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{
|
||
Logger: logger.Default.LogMode(logger.Info),
|
||
})
|
||
if err != nil {
|
||
return nil, fmt.Errorf("failed to connect database: %w", err)
|
||
}
|
||
|
||
log.Println("Database connected successfully")
|
||
return db, nil
|
||
}
|