mirror of
https://github.com/zoriya/Kyoo.git
synced 2025-05-24 02:02:36 -04:00
wip: Start subscriptions
This commit is contained in:
parent
96ad9424bb
commit
c1ebde75fd
@ -5,6 +5,7 @@ import "errors"
|
|||||||
type Client struct {
|
type Client struct {
|
||||||
id string
|
id string
|
||||||
messages chan []byte
|
messages chan []byte
|
||||||
|
subscriptions []string
|
||||||
}
|
}
|
||||||
|
|
||||||
type Message struct {
|
type Message struct {
|
||||||
@ -18,11 +19,24 @@ type Response struct {
|
|||||||
Error string `json:"error,omitempty"`
|
Error string `json:"error,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func NewClient() *Client{
|
||||||
|
return &Client{
|
||||||
|
messages: make(chan []byte, MAX_MESSAGE_QUEUE),
|
||||||
|
subscriptions: make([]string, 0),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (c *Client) HandleMessage(message Message) (interface{}, error) {
|
func (c *Client) HandleMessage(message Message) (interface{}, error) {
|
||||||
switch message.Action {
|
switch message.Action {
|
||||||
case "ping":
|
case "ping":
|
||||||
return "pong", nil
|
return "pong", nil
|
||||||
|
case "subscribe":
|
||||||
|
c.subscriptions = append(c.subscriptions, message.Value.(string))
|
||||||
|
return nil, nil
|
||||||
default:
|
default:
|
||||||
return nil, errors.New("invalid action")
|
return nil, errors.New("invalid action")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *Client) ListenChannels() {
|
||||||
|
}
|
||||||
|
@ -23,12 +23,11 @@ func main() {
|
|||||||
}
|
}
|
||||||
defer c.CloseNow()
|
defer c.CloseNow()
|
||||||
|
|
||||||
client := &Client{
|
client := NewClient()
|
||||||
messages: make(chan []byte, MAX_MESSAGE_QUEUE),
|
|
||||||
}
|
|
||||||
server.RegisterClient(client)
|
server.RegisterClient(client)
|
||||||
defer server.DeleteClient(client)
|
defer server.DeleteClient(client)
|
||||||
|
|
||||||
|
go client.ListenChannels()
|
||||||
for {
|
for {
|
||||||
var message Message
|
var message Message
|
||||||
err := wsjson.Read(r.Context(), c, &message)
|
err := wsjson.Read(r.Context(), c, &message)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user