/
clients.go
71 lines (59 loc) · 1.35 KB
/
clients.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
package container
import (
"sync"
"github.com/joeyscat/qim"
"go.uber.org/zap"
)
type ClientMap interface {
Add(client qim.Client)
Remove(id string)
Get(id string) (qim.Client, bool)
Services(kvs ...string) []qim.Service
}
type ClientsImpl struct {
clients *sync.Map
}
// Add implements ClientMap
func (ch *ClientsImpl) Add(client qim.Client) {
if client.ServiceID() == "" {
c.lg.Error("client id is required", zap.String("module", "ClientsImpl"))
}
ch.clients.Store(client.ServiceID(), client)
}
// Get implements ClientMap
func (ch *ClientsImpl) Get(id string) (qim.Client, bool) {
if id == "" {
c.lg.Error("client id is required", zap.String("module", "ClientsImpl"))
}
val, ok := ch.clients.Load(id)
if !ok {
return nil, false
}
return val.(qim.Client), true
}
// Remove implements ClientMap
func (ch *ClientsImpl) Remove(id string) {
ch.clients.Delete(id)
}
// Services implements ClientMap
func (ch *ClientsImpl) Services(kvs ...string) []qim.Service {
kvLen := len(kvs)
if kvLen != 0 && kvLen != 2 {
return nil
}
arr := make([]qim.Service, 0)
ch.clients.Range(func(key, value any) bool {
svc := value.(qim.Service)
if kvLen > 0 && svc.GetMeta()[kvs[0]] != kvs[1] {
return true
}
arr = append(arr, svc)
return true
})
return arr
}
func NewClients(num int) ClientMap {
return &ClientsImpl{
clients: new(sync.Map),
}
}