package ws

import "sync"

//IOnline 联机在线管理（已废弃，请使用 IClientsMgr ）
type IOnline interface {
	AddClient(c *Client)
	DeleteClient(c *Client)

	Clients() []*Client //return all clients
	Conns() int64       //return clients num
}

func NewOnlineWorker() *OnlineWorker {
	return &OnlineWorker{
		clients: make(map[*Client]bool, 128),
	}
}

//OnlineWorker 联机管理worker
type OnlineWorker struct {
	clients map[*Client]bool

	//总连接数量
	conn int64

	mu sync.RWMutex
}

func (w *OnlineWorker) AddClient(c *Client) {
	w.mu.Lock()
	w.clients[c] = true
	w.conn += int64(1)
	w.mu.Unlock()
}

func (w *OnlineWorker) DeleteClient(c *Client) {
	w.mu.RLock()
	_, exist := w.clients[c]
	w.mu.RUnlock()

	if !exist {
		return
	}

	w.mu.Lock()
	delete(w.clients, c)
	w.conn -= int64(1)
	w.mu.Unlock()

}

func (w *OnlineWorker) Clients() []*Client {
	clients := make([]*Client, 0)

	w.mu.RLock()
	for c := range w.clients {
		clients = append(clients, c)
	}
	w.mu.RUnlock()

	return clients
}

func (w *OnlineWorker) Conns() int64 {
	return w.conn
}