47 lines
843 B
Go
47 lines
843 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
connCenter ConnCenter
|
||
|
|
)
|
||
|
|
|
||
|
|
type ConnCenter struct {
|
||
|
|
Conns map[uint16]net.Conn
|
||
|
|
Mu sync.Mutex
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConnCenter) Init() {
|
||
|
|
c.Conns = make(map[uint16]net.Conn)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConnCenter) AddConn(id uint16, con net.Conn) (uint16, error) {
|
||
|
|
c.Mu.Lock()
|
||
|
|
defer c.Mu.Unlock()
|
||
|
|
log.Printf("Conns: add %v with %d id\n", con, id)
|
||
|
|
c.Conns[id] = con
|
||
|
|
return id, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConnCenter) DeleteIfHaveOne(id uint16) {
|
||
|
|
name, found := c.Conns[id]
|
||
|
|
if !found {
|
||
|
|
log.Printf("Conn with %d id is not found; Can not delete\n", id)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
delete(c.Conns, id)
|
||
|
|
log.Printf("Conn with %v con and %d id was found; Conn is deleted\n", name, id)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *ConnCenter) GetConn(id uint16) (net.Conn, error) {
|
||
|
|
con, have := c.Conns[id]
|
||
|
|
if !have {
|
||
|
|
return nil, ERROR_DONT_HAVE
|
||
|
|
}
|
||
|
|
return con, nil
|
||
|
|
}
|