package netService

import (
	"gitee.com/ling-bin/netwebSocket/netInterface"
	"sync"
)

// 连接管理模块
type ConnManager struct {
	connections sync.Map
}

//实例化管理
func NewConnManager() netInterface.IConnManager {
	return &ConnManager{}
}

//添加链接
func (c *ConnManager) Add(conn netInterface.IConnection) {
	connId := conn.GetConnId()
	c.connections.Store(connId, conn)
}

//删除连接
func (c *ConnManager) Remove(conn netInterface.IConnection) {
	connId := conn.GetConnId()
	c.connections.Delete(connId)
}

//利用ConnID获取链接
func (c *ConnManager) Get(connID uint64) (netInterface.IConnection, bool) {
	load, ok := c.connections.Load(connID)
	if ok {
		return load.(netInterface.IConnection), ok
	}
	return nil, false
}

//遍历连接
func (c *ConnManager) Range(hFunc func(key uint64, value netInterface.IConnection) bool) {
	c.connections.Range(func(key, value interface{}) bool {
		return hFunc(key.(uint64), value.(netInterface.IConnection))
	})
}

//获取个数
func (c *ConnManager) Count() int {
	count := 0
	c.connections.Range(func(key, value interface{}) bool {
		count++
		return true
	})
	return count
}

//清除并停止所有连接
func (c *ConnManager) ClearConn() {
	c.connections.Range(func(key, value interface{}) bool {
		value.(netInterface.IConnection).Stop()
		return true
	})
}