修复前端验证问题,修复healthcheck问题

This commit is contained in:
2025-11-11 00:50:24 +08:00
parent 48b443c33b
commit 4eab8dea65
3 changed files with 32 additions and 4 deletions

View File

@@ -2,6 +2,7 @@ package db
import ( import (
"ai-gateway/internal/models" "ai-gateway/internal/models"
"log"
"os" "os"
"gorm.io/driver/sqlite" "gorm.io/driver/sqlite"
@@ -34,9 +35,36 @@ func InitDB() (*gorm.DB, error) {
return nil, err return nil, err
} }
// 初始化默认数据
if err := initializeDefaultData(db); err != nil {
log.Printf("Warning: Failed to initialize default data: %v", err)
}
return db, nil return db, nil
} }
// initializeDefaultData 初始化默认数据如默认API密钥
func initializeDefaultData(db *gorm.DB) error {
// 检查是否已存在API密钥
var count int64
if err := db.Model(&models.APIKey{}).Count(&count).Error; err != nil {
return err
}
// 如果没有API密钥创建默认密钥
if count == 0 {
defaultKey := &models.APIKey{
Key: "sk-dev-key-789012", // 与前端配置保持一致
}
if err := db.Create(defaultKey).Error; err != nil {
return err
}
log.Println("✓ Created default API key: sk-dev-key-789012")
}
return nil
}
// GetProviders 从数据库中获取所有服务商列表 // GetProviders 从数据库中获取所有服务商列表
func GetProviders(db *gorm.DB) ([]models.Provider, error) { func GetProviders(db *gorm.DB) ([]models.Provider, error) {
var providers []models.Provider var providers []models.Provider

View File

@@ -21,7 +21,7 @@ services:
networks: networks:
- ai-router-network - ai-router-network
healthcheck: healthcheck:
test: ["CMD", "wget", "--no-verbose", "--tries=1", "-O", "/dev/null", "http://localhost:8080/health"] test: ["CMD", "wget", "--no-verbose", "--tries=1", "-O", "/dev/null", "http://backend:8080/health"]
interval: 30s interval: 30s
timeout: 10s timeout: 10s
retries: 3 retries: 3
@@ -42,7 +42,7 @@ services:
networks: networks:
- ai-router-network - ai-router-network
healthcheck: healthcheck:
test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://localhost:80/health"] test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://frontend:80/health"]
interval: 30s interval: 30s
timeout: 10s timeout: 10s
retries: 3 retries: 3

View File

@@ -17,7 +17,7 @@ services:
networks: networks:
- ai-router-network - ai-router-network
healthcheck: healthcheck:
test: ["CMD", "wget", "--no-verbose", "--tries=1", "-O", "/dev/null", "http://localhost:8080/health"] test: ["CMD", "wget", "--no-verbose", "--tries=1", "-O", "/dev/null", "http://backend:8080/health"]
interval: 30s interval: 30s
timeout: 10s timeout: 10s
retries: 3 retries: 3
@@ -38,7 +38,7 @@ services:
networks: networks:
- ai-router-network - ai-router-network
healthcheck: healthcheck:
test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://localhost:80/health"] test: ["CMD", "wget", "--no-verbose", "--tries=1", "--spider", "http://frontend:80/health"]
interval: 30s interval: 30s
timeout: 10s timeout: 10s
retries: 3 retries: 3