let5see/huobao-drama
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
Connor
2026-01-12 14:25:39 +08:00
parent 7db3545a01
commit 6f4f0b08f5
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
infrastructure/database/database.go
View File

@@ -2,6 +2,8 @@ package database
import ( import (
"fmt" "fmt"
"os"
"path/filepath"
"time" "time"
"github.com/drama-generator/backend/domain/models" "github.com/drama-generator/backend/domain/models"
@@ -15,6 +17,13 @@ import (
func NewDatabase(cfg config.DatabaseConfig) (*gorm.DB, error) { func NewDatabase(cfg config.DatabaseConfig) (*gorm.DB, error) {
dsn := cfg.DSN() dsn := cfg.DSN()
if cfg.Type == "sqlite" {
dbDir := filepath.Dir(dsn)
if err := os.MkdirAll(dbDir, 0755); err != nil {
return nil, fmt.Errorf("failed to create database directory: %w", err)
}
}
gormConfig := &gorm.Config{ gormConfig := &gorm.Config{
Logger: logger.Default.LogMode(logger.Info), Logger: logger.Default.LogMode(logger.Info),
} }