/
rdbcache.go
102 lines (92 loc) · 2.52 KB
/
rdbcache.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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
package logdb
import (
"sync"
"github.com/LiuzhouChan/go-paxos/paxosio"
"github.com/LiuzhouChan/go-paxos/paxospb"
)
type rdbcache struct {
nodeInfo map[paxosio.NodeInfo]struct{}
ps map[paxosio.NodeInfo]paxospb.State
maxInstance map[paxosio.NodeInfo]uint64
mu sync.Mutex
// lastEntryBatch map[paxosio.NodeInfo]paxospb.EntryBatch
}
func newRDBCache() *rdbcache {
return &rdbcache{
nodeInfo: make(map[paxosio.NodeInfo]struct{}),
ps: make(map[paxosio.NodeInfo]paxospb.State),
maxInstance: make(map[paxosio.NodeInfo]uint64),
// lastEntryBatch: make(map[paxosio.NodeInfo]paxospb.EntryBatch),
}
}
func (r *rdbcache) setNodeInfo(groupID uint64, nodeID uint64) bool {
key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
r.mu.Lock()
defer r.mu.Unlock()
_, ok := r.nodeInfo[key]
if !ok {
r.nodeInfo[key] = struct{}{}
}
return !ok
}
func (r *rdbcache) setState(groupID uint64,
nodeID uint64, st paxospb.State) bool {
key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
r.mu.Lock()
defer r.mu.Unlock()
v, ok := r.ps[key]
if !ok {
r.ps[key] = st
return true
}
if paxospb.IsStateEqual(v, st) {
return false
}
r.ps[key] = st
return true
}
func (r *rdbcache) setMaxInstance(groupID uint64,
nodeID uint64, maxInstance uint64) {
r.mu.Lock()
defer r.mu.Unlock()
key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
r.maxInstance[key] = maxInstance
}
func (r *rdbcache) getMaxInstance(groupID uint64,
nodeID uint64) (uint64, bool) {
r.mu.Lock()
defer r.mu.Unlock()
key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
v, ok := r.maxInstance[key]
if !ok {
return 0, false
}
return v, true
}
// func (r *rdbcache) setLastEntryBatch(groupID uint64,
// nodeID uint64, eb paxospb.EntryBatch) {
// r.mu.Lock()
// defer r.mu.Unlock()
// key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
// oeb, ok := r.lastEntryBatch[key]
// if !ok {
// oeb = paxospb.EntryBatch{Entries: make([]paxospb.Entry, 0, len(eb.Entries))}
// } else {
// oeb.Entries = oeb.Entries[:0]
// }
// oeb.Entries = append(oeb.Entries, eb.Entries...)
// r.lastEntryBatch[key] = oeb
// }
// func (r *rdbcache) getLastEntryBatch(groupID uint64,
// nodeID uint64, lb paxospb.EntryBatch) (paxospb.EntryBatch, bool) {
// r.mu.Lock()
// defer r.mu.Unlock()
// key := paxosio.NodeInfo{GroupID: groupID, NodeID: nodeID}
// v, ok := r.lastEntryBatch[key]
// if !ok {
// return v, false
// }
// lb.Entries = lb.Entries[:0]
// lb.Entries = append(lb.Entries, v.Entries...)
// return lb, true
// }