-
Notifications
You must be signed in to change notification settings - Fork 2
/
storage.go
100 lines (86 loc) · 2.17 KB
/
storage.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
package signalling
import (
"github.com/irdkwmnsb/webrtc-grabber/packages/relay/internal/api"
"github.com/irdkwmnsb/webrtc-grabber/packages/relay/internal/sockets"
"sync"
"time"
)
type Storage struct {
peers map[sockets.SocketID]api.Peer
participants []string
mutex sync.Mutex
}
func NewStorage() *Storage {
return &Storage{
peers: make(map[sockets.SocketID]api.Peer),
participants: make([]string, 0),
}
}
func (s *Storage) addPeer(name string, socketId sockets.SocketID) *Storage {
newPeer := api.Peer{Name: name, SocketId: socketId}
s.mutex.Lock()
defer s.mutex.Unlock()
s.peers[newPeer.SocketId] = newPeer
return s
}
func (s *Storage) getPeerByName(name string) (api.Peer, bool) {
s.mutex.Lock()
defer s.mutex.Unlock()
var peer api.Peer
isFind := false
for _, p := range s.peers {
if p.Name != name {
continue
}
if !isFind || peer.LastPing == nil || p.LastPing != nil && peer.LastPing.Before(*p.LastPing) {
peer = p
isFind = true
}
}
return peer, isFind
}
func (s *Storage) deletePeer(streamId sockets.SocketID) {
s.mutex.Lock()
defer s.mutex.Unlock()
delete(s.peers, streamId)
}
func (s *Storage) deleteOldPeers() {
for peerSocketId, peer := range s.peers {
if peer.LastPing != nil && time.Since(*peer.LastPing).Seconds() > 60 {
s.deletePeer(peerSocketId)
}
}
}
func (s *Storage) ping(socketId sockets.SocketID, status api.PeerStatus) {
now := time.Now()
s.mutex.Lock()
defer s.mutex.Unlock()
peer := s.peers[socketId]
peer.LastPing = &now
peer.ConnectionsCount = status.ConnectionsCount
peer.StreamTypes = status.StreamTypes
s.peers[socketId] = peer
}
func (s *Storage) getAll() []api.Peer {
s.mutex.Lock()
defer s.mutex.Unlock()
var peers []api.Peer
for _, peer := range s.peers {
peers = append(peers, peer)
}
return peers
}
func (s *Storage) getParticipantsStatus() []api.Peer {
var peers []api.Peer
for _, participant := range s.participants {
if peer, ok := s.getPeerByName(participant); ok {
peers = append(peers, peer)
} else {
peers = append(peers, api.Peer{Name: participant})
}
}
return peers
}
func (s *Storage) setParticipants(participants []string) {
s.participants = participants
}