123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179 |
- package data
- import (
- "bytes"
- "database/sql"
- "encoding/gob"
- "sync"
- "time"
- // sqlite requires a blank import.
- _ "github.com/mattn/go-sqlite3"
- "github.com/owncast/owncast/config"
- "github.com/owncast/owncast/db"
- log "github.com/sirupsen/logrus"
- )
- // Datastore is the global key/value store for configuration values.
- type Datastore struct {
- DB *sql.DB
- cache map[string][]byte
- DbLock *sync.Mutex
- }
- func (ds *Datastore) warmCache() {
- log.Traceln("Warming config value cache")
- res, err := ds.DB.Query("SELECT key, value FROM datastore")
- if err != nil || res.Err() != nil {
- log.Errorln("error warming config cache", err, res.Err())
- }
- defer res.Close()
- for res.Next() {
- var rowKey string
- var rowValue []byte
- if err := res.Scan(&rowKey, &rowValue); err != nil {
- log.Errorln("error pre-caching config row", err)
- }
- ds.cache[rowKey] = rowValue
- }
- }
- // GetQueries will return the shared instance of the SQL query generator.
- func (ds *Datastore) GetQueries() *db.Queries {
- return db.New(ds.DB)
- }
- // Get will query the database for the key and return the entry.
- func (ds *Datastore) Get(key string) (ConfigEntry, error) {
- cachedValue, err := ds.GetCachedValue(key)
- if err == nil {
- return ConfigEntry{
- Key: key,
- Value: cachedValue,
- }, nil
- }
- var resultKey string
- var resultValue []byte
- row := ds.DB.QueryRow("SELECT key, value FROM datastore WHERE key = ? LIMIT 1", key)
- if err := row.Scan(&resultKey, &resultValue); err != nil {
- return ConfigEntry{}, err
- }
- result := ConfigEntry{
- Key: resultKey,
- Value: resultValue,
- }
- ds.SetCachedValue(resultKey, resultValue)
- return result, nil
- }
- // Save will save the ConfigEntry to the database.
- func (ds *Datastore) Save(e ConfigEntry) error {
- ds.DbLock.Lock()
- defer ds.DbLock.Unlock()
- var dataGob bytes.Buffer
- enc := gob.NewEncoder(&dataGob)
- if err := enc.Encode(e.Value); err != nil {
- return err
- }
- tx, err := ds.DB.Begin()
- if err != nil {
- return err
- }
- var stmt *sql.Stmt
- stmt, err = tx.Prepare("INSERT INTO datastore (key, value) VALUES(?, ?) ON CONFLICT(key) DO UPDATE SET value=excluded.value")
- if err != nil {
- return err
- }
- _, err = stmt.Exec(e.Key, dataGob.Bytes())
- if err != nil {
- return err
- }
- defer stmt.Close()
- if err = tx.Commit(); err != nil {
- log.Fatalln(err)
- }
- ds.SetCachedValue(e.Key, dataGob.Bytes())
- return nil
- }
- // Setup will create the datastore table and perform initial initialization.
- func (ds *Datastore) Setup() {
- ds.cache = make(map[string][]byte)
- ds.DB = GetDatabase()
- ds.DbLock = &sync.Mutex{}
- createTableSQL := `CREATE TABLE IF NOT EXISTS datastore (
- "key" string NOT NULL PRIMARY KEY,
- "value" BLOB,
- "timestamp" DATE DEFAULT CURRENT_TIMESTAMP NOT NULL
- );`
- ds.MustExec(createTableSQL)
- if !HasPopulatedDefaults() {
- PopulateDefaults()
- }
- if !hasPopulatedFederationDefaults() {
- if err := SetFederationGoLiveMessage(config.GetDefaults().FederationGoLiveMessage); err != nil {
- log.Errorln(err)
- }
- if err := _datastore.SetBool("HAS_POPULATED_FEDERATION_DEFAULTS", true); err != nil {
- log.Errorln(err)
- }
- }
- // Set the server initialization date if needed.
- if hasSetInitDate, _ := GetServerInitTime(); hasSetInitDate == nil || !hasSetInitDate.Valid {
- _ = SetServerInitTime(time.Now())
- }
- migrateDatastoreValues(_datastore)
- }
- // Reset will delete all config entries in the datastore and start over.
- func (ds *Datastore) Reset() {
- sql := "DELETE FROM datastore"
- stmt, err := ds.DB.Prepare(sql)
- if err != nil {
- log.Fatalln(err)
- }
- defer stmt.Close()
- if _, err = stmt.Exec(); err != nil {
- log.Fatalln(err)
- }
- PopulateDefaults()
- }
- // GetDatastore returns the shared instance of the owncast datastore.
- func GetDatastore() *Datastore {
- return _datastore
- }
- // MustExec will execute a SQL statement on a provided database instance.
- func (ds *Datastore) MustExec(s string) {
- stmt, err := ds.DB.Prepare(s)
- if err != nil {
- log.Panic(err)
- }
- defer stmt.Close()
- _, err = stmt.Exec()
- if err != nil {
- log.Warnln(err)
- }
- }
|