Cleaned up a bit

This commit is contained in:
Atridad Lahiji 2024-06-04 16:07:13 -06:00
parent e668d62b57
commit e57121715e
No known key found for this signature in database
2 changed files with 9 additions and 9 deletions

View file

@ -31,7 +31,7 @@ func SSEDemoSend(c echo.Context) error {
}
// Send message
lib.SendSSE(channel, message)
lib.SSEServer.SendSSE(channel, message)
return c.JSON(http.StatusOK, map[string]string{"status": "message sent"})
}

View file

@ -52,15 +52,15 @@ func (s *SSEServerType) ClientCount(channel string) int {
return len(s.clients[channel])
}
func SendSSE(channel string, message string) error {
SSEServer.mu.Lock()
defer SSEServer.mu.Unlock()
func (s *SSEServerType) SendSSE(channel string, message string) {
s.mu.Lock()
defer s.mu.Unlock()
for client := range SSEServer.clients[channel] {
client <- message
}
go func() {
for client := range s.clients[channel] {
client <- message
}
}()
LogDebug.Printf("\nMessage broadcast on channel %s: %s\n", channel, message)
return nil
}