53 lines
1.7 KiB
Go
53 lines
1.7 KiB
Go
package socket
|
|
|
|
import (
|
|
connection "g365sfu/socket/struct"
|
|
"sync"
|
|
)
|
|
|
|
// Это хранилище для всех подключённых сокетов.
|
|
// Оно обеспечивает потокобезопасный доступ к ним,
|
|
// позволяя добавлять, удалять и получать сокеты без риска
|
|
// гонок данных.
|
|
|
|
// Здесь содержатся только соединения прошедшие рукопожатие
|
|
|
|
// Потокобезопасное хранилище подключённых сокетов
|
|
var (
|
|
handshakeCompletedSockets = make(map[string]*connection.Connection)
|
|
socketsMutex = sync.RWMutex{}
|
|
)
|
|
|
|
// Добавление сокета в хранилище
|
|
func AddSocket(conn *connection.Connection) {
|
|
socketsMutex.Lock()
|
|
defer socketsMutex.Unlock()
|
|
handshakeCompletedSockets[conn.Identificator] = conn
|
|
}
|
|
|
|
// Удаление сокета из хранилища
|
|
func RemoveSocket(identificator string) {
|
|
socketsMutex.Lock()
|
|
defer socketsMutex.Unlock()
|
|
delete(handshakeCompletedSockets, identificator)
|
|
}
|
|
|
|
// Получение сокета по идентификатору
|
|
func GetSocket(identificator string) (*connection.Connection, bool) {
|
|
socketsMutex.RLock()
|
|
defer socketsMutex.RUnlock()
|
|
conn, exists := handshakeCompletedSockets[identificator]
|
|
return conn, exists
|
|
}
|
|
|
|
// Получение всех сокетов
|
|
func GetAllSockets() []*connection.Connection {
|
|
socketsMutex.RLock()
|
|
defer socketsMutex.RUnlock()
|
|
connections := make([]*connection.Connection, 0, len(handshakeCompletedSockets))
|
|
for _, conn := range handshakeCompletedSockets {
|
|
connections = append(connections, conn)
|
|
}
|
|
return connections
|
|
}
|