forked from hyperledger-archives/aries-framework-go
/
message.go
52 lines (41 loc) · 1.09 KB
/
message.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
/*
Copyright SecureKey Technologies Inc. All Rights Reserved.
SPDX-License-Identifier: Apache-2.0
*/
package service
import "sync"
// Message thread-safe message register structure.
type Message struct {
mu sync.RWMutex
events []chan<- StateMsg
}
// MsgEvents returns event message channels.
func (m *Message) MsgEvents() []chan<- StateMsg {
m.mu.RLock()
events := append(m.events[:0:0], m.events...)
m.mu.RUnlock()
return events
}
// RegisterMsgEvent on protocol messages. The message events are triggered for incoming messages. Event
// will not expect any callback on these events unlike Action events.
func (m *Message) RegisterMsgEvent(ch chan<- StateMsg) error {
if ch == nil {
return ErrNilChannel
}
m.mu.Lock()
m.events = append(m.events, ch)
m.mu.Unlock()
return nil
}
// UnregisterMsgEvent on protocol messages. Refer RegisterMsgEvent().
func (m *Message) UnregisterMsgEvent(ch chan<- StateMsg) error {
m.mu.Lock()
for i := 0; i < len(m.events); i++ {
if m.events[i] == ch {
m.events = append(m.events[:i], m.events[i+1:]...)
i--
}
}
m.mu.Unlock()
return nil
}