-
-
Notifications
You must be signed in to change notification settings - Fork 70
/
persist.go
88 lines (79 loc) · 1.76 KB
/
persist.go
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package lolicon_pool
import (
"encoding/json"
localdb "github.com/Sora233/DDBOT/lsp/buntdb"
"github.com/tidwall/buntdb"
)
func (pool *LoliconPool) load() {
pool.cond.L.Lock()
defer pool.cond.L.Unlock()
logger.Debug("image pool load cache start")
for k, v := range pool.cache {
var img []*Setu
err := localdb.RTxCover(func(tx *buntdb.Tx) error {
key := localdb.LoliconPoolStoreKey(k.String())
val, err := tx.Get(key)
if err == buntdb.ErrNotFound {
return nil
} else if err != nil {
return err
}
return json.Unmarshal([]byte(val), &img)
})
if err != nil {
logger.WithField("r18", k.String()).
Errorf("image pool cache load failed %v", err)
continue
}
for _, i := range img {
v.PushBack(i)
}
logger.WithField("r18", k.String()).
WithField("image_count", v.Len()).
Debug("image cache loaded")
}
}
func (pool *LoliconPool) store() {
pool.cond.L.Lock()
defer pool.cond.L.Unlock()
if !pool.changed {
return
}
defer func() {
pool.changed = false
}()
logger.Debug("image pool store cache start")
for k, v := range pool.cache {
var img []*Setu
root := v.Front()
if root == nil {
continue
}
for {
img = append(img, root.Value.(*Setu))
if root == v.Back() {
break
}
root = root.Next()
if root == nil {
break
}
}
err := localdb.RWTxCover(func(tx *buntdb.Tx) error {
key := localdb.LoliconPoolStoreKey(k.String())
b, err := json.Marshal(img)
if err != nil {
return err
}
_, _, err = tx.Set(key, string(b), nil)
return err
})
if err != nil {
logger.WithField("r18", k.String()).Errorf("image pool cache store failed %v", err)
continue
}
logger.WithField("r18", k.String()).
WithField("image_count", v.Len()).
Debug("image pool cache stored")
}
}