fix: improve database file handling (#14043)

This commit is contained in:
ssongliu
2026-10-13 09:39:27 +08:00
committed by GitHub
parent f9c43989fe
commit bc6cf519ce
6 changed files with 62 additions and 51 deletions
+25 -12
View File
@@ -13,20 +13,9 @@ import (
)
func LoadDBConnByPath(fullPath, dbName string) *gorm.DB {
if _, err := CreateDirWhenNotExist(true, path.Dir(fullPath)); err != nil {
panic(fmt.Errorf("init db dir failed, err: %v", err))
}
if _, err := os.Stat(fullPath); err != nil {
f, err := os.Create(fullPath)
if err != nil {
panic(fmt.Errorf("init %s db file failed, err: %v", dbName, err))
}
_ = f.Close()
}
db, err := GetDBWithPath(fullPath)
if err != nil {
panic(err)
panic(fmt.Errorf("init %s db failed, err: %w", dbName, err))
}
return db
}
@@ -40,6 +29,30 @@ func CloseDB(db *gorm.DB) {
}
func GetDBWithPath(dbPath string) (*gorm.DB, error) {
dbDir := path.Dir(dbPath)
if err := os.MkdirAll(dbDir, 0700); err != nil {
return nil, fmt.Errorf("init db dir failed, err: %w", err)
}
if err := os.Chmod(dbDir, 0700); err != nil {
return nil, fmt.Errorf("set db dir permissions failed, err: %w", err)
}
file, err := os.OpenFile(dbPath, os.O_CREATE|os.O_RDWR, 0600)
if err != nil {
return nil, fmt.Errorf("init db file failed, err: %w", err)
}
if err := file.Chmod(0600); err != nil {
_ = file.Close()
return nil, fmt.Errorf("set db file permissions failed, err: %w", err)
}
if err := file.Close(); err != nil {
return nil, err
}
for _, suffix := range []string{"-wal", "-shm", "-journal"} {
if err := os.Chmod(dbPath+suffix, 0600); err != nil && !os.IsNotExist(err) {
return nil, fmt.Errorf("set db sidecar permissions failed, err: %w", err)
}
}
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{
DisableForeignKeyConstraintWhenMigrating: true,
Logger: newLogger(),