package components
  
  import "sync"
  
  type ConnManage struct {
  	mu    sync.RWMutex
  	conns map[int]IConnection
  }
  
  func NewConnManage() *ConnManage {
  	return &ConnManage{
  		mu:    sync.RWMutex{},
  		conns: make(map[int]IConnection),
  	}
  }
  
  func (c *ConnManage) AddConn(id int, connection IConnection) {
  	c.mu.Lock()
  	defer c.mu.Unlock()
  	c.conns[id] = connection
  }
  
  func (c *ConnManage) GetConn(id int) IConnection {
  	c.mu.RLock()
  	defer c.mu.RUnlock()
  	return c.conns[id]
  }
  
  func (c *ConnManage) DelConn(id int) IConnection {
  	c.mu.Lock()
  	defer c.mu.Unlock()
  	conn := c.conns[id]
  	delete(c.conns, id)
  	return conn
  }
  
  func (c *ConnManage) Range(f func(key interface{}, value interface{}) bool) {
  	for k, v := range c.conns {
  		c.mu.Lock()
  		if ok := f(k, v); !ok {
  			c.mu.Unlock()
  			return
  		}
  		c.mu.Unlock()
  
  	}
  }
  
  func (c *ConnManage) StopAllConns() {
  	c.Range(func(key interface{}, value interface{}) bool {
  		conn := value.(IConnection)
  		conn.Stop()
  		return true
  	})
  }