@@ -4,29 +4,56 @@ import (
|
||||
"fmt"
|
||||
|
||||
"github.com/glebarez/sqlite"
|
||||
"gorm.io/driver/mysql"
|
||||
"gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
|
||||
"oci-portal/internal/model"
|
||||
)
|
||||
|
||||
// Open 打开 SQLite 数据库并自动迁移全部模型。
|
||||
func Open(path string) (*gorm.DB, error) {
|
||||
db, err := gorm.Open(sqlite.Open(path), &gorm.Config{
|
||||
// Open 按驱动打开数据库并自动迁移全部模型。
|
||||
// driver 取值 sqlite(默认)/mysql/postgres;sqlite 用 path,其余用 dsn。
|
||||
func Open(driver, dsn, path string) (*gorm.DB, error) {
|
||||
dialector, err := buildDialector(driver, dsn, path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
db, err := gorm.Open(dialector, &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Warn),
|
||||
})
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("open sqlite %s: %w", path, err)
|
||||
return nil, fmt.Errorf("open %s database: %w", dialector.Name(), err)
|
||||
}
|
||||
if err := db.AutoMigrate(
|
||||
if err := autoMigrate(db); err != nil {
|
||||
return nil, fmt.Errorf("auto migrate: %w", err)
|
||||
}
|
||||
return db, nil
|
||||
}
|
||||
|
||||
// buildDialector 把驱动名映射为 GORM dialector;三个驱动均为纯 Go,不引入 cgo。
|
||||
// MySQL 设 DefaultStringSize=512:未标 size 的 string 建为 varchar(512),
|
||||
// 避免 OCID / URL 类字段被默认 varchar(191) 截断(SQLite/PG 的 string 天然无长度上限)。
|
||||
func buildDialector(driver, dsn, path string) (gorm.Dialector, error) {
|
||||
switch driver {
|
||||
case "", "sqlite":
|
||||
return sqlite.Open(path), nil
|
||||
case "mysql":
|
||||
return mysql.New(mysql.Config{DSN: dsn, DefaultStringSize: 512}), nil
|
||||
case "postgres":
|
||||
return postgres.Open(dsn), nil
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported DB_DRIVER %q (sqlite/mysql/postgres)", driver)
|
||||
}
|
||||
}
|
||||
|
||||
func autoMigrate(db *gorm.DB) error {
|
||||
return db.AutoMigrate(
|
||||
&model.User{}, &model.UserIdentity{}, &model.OciConfig{}, &model.Task{}, &model.TaskLog{},
|
||||
&model.CheckSnapshot{}, &model.CostSnapshot{},
|
||||
&model.RegionCache{}, &model.CompartmentCache{}, &model.Setting{},
|
||||
&model.SystemLog{}, &model.LogEvent{}, &model.Proxy{},
|
||||
&model.AiKey{}, &model.AiChannel{}, &model.AiModelCache{}, &model.AiCallLog{},
|
||||
&model.AiContentLog{},
|
||||
); err != nil {
|
||||
return nil, fmt.Errorf("auto migrate: %w", err)
|
||||
}
|
||||
return db, nil
|
||||
)
|
||||
}
|
||||
|
||||
@@ -0,0 +1,52 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestBuildDialector(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
driver string
|
||||
wantName string
|
||||
wantErr string
|
||||
}{
|
||||
{name: "默认空驱动走 sqlite", driver: "", wantName: "sqlite"},
|
||||
{name: "显式 sqlite", driver: "sqlite", wantName: "sqlite"},
|
||||
{name: "mysql", driver: "mysql", wantName: "mysql"},
|
||||
{name: "postgres", driver: "postgres", wantName: "postgres"},
|
||||
{name: "未知驱动报错", driver: "oracle", wantErr: "unsupported DB_DRIVER"},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
d, err := buildDialector(tt.driver, "user:pass@tcp(127.0.0.1)/db", "test.db")
|
||||
if tt.wantErr != "" {
|
||||
if err == nil || !strings.Contains(err.Error(), tt.wantErr) {
|
||||
t.Fatalf("err = %v, want contains %q", err, tt.wantErr)
|
||||
}
|
||||
return
|
||||
}
|
||||
if err != nil {
|
||||
t.Fatalf("buildDialector: %v", err)
|
||||
}
|
||||
if got := d.Name(); got != tt.wantName {
|
||||
t.Fatalf("dialector name = %q, want %q", got, tt.wantName)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestOpenSQLiteMigrates(t *testing.T) {
|
||||
path := filepath.Join(t.TempDir(), "test.db")
|
||||
db, err := Open("sqlite", "", path)
|
||||
if err != nil {
|
||||
t.Fatalf("Open sqlite: %v", err)
|
||||
}
|
||||
for _, table := range []string{"users", "oci_configs", "ai_channels", "log_events"} {
|
||||
if !db.Migrator().HasTable(table) {
|
||||
t.Fatalf("table %s not migrated", table)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user