/
slots.go
70 lines (58 loc) · 1.2 KB
/
slots.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
// Copyright 2016 CodisLabs. All Rights Reserved.
// Licensed under the MIT (MIT-LICENSE.txt) license.
package proxy
import (
"sync"
"github.com/CodisLabs/codis/pkg/models"
)
type Slot struct {
id int
lock struct {
hold bool
sync.RWMutex
}
refs sync.WaitGroup
switched bool
backend, migrate struct {
id int
bc *sharedBackendConn
}
replicaGroups [][]*sharedBackendConn
method forwardMethod
}
func (s *Slot) snapshot() *models.Slot {
var m = &models.Slot{
Id: s.id,
Locked: s.lock.hold,
BackendAddr: s.backend.bc.Addr(),
BackendAddrGroupId: s.backend.id,
MigrateFrom: s.migrate.bc.Addr(),
MigrateFromGroupId: s.migrate.id,
ForwardMethod: s.method.GetId(),
}
for i := range s.replicaGroups {
var group []string
for _, bc := range s.replicaGroups[i] {
group = append(group, bc.Addr())
}
m.ReplicaGroups = append(m.ReplicaGroups, group)
}
return m
}
func (s *Slot) blockAndWait() {
if !s.lock.hold {
s.lock.hold = true
s.lock.Lock()
}
s.refs.Wait()
}
func (s *Slot) unblock() {
if !s.lock.hold {
return
}
s.lock.hold = false
s.lock.Unlock()
}
func (s *Slot) forward(r *Request, hkey []byte) error {
return s.method.Forward(s, r, hkey)
}