2020-07-28 19:17:07 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"database/sql"
|
2021-05-29 11:32:00 +00:00
|
|
|
"errors"
|
2021-03-26 08:33:46 +00:00
|
|
|
"log"
|
|
|
|
"os"
|
2020-10-18 08:01:46 +00:00
|
|
|
|
2020-12-16 20:24:53 +00:00
|
|
|
sqlite "github.com/mattn/go-sqlite3"
|
2021-03-26 08:33:46 +00:00
|
|
|
"github.com/schollz/sqlite3dump"
|
2020-07-28 19:17:07 +00:00
|
|
|
)
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
var appDb *goblogDb
|
|
|
|
|
|
|
|
type goblogDb struct {
|
|
|
|
db *sql.DB
|
|
|
|
statementCache map[string]*sql.Stmt
|
|
|
|
}
|
2020-07-28 19:17:07 +00:00
|
|
|
|
|
|
|
func initDatabase() (err error) {
|
2021-05-29 11:32:00 +00:00
|
|
|
// Setup db
|
2020-12-16 20:24:53 +00:00
|
|
|
sql.Register("goblog_db", &sqlite.SQLiteDriver{
|
|
|
|
ConnectHook: func(c *sqlite.SQLiteConn) error {
|
|
|
|
if err := c.RegisterFunc("tolocal", toLocalSafe, true); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-04-28 18:03:20 +00:00
|
|
|
if err := c.RegisterFunc("wordcount", wordCount, true); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-05-09 07:08:31 +00:00
|
|
|
if err := c.RegisterFunc("mdtext", renderText, true); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2020-12-16 20:24:53 +00:00
|
|
|
return nil
|
|
|
|
},
|
|
|
|
})
|
2021-05-29 11:32:00 +00:00
|
|
|
db, err := sql.Open("goblog_db", appConfig.Db.File+"?cache=shared&mode=rwc&_journal_mode=WAL")
|
2020-07-28 19:17:07 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
db.SetMaxOpenConns(1)
|
|
|
|
err = db.Ping()
|
2021-03-26 08:33:46 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
// Check available SQLite features
|
|
|
|
rows, err := db.Query("pragma compile_options")
|
2021-03-26 08:33:46 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
cos := map[string]bool{}
|
|
|
|
var co string
|
|
|
|
for rows.Next() {
|
|
|
|
err = rows.Scan(&co)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
cos[co] = true
|
|
|
|
}
|
|
|
|
if _, ok := cos["ENABLE_FTS5"]; !ok {
|
|
|
|
return errors.New("sqlite not compiled with FTS5")
|
|
|
|
}
|
|
|
|
// Migrate DB
|
|
|
|
err = migrateDb(db)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
// Create appDB
|
|
|
|
appDb = &goblogDb{
|
|
|
|
db: db,
|
|
|
|
statementCache: map[string]*sql.Stmt{},
|
|
|
|
}
|
|
|
|
appDb.vacuum()
|
|
|
|
addShutdownFunc(func() {
|
|
|
|
_ = appDb.close()
|
|
|
|
log.Println("Closed database")
|
|
|
|
})
|
2021-03-26 08:33:46 +00:00
|
|
|
if appConfig.Db.DumpFile != "" {
|
2021-05-29 11:32:00 +00:00
|
|
|
hourlyHooks = append(hourlyHooks, func() {
|
|
|
|
appDb.dump()
|
|
|
|
})
|
|
|
|
appDb.dump()
|
2021-03-26 08:33:46 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) dump() {
|
2021-03-26 08:33:46 +00:00
|
|
|
f, err := os.Create(appConfig.Db.DumpFile)
|
|
|
|
if err != nil {
|
|
|
|
log.Println("Error while dump db:", err.Error())
|
2021-04-16 18:00:38 +00:00
|
|
|
return
|
2021-03-26 08:33:46 +00:00
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
if err = sqlite3dump.DumpDB(db.db, f); err != nil {
|
2021-03-26 08:33:46 +00:00
|
|
|
log.Println("Error while dump db:", err.Error())
|
|
|
|
}
|
2020-07-28 19:17:07 +00:00
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) close() error {
|
|
|
|
db.vacuum()
|
|
|
|
return db.db.Close()
|
2020-07-28 19:17:07 +00:00
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) vacuum() {
|
|
|
|
_, _ = db.exec("VACUUM")
|
2020-07-28 19:17:07 +00:00
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) prepare(query string) (*sql.Stmt, error) {
|
2020-11-09 15:40:12 +00:00
|
|
|
stmt, err, _ := cacheGroup.Do(query, func() (interface{}, error) {
|
2021-05-29 11:32:00 +00:00
|
|
|
stmt, ok := db.statementCache[query]
|
2020-11-09 15:40:12 +00:00
|
|
|
if ok && stmt != nil {
|
|
|
|
return stmt, nil
|
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
stmt, err := db.db.Prepare(query)
|
2020-11-09 15:40:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
2021-05-29 11:32:00 +00:00
|
|
|
db.statementCache[query] = stmt
|
2020-11-09 15:40:12 +00:00
|
|
|
return stmt, nil
|
|
|
|
})
|
2020-12-09 16:25:09 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return stmt.(*sql.Stmt), nil
|
2020-11-09 15:40:12 +00:00
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) exec(query string, args ...interface{}) (sql.Result, error) {
|
|
|
|
stmt, err := db.prepare(query)
|
2020-11-09 15:40:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return stmt.Exec(args...)
|
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) execMulti(query string, args ...interface{}) (sql.Result, error) {
|
|
|
|
// Can't prepare the statement
|
|
|
|
return db.db.Exec(query, args...)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (db *goblogDb) query(query string, args ...interface{}) (*sql.Rows, error) {
|
|
|
|
stmt, err := db.prepare(query)
|
2020-11-09 15:40:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return stmt.Query(args...)
|
|
|
|
}
|
|
|
|
|
2021-05-29 11:32:00 +00:00
|
|
|
func (db *goblogDb) queryRow(query string, args ...interface{}) (*sql.Row, error) {
|
|
|
|
stmt, err := db.prepare(query)
|
2020-11-09 15:40:12 +00:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return stmt.QueryRow(args...), nil
|
2020-07-30 14:47:46 +00:00
|
|
|
}
|