blob: 3841d83b3ca89c60611fa9fee894a7aedde67f68 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
package svc // import "github.com/getwtxt/getwtxt/svc"
import (
"time"
"github.com/syndtr/goleveldb/leveldb"
"golang.org/x/sys/unix"
)
type dbase interface {
push() error
pull()
}
// Pull DB data into cache, if available.
func initDatabase() {
var db dbase
confObj.Mu.RLock()
dbpath := confObj.DBPath
confObj.Mu.RUnlock()
switch confObj.DBType {
case "leveldb":
lvl, err := leveldb.OpenFile(dbpath, nil)
errFatal("", err)
db = &dbLevel{db: lvl}
case "sqlite":
db = initSqlite()
}
dbChan <- db
pullDB()
}
func dbTimer() bool {
confObj.Mu.RLock()
answer := time.Since(confObj.LastPush) > confObj.DBInterval
confObj.Mu.RUnlock()
return answer
}
// Pushes the registry's cache data to a local
// database for safe keeping.
func pushDB() error {
db := <-dbChan
err := db.push()
dbChan <- db
unix.Sync()
return err
}
func pullDB() {
db := <-dbChan
db.pull()
dbChan <- db
}
|