27 lines
498 B
Go
27 lines
498 B
Go
package models
|
|
|
|
import "github.com/coder/websocket"
|
|
|
|
type Subscribers map[*websocket.Conn]*bool
|
|
|
|
func InitSubsrcibers() Subscribers {
|
|
return make(Subscribers)
|
|
}
|
|
|
|
func (s *Subscribers) Connect(conn *websocket.Conn) {
|
|
b := true
|
|
(*s)[conn] = &b
|
|
}
|
|
|
|
func (s *Subscribers) DeleteSubsrcibers(conn *websocket.Conn) {
|
|
delete(*s, conn)
|
|
}
|
|
|
|
func (s *Subscribers) GetPointer(conn *websocket.Conn) *bool {
|
|
return (*s)[conn]
|
|
}
|
|
|
|
func (s *Subscribers) Disconnect(conn *websocket.Conn) {
|
|
*(*s)[conn] = false
|
|
}
|