You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.0 KiB
97 lines
2.0 KiB
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"sync"
|
|
|
|
sqlite "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var (
|
|
appDb *sql.DB
|
|
appDbWriteMutex = &sync.Mutex{}
|
|
dbStatementCache = map[string]*sql.Stmt{}
|
|
dbStatementCacheMutex = &sync.RWMutex{}
|
|
)
|
|
|
|
func initDatabase() (err error) {
|
|
sql.Register("goblog_db", &sqlite.SQLiteDriver{
|
|
ConnectHook: func(c *sqlite.SQLiteConn) error {
|
|
if err := c.RegisterFunc("tolocal", toLocalSafe, true); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
},
|
|
})
|
|
appDb, err = sql.Open("goblog_db", appConfig.Db.File+"?cache=shared&mode=rwc&_journal_mode=WAL")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return migrateDb()
|
|
}
|
|
|
|
func startWritingToDb() {
|
|
appDbWriteMutex.Lock()
|
|
}
|
|
|
|
func finishWritingToDb() {
|
|
appDbWriteMutex.Unlock()
|
|
}
|
|
|
|
func closeDb() error {
|
|
vacuumDb()
|
|
return appDb.Close()
|
|
}
|
|
|
|
func vacuumDb() {
|
|
_, _ = appDbExec("VACUUM;")
|
|
}
|
|
|
|
func prepareAppDbStatement(query string) (*sql.Stmt, error) {
|
|
stmt, err, _ := cacheGroup.Do(query, func() (interface{}, error) {
|
|
dbStatementCacheMutex.RLock()
|
|
stmt, ok := dbStatementCache[query]
|
|
dbStatementCacheMutex.RUnlock()
|
|
if ok && stmt != nil {
|
|
return stmt, nil
|
|
}
|
|
stmt, err := appDb.Prepare(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
dbStatementCacheMutex.Lock()
|
|
dbStatementCache[query] = stmt
|
|
dbStatementCacheMutex.Unlock()
|
|
return stmt, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return stmt.(*sql.Stmt), nil
|
|
}
|
|
|
|
func appDbExec(query string, args ...interface{}) (sql.Result, error) {
|
|
stmt, err := prepareAppDbStatement(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
startWritingToDb()
|
|
defer finishWritingToDb()
|
|
return stmt.Exec(args...)
|
|
}
|
|
|
|
func appDbQuery(query string, args ...interface{}) (*sql.Rows, error) {
|
|
stmt, err := prepareAppDbStatement(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return stmt.Query(args...)
|
|
}
|
|
|
|
func appDbQueryRow(query string, args ...interface{}) (*sql.Row, error) {
|
|
stmt, err := prepareAppDbStatement(query)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return stmt.QueryRow(args...), nil
|
|
}
|