connect/server/hub.go

package server

import (
	"sync"

	"sikey.com/websocket/stackexchange"
)

type HubConfig struct {
	ConnectSize    int
	DisconnectSize int
	MessageSize    int

	StackExchange *stackexchange.StackExchange
}

type Hub struct {
	clients map[string]*Client
	mutex   sync.RWMutex

	Connect    chan *Client
	Disconnect chan *Client
	Message    chan *Message
}

func NewHub(cfg HubConfig) *Hub {
	hub := &Hub{
		clients: make(map[string]*Client),
		mutex:   sync.RWMutex{},

		Connect:    make(chan *Client, cfg.ConnectSize),
		Disconnect: make(chan *Client, cfg.DisconnectSize),
		Message:    make(chan *Message, cfg.MessageSize),
	}

	go hub.run()
	go hub.remote() // 远程消息
	return hub
}

func (h *Hub) run() {
	for {
		select {
		case client := <-h.Connect:
			h.clients[client.UserId] = client
		case client := <-h.Disconnect:
			close(client.Send)
			delete(h.clients, client.UserId)
		case message := <-h.Message:
			if client, ok := h.clients[message.receiver]; ok {
				client.Send <- message
			} else {
				// 不在同一台服务器, 这里将消息发送至拓展应用
			}
		}
	}
}

func (h *Hub) remote() {
}