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

websocket: fix a bug on emit on specific room when it doesn't exist it sends to all connections caused by a third-party contributor's PR...

Browse Source 
Former-commit-id: de6fddadd7cef8537ad1d1aff1acd991e4cf99cf
This commit is contained in:
Gerasimos Maropoulos 2018-05-02 17:47:14 +03:00
parent 087c8c8b3a
commit 0db2afea93
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
websocket/connection.go
View File

@ -511,6 +511,7 @@ func (c *connection) To(to string) Emitter {
} else if to == c.id {
return c.self
}
// is an emitter to another client/connection
return newEmitter(c, to)
}

24
websocket/server.go
View File

@ -375,18 +375,20 @@ func (s *Server) GetConnectionsByRoom(roomName string) []Connection {
// You SHOULD use connection.EmitMessage/Emit/To().Emit/EmitMessage instead.
// let's keep it unexported for the best.
func (s *Server) emitMessage(from, to string, data []byte) {
if to != All && to != Broadcast && s.rooms[to] != 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 := s.connections.get(connectionIDInsideRoom); c != nil {
c.writeDefault(data) //send the message to the client(s)
} else {
// the connection is not connected but it's inside the room, we remove it on disconnect but for ANY CASE:
cid := connectionIDInsideRoom
if c != nil {
cid = c.id
if to != All && to != Broadcast {
if s.rooms[to] != 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 := s.connections.get(connectionIDInsideRoom); c != nil {
c.writeDefault(data) //send the message to the client(s)
} else {
// the connection is not connected but it's inside the room, we remove it on disconnect but for ANY CASE:
cid := connectionIDInsideRoom
if c != nil {
cid = c.id
}
s.Leave(cid, to)
}
s.Leave(cid, to)
}
}
} else {