chore: sqlite 默认数据库文件调整为 db/openteam.db,运行时自动创建目录
This commit is contained in:
@@ -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 {
|
||||||
|
|||||||
Reference in New Issue
Block a user