mirror of https://github.com/mickael-menu/zk
Add the initial DB schema
parent
c89ff1e58a
commit
4dad3a2309
@ -0,0 +1,112 @@
|
|||||||
|
package sqlite
|
||||||
|
|
||||||
|
import (
|
||||||
|
"database/sql"
|
||||||
|
|
||||||
|
_ "github.com/mattn/go-sqlite3"
|
||||||
|
"github.com/mickael-menu/zk/util/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
// DB holds the connections to a SQLite database.
|
||||||
|
type DB struct {
|
||||||
|
db *sql.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open creates a new DB instance for the SQLite database at the given path.
|
||||||
|
func Open(path string) (*DB, error) {
|
||||||
|
db, err := sql.Open("sqlite3", "file:"+path)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "failed to open the database")
|
||||||
|
}
|
||||||
|
return &DB{db}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Close terminates the connections to the SQLite database.
|
||||||
|
func (db *DB) Close() error {
|
||||||
|
err := db.db.Close()
|
||||||
|
return errors.Wrap(err, "failed to close the database")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Migrate upgrades the SQL schema of the database.
|
||||||
|
func (db *DB) Migrate() error {
|
||||||
|
wrap := errors.Wrapper("database migration failed")
|
||||||
|
|
||||||
|
tx, err := db.db.Begin()
|
||||||
|
if err != nil {
|
||||||
|
return wrap(err)
|
||||||
|
}
|
||||||
|
defer tx.Rollback()
|
||||||
|
|
||||||
|
var version int
|
||||||
|
err = tx.QueryRow("PRAGMA user_version").Scan(&version)
|
||||||
|
if err != nil {
|
||||||
|
return wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if version == 0 {
|
||||||
|
err = execMultiple(tx, []string{
|
||||||
|
`
|
||||||
|
CREATE TABLE IF NOT EXISTS notes (
|
||||||
|
id INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
|
||||||
|
filename TEXT NOT NULL,
|
||||||
|
dir TEXT NOT NULL,
|
||||||
|
title TEXT DEFAULT('') NOT NULL,
|
||||||
|
content TEXT DEFAULT('') NOT NULL,
|
||||||
|
word_count INTEGER DEFAULT(0) NOT NULL,
|
||||||
|
checksum TEXT NOT NULL,
|
||||||
|
created TEXT DEFAULT(CURRENT_TIMESTAMP) NOT NULL,
|
||||||
|
modified TEXT DEFAULT(CURRENT_TIMESTAMP) NOT NULL,
|
||||||
|
UNIQUE(filename, dir)
|
||||||
|
)
|
||||||
|
`,
|
||||||
|
`CREATE INDEX IF NOT EXISTS notes_checksum_idx ON notes(checksum)`,
|
||||||
|
`
|
||||||
|
CREATE VIRTUAL TABLE IF NOT EXISTS notes_fts USING fts5(
|
||||||
|
title, content,
|
||||||
|
content = notes,
|
||||||
|
content_rowid = id,
|
||||||
|
tokenize = 'porter unicode61 remove_diacritics 1'
|
||||||
|
)
|
||||||
|
`,
|
||||||
|
// Triggers to keep the FTS index up to date.
|
||||||
|
`
|
||||||
|
CREATE TRIGGER IF NOT EXISTS notes_ai AFTER INSERT ON notes BEGIN
|
||||||
|
INSERT INTO notes_fts(rowid, title, content) VALUES (new.id, new.title, new.content);
|
||||||
|
END
|
||||||
|
`,
|
||||||
|
`
|
||||||
|
CREATE TRIGGER IF NOT EXISTS notes_ad AFTER DELETE ON notes BEGIN
|
||||||
|
INSERT INTO notes_fts(notes_fts, rowid, title, content) VALUES('delete', old.id, old.title, old.content);
|
||||||
|
END
|
||||||
|
`,
|
||||||
|
`
|
||||||
|
CREATE TRIGGER IF NOT EXISTS notes_au AFTER UPDATE ON notes BEGIN
|
||||||
|
INSERT INTO notes_fts(notes_fts, rowid, title, content) VALUES('delete', old.id, old.title, old.content);
|
||||||
|
INSERT INTO notes_fts(rowid, title, content) VALUES (new.id, new.title, new.content);
|
||||||
|
END
|
||||||
|
`,
|
||||||
|
`PRAGMA user_version = 1`,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = tx.Commit()
|
||||||
|
if err != nil {
|
||||||
|
return wrap(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func execMultiple(tx *sql.Tx, stmts []string) error {
|
||||||
|
var err error
|
||||||
|
for _, stmt := range stmts {
|
||||||
|
if err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
_, err = tx.Exec(stmt)
|
||||||
|
}
|
||||||
|
return err
|
||||||
|
}
|
Loading…
Reference in New Issue