forked from nknorg/nkn
-
Notifications
You must be signed in to change notification settings - Fork 0
/
messagebuffer.go
39 lines (33 loc) · 1.02 KB
/
messagebuffer.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
package messagebuffer
import (
"encoding/hex"
"sync"
"github.com/nknorg/nkn/v2/pb"
)
// MessageBuffer is the buffer to hold message for clients not online
type MessageBuffer struct {
sync.Mutex
buffer map[string][]*pb.Relay
}
// NewMessageBuffer creates a MessageBuffer
func NewMessageBuffer() *MessageBuffer {
return &MessageBuffer{
buffer: make(map[string][]*pb.Relay),
}
}
// AddMessage adds a message to message buffer
func (messageBuffer *MessageBuffer) AddMessage(clientID []byte, msg *pb.Relay) {
clientIDStr := hex.EncodeToString(clientID)
messageBuffer.Lock()
defer messageBuffer.Unlock()
messageBuffer.buffer[clientIDStr] = append(messageBuffer.buffer[clientIDStr], msg)
}
// PopMessages reads and clears all messages of a client
func (messageBuffer *MessageBuffer) PopMessages(clientID []byte) []*pb.Relay {
clientIDStr := hex.EncodeToString(clientID)
messageBuffer.Lock()
defer messageBuffer.Unlock()
messages := messageBuffer.buffer[clientIDStr]
messageBuffer.buffer[clientIDStr] = nil
return messages
}