gosocketio/vendor/github.com/googollee/go-socket.io/namespaces.go

46 lines
777 B
Go

package socketio
import "sync"
type namespaces struct {
namespaces map[string]*namespaceConn
mu sync.RWMutex
}
func newNamespaces() *namespaces {
return &namespaces{
namespaces: make(map[string]*namespaceConn),
}
}
func (n *namespaces) Get(ns string) (*namespaceConn, bool) {
n.mu.RLock()
defer n.mu.RUnlock()
namespace, ok := n.namespaces[ns]
return namespace, ok
}
func (n *namespaces) Set(ns string, conn *namespaceConn) {
n.mu.Lock()
defer n.mu.Unlock()
n.namespaces[ns] = conn
}
func (n *namespaces) Delete(ns string) {
n.mu.Lock()
defer n.mu.Unlock()
delete(n.namespaces, ns)
}
func (n *namespaces) Range(fn func(ns string, nc *namespaceConn)) {
n.mu.RLock()
defer n.mu.RUnlock()
for ns, nc := range n.namespaces {
fn(ns, nc)
}
}