go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-01-23 02:31:04 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix panic error in concurrent calling with websocket.Connection.Emit()

Browse Source 
Former-commit-id: b5d0ecd61547a297c320867af0a8abf208f0f743
This commit is contained in:
Eryx 2018-10-16 13:05:14 +08:00
parent a1d80ce4b6
commit d2c3b90bc4
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
websocket/server.go
View File

@ -52,9 +52,9 @@ type (
func New(cfg Config) *Server {
cfg = cfg.Validate()
return &Server{
config: cfg,
connections: sync.Map{}, // ready-to-use, this is not necessary.
rooms: make(map[string][]string),
config: cfg,
connections: sync.Map{}, // ready-to-use, this is not necessary.
rooms: make(map[string][]string),
onConnectionListeners: make([]ConnectionFunc, 0),
upgrader: websocket.Upgrader{
HandshakeTimeout: cfg.HandshakeTimeout,
@ -352,7 +352,10 @@ func (s *Server) GetConnectionsByRoom(roomName string) []Connection {
// let's keep it unexported for the best.
func (s *Server) emitMessage(from, to string, data []byte) {
if to != All && to != Broadcast {
if s.rooms[to] != nil {
s.mu.RLock()
room := s.rooms[to]
s.mu.RUnlock()
if room != nil {
// it suppose to send the message to a specific room/or a user inside its own room
for _, connectionIDInsideRoom := range s.rooms[to] {
if c, ok := s.getConnection(connectionIDInsideRoom); ok {