122 lines
2.4 KiB
Go
122 lines
2.4 KiB
Go
|
package socketio
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"github.com/googollee/go-socket.io/parser"
|
||
|
|
)
|
||
|
|
|
||
|
|
var readHandlerMapping = map[parser.Type]readHandler{
|
||
|
|
parser.Ack: ackPacketHandler,
|
||
|
|
parser.Connect: connectPacketHandler,
|
||
|
|
parser.Disconnect: disconnectPacketHandler,
|
||
|
|
}
|
||
|
|
|
||
|
|
func ackPacketHandler(c *conn, header parser.Header) error {
|
||
|
|
conn, ok := c.namespaces.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
_ = c.decoder.DiscardLast()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
conn.dispatch(header)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func eventPacketHandler(c *conn, event string, header parser.Header) error {
|
||
|
|
conn, ok := c.namespaces.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
_ = c.decoder.DiscardLast()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
handler, ok := c.handlers.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
_ = c.decoder.DiscardLast()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
args, err := c.decoder.DecodeArgs(handler.getEventTypes(event))
|
||
|
|
if err != nil {
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return errDecodeArgs
|
||
|
|
}
|
||
|
|
|
||
|
|
ret, err := handler.dispatchEvent(conn, event, args...)
|
||
|
|
if err != nil {
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return errHandleDispatch
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(ret) > 0 {
|
||
|
|
header.Type = parser.Ack
|
||
|
|
c.write(header, ret...)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func connectPacketHandler(c *conn, header parser.Header) error {
|
||
|
|
if err := c.decoder.DiscardLast(); err != nil {
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
handler, ok := c.handlers.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
c.onError(header.Namespace, errFailedConnectNamespace)
|
||
|
|
return errFailedConnectNamespace
|
||
|
|
}
|
||
|
|
|
||
|
|
conn, ok := c.namespaces.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
conn = newNamespaceConn(c, header.Namespace, handler.broadcast)
|
||
|
|
c.namespaces.Set(header.Namespace, conn)
|
||
|
|
conn.Join(c.ID())
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err := handler.dispatch(conn, header)
|
||
|
|
if err != nil {
|
||
|
|
log.Println("dispatch connect packet", err)
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return errHandleDispatch
|
||
|
|
}
|
||
|
|
|
||
|
|
c.write(header)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func disconnectPacketHandler(c *conn, header parser.Header) error {
|
||
|
|
args, err := c.decoder.DecodeArgs(defaultHeaderType)
|
||
|
|
if err != nil {
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return errDecodeArgs
|
||
|
|
}
|
||
|
|
|
||
|
|
conn, ok := c.namespaces.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
_ = c.decoder.DiscardLast()
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
conn.LeaveAll()
|
||
|
|
|
||
|
|
c.namespaces.Delete(header.Namespace)
|
||
|
|
|
||
|
|
handler, ok := c.handlers.Get(header.Namespace)
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = handler.dispatch(conn, header, args...)
|
||
|
|
if err != nil {
|
||
|
|
log.Println("dispatch disconnect packet", err)
|
||
|
|
c.onError(header.Namespace, err)
|
||
|
|
return errHandleDispatch
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|