修复前端验证问题,修复healthcheck问题
This commit is contained in:
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user