Browse Source

db 新建的时候如果存储的目录不存在则需要创建

Signed-off-by: allan716 <[email protected]>
allan716 4 years ago
parent
commit
40c8e045a1
1 changed files with 6 additions and 0 deletions
  1. 6 0
      internal/dao/init.go

+ 6 - 0
internal/dao/init.go

@@ -8,6 +8,7 @@ import (
 	"github.com/allanpk716/ChineseSubFinder/internal/pkg/sqlite"
 	"gorm.io/gorm"
 	"os"
+	"path"
 	"runtime"
 )
 
@@ -16,6 +17,11 @@ func InitDb() error {
 	var err error
 	// 新建数据库
 	nowDbFileName := getDbName()
+
+	dbDir := path.Dir(nowDbFileName)
+	if pkg.IsDir(dbDir) == false {
+		_ = os.MkdirAll(dbDir, os.ModePerm)
+	}
 	db, err = gorm.Open(sqlite.Open(nowDbFileName), &gorm.Config{})
 	if err != nil {
 		return errors.New(fmt.Sprintf("failed to connect database, %s", err.Error()))