ws

type Client interface {
	Connect(context.Context) error
	Close()
	OnConnect(func(Connect))
	OnMessage(func(websocket.MessageType, []byte))
	OnClose(func(error))
}

type Connect interface {
	// SendText queues a text message. It returns nil if queued, error if closed.
	// Context is used for queuing timeout if channel is full.
	SendText(context.Context, string) error

	// SendBinary queues a binary message.
	SendBinary(context.Context, []byte) error

	// SendJSON queues a JSON message.
	SendJSON(context.Context, interface{}) error

	// ReadMessage blocks until a message is received or context done.
	ReadMessage(context.Context) (websocket.MessageType, []byte, error)

	// Close closes the connection and loops.
	Close() error

	// Conn returns underlying connection (use with caution).
	Conn() *websocket.Conn

	// ID returns the unique identifier for this connection
	ID() string
	// SetID sets the unique identifier
	SetID(string)

	// Get retrieves a value from metadata
	Get(key string) (value interface{}, exists bool)
	// Set stores a value in metadata
	Set(key string, value interface{})
}

type Hub interface {
	// Register 注册连接
	Register(Connect)
	// Unregister 注销连接
	Unregister(Connect)

	// Broadcast 广播消息给所有连接 (分布式)
	Broadcast(msg []byte)

	// SendTo 向特定 UserID 的连接发送消息 (分布式)
	SendTo(userID string, msg []byte)

	// Count 返回当前在线连接数 (本地)
	Count() int64

	// Close 关闭所有连接
	Close()
}

type MessageBroker interface {
	// Subscribe 订阅频道，当收到消息时调用 handler
	Subscribe(ctx context.Context, channel string, handler func(msg []byte)) error
	// Publish 发布消息到频道
	Publish(ctx context.Context, channel string, msg []byte) error
	// Close 关闭代理连接
	Close() error
}

type RedisBroker struct {
	// contains filtered or unexported fields
}

type Server interface {
	Start(func(Connect)) error
	Shutdown(context.Context) error
	Handler(func(Connect)) http.HandlerFunc
}

type ServerConfig struct {
	Addr string
}