mirror of
https://github.com/elyby/chrly.git
synced 2024-12-05 21:19:19 +05:30
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
|
package queue
|
||
|
|
||
|
import (
|
||
|
"sync"
|
||
|
|
||
|
"github.com/elyby/chrly/api/mojang"
|
||
|
)
|
||
|
|
||
|
type broadcastMap struct {
|
||
|
lock sync.Mutex
|
||
|
listeners map[string][]chan *mojang.SignedTexturesResponse
|
||
|
}
|
||
|
|
||
|
func newBroadcaster() *broadcastMap {
|
||
|
return &broadcastMap{
|
||
|
listeners: make(map[string][]chan *mojang.SignedTexturesResponse),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Returns a boolean value, which will be true if the username passed didn't exist before
|
||
|
func (c *broadcastMap) AddListener(username string, resultChan chan *mojang.SignedTexturesResponse) bool {
|
||
|
c.lock.Lock()
|
||
|
defer c.lock.Unlock()
|
||
|
|
||
|
val, alreadyHasSource := c.listeners[username]
|
||
|
if alreadyHasSource {
|
||
|
c.listeners[username] = append(val, resultChan)
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
c.listeners[username] = []chan *mojang.SignedTexturesResponse{resultChan}
|
||
|
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func (c *broadcastMap) BroadcastAndRemove(username string, result *mojang.SignedTexturesResponse) {
|
||
|
c.lock.Lock()
|
||
|
defer c.lock.Unlock()
|
||
|
|
||
|
val, ok := c.listeners[username]
|
||
|
if !ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
for _, channel := range val {
|
||
|
go func(channel chan *mojang.SignedTexturesResponse) {
|
||
|
channel <- result
|
||
|
close(channel)
|
||
|
}(channel)
|
||
|
}
|
||
|
|
||
|
delete(c.listeners, username)
|
||
|
}
|