chore: sqlite 默认数据库文件调整为 db/openteam.db,运行时自动创建目录

This commit is contained in:
Sakurasan
2026-09-01 21:36:15 +08:00
parent a2cef00908
commit d28fca8ee6
+19 -5
View File
@@ -3,6 +3,8 @@ package store
import ( import (
"fmt" "fmt"
"log" "log"
"os"
"path/filepath"
"opencatd-open/pkg/config" "opencatd-open/pkg/config"
_ "github.com/lib/pq" _ "github.com/lib/pq"
@@ -15,11 +17,17 @@ import (
var DB *gorm.DB var DB *gorm.DB
func InitDB(cfg *config.Config) (*gorm.DB, error) { func InitDB(cfg *config.Config) (*gorm.DB, error) {
var dialector gorm.Dialector var (
dialector gorm.Dialector
err error
)
switch cfg.DB_Type { switch cfg.DB_Type {
case "sqlite": case "sqlite":
dialector = sqliteDialector(cfg.DSN) dialector, err = sqliteDialector(cfg.DSN)
if err != nil {
return nil, err
}
case "postgres": case "postgres":
dialector = postgresDialector(cfg.DSN) dialector = postgresDialector(cfg.DSN)
case "mysql": case "mysql":
@@ -48,11 +56,17 @@ func InitDB(cfg *config.Config) (*gorm.DB, error) {
return db, nil return db, nil
} }
func sqliteDialector(dsn string) gorm.Dialector { func sqliteDialector(dsn string) (gorm.Dialector, error) {
if dsn == "" { if dsn == "" {
dsn = "opencatd.db" dsn = "db/openteam.db"
} }
return gormlite.Open(dsn) // sqlite 不会自动创建上级目录,先确保它存在(与 docker-compose 挂载的 /app/db 对应)
if dir := filepath.Dir(dsn); dir != "." && dir != string(filepath.Separator) {
if err := os.MkdirAll(dir, 0o755); err != nil {
return nil, fmt.Errorf("failed to create database directory %s: %w", dir, err)
}
}
return gormlite.Open(dsn), nil
} }
func postgresDialector(dsn string) gorm.Dialector { func postgresDialector(dsn string) gorm.Dialector {