YourSelfHosted/memos
1
0
Fork 0
mirror of https://github.com/usememos/memos.git synced 2025-06-05 22:09:59 +02:00
Code Issues Packages Projects Releases Wiki Activity

update: db initialization

Browse Source
This commit is contained in:
boojack
2022-05-01 11:06:29 +08:00
parent eae858e20b
commit 406109b091
7 changed files with 32 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
store/sqlite.go
View File

@@ -7,6 +7,7 @@ import (
"fmt"
"io/fs"
"memos/common"
"os"
"sort"
_ "github.com/mattn/go-sqlite3"
@@ -42,18 +43,31 @@ func (db *DB) Open() (err error) {
}
// Connect to the database.
if db.Db, err = sql.Open("sqlite3", db.DSN); err != nil {
sqlDB, err := sql.Open("sqlite3", db.DSN)
if err != nil {
return fmt.Errorf("failed to open db with dsn: %s, err: %w", db.DSN, err)
}
if db.mode == "dev" {
// If mode is dev, then we will migrate and seed the database.
db.Db = sqlDB
// If db file not exists, we should migrate and seed the database.
if _, err := os.Stat(db.DSN); errors.Is(err, os.ErrNotExist) {
if err := db.migrate(); err != nil {
return fmt.Errorf("failed to migrate: %w", err)
}
if err := db.seed(); err != nil {
return fmt.Errorf("failed to seed: %w", err)
}
} else {
// If db file exists and mode is dev, we should migrate and seed the database.
if db.mode == "dev" {
if err := db.migrate(); err != nil {
return fmt.Errorf("failed to migrate: %w", err)
}
if err := db.seed(); err != nil {
return fmt.Errorf("failed to seed: %w", err)
}
}
}
return err