mirror of
https://github.com/woodchen-ink/aimodels-prices.git
synced 2025-07-18 05:32:00 +08:00
wood chen
315933d57a
- Update config to include SQLite path for database migration - Modify model type handlers to use new database schema - Update frontend to handle new model type key and label fields - Add database migration script to start process - Import SQLite driver for potential database usage
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package config
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"path/filepath"
|
||
|
||
"github.com/joho/godotenv"
|
||
)
|
||
|
||
type Config struct {
|
||
// MySQL配置
|
||
DBHost string
|
||
DBPort string
|
||
DBUser string
|
||
DBPassword string
|
||
DBName string
|
||
|
||
// 其他配置
|
||
ServerPort string
|
||
|
||
// SQLite配置（用于数据迁移）
|
||
SQLitePath string
|
||
}
|
||
|
||
func LoadConfig() (*Config, error) {
|
||
// 确保数据目录存在
|
||
dbDir := "./data"
|
||
if err := os.MkdirAll(dbDir, 0755); err != nil {
|
||
return nil, fmt.Errorf("failed to create data directory: %v", err)
|
||
}
|
||
|
||
// 尝试从 data 目录加载 .env 文件
|
||
envPath := filepath.Join(dbDir, ".env")
|
||
if err := godotenv.Load(envPath); err != nil {
|
||
fmt.Printf("Warning: .env file not found in data directory: %v\n", err)
|
||
// 如果 data/.env 不存在，尝试加载项目根目录的 .env
|
||
if err := godotenv.Load(); err != nil {
|
||
fmt.Printf("Warning: .env file not found in root directory: %v\n", err)
|
||
}
|
||
}
|
||
|
||
config := &Config{
|
||
// MySQL配置
|
||
DBHost: getEnv("DB_HOST", "localhost"),
|
||
DBPort: getEnv("DB_PORT", "3306"),
|
||
DBUser: getEnv("DB_USER", "root"),
|
||
DBPassword: getEnv("DB_PASSWORD", ""),
|
||
DBName: getEnv("DB_NAME", "aimodels"),
|
||
|
||
// 其他配置
|
||
ServerPort: getEnv("PORT", "8080"),
|
||
|
||
// SQLite路径（用于数据迁移）
|
||
SQLitePath: filepath.Join(dbDir, "aimodels.db"),
|
||
}
|
||
|
||
return config, nil
|
||
}
|
||
|
||
func getEnv(key, defaultValue string) string {
|
||
value := os.Getenv(key)
|
||
if value == "" {
|
||
return defaultValue
|
||
}
|
||
return value
|
||
}
|