31 lines
597 B
Go
31 lines
597 B
Go
package db
|
|
|
|
import (
|
|
"ai-gateway/internal/models"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// InitDB 初始化数据库连接并执行自动迁移
|
|
func InitDB() (*gorm.DB, error) {
|
|
// 使用SQLite驱动连接到gateway.db数据库文件
|
|
db, err := gorm.Open(sqlite.Open("gateway.db"), &gorm.Config{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// 自动迁移所有模型，创建相应的表
|
|
err = db.AutoMigrate(
|
|
&models.Provider{},
|
|
&models.APIKey{},
|
|
&models.VirtualModel{},
|
|
&models.BackendModel{},
|
|
&models.RequestLog{},
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return db, nil
|
|
} |