socketio/server.v4.runback.go

package socketio

import (
	"errors"
	"fmt"
	
	siop "github.com/oarkflow/socketio/protocol"
	siot "github.com/oarkflow/socketio/transport"
)

func doConnectPacketV4(v4 *ServerV4) func(SocketID, siot.Socket, *Request) error {
	return func(socketID SocketID, socket siot.Socket, req *Request) (err error) {
		unlock := v4.prev.prev.prev.r()
		tr := v4.tr()
		unlock()
		
		tr.Join(socket.Namespace, socketID, socketID.Room(socketIDPrefix))
		
		stopBuffer := tr.(rawTransport).Transport(socketID).StartBuffer()
		defer stopBuffer()
		
		v4.setPrefix()
		v4.setSocketID(socketID)
		v4.setNsp(socket.Namespace)
		
		var h = handshakeV4{}
		switch val := socket.Data.(type) {
		case map[string]interface{}:
			h.Auth = func() map[string]interface{} {
				return val
			}
		}
		
		if fn, ok := v4.onConnect[socket.Namespace]; ok {
			return fn(&SocketV4{inSocketV4: v4.inSocketV4.clone(), req: req, han: h})
		}
		
		return ErrNamespaceNotFound.F(socket.Namespace)
	}
}

func runV4(v4 *ServerV4) func(SocketID, *Request) error {
	return func(socketID SocketID, req *Request) error {
		unlock := v4.prev.prev.prev.r()
		tr := v4.tr()
		unlock()
		
		for socket := range tr.Receive(socketID) {
			if err := doV4(v4, socketID, socket, req); err != nil {
				return err
			}
		}
		return nil
	}
}

func doV4(v4 *ServerV4, socketID SocketID, socket siot.Socket, req *Request) error {
	v1 := v4.prev.prev.prev
	
	switch socket.Type {
	case siop.ConnectPacket.Byte():
		unlock := v1.r()
		tr := v4.tr()
		unlock()
		
		if err := v1.doConnectPacket(socketID, socket, req); err != nil {
			if errors.Is(err, ErrNamespaceNotFound) {
				tr.Send(socketID, serviceError(fmt.Errorf("%valid namespace", "Inv")), siop.WithNamespace(socket.Namespace), siop.WithType(byte(siop.ConnectErrorPacket)))
				return nil
			}
			tr.Send(socketID, serviceError(err), siop.WithType(byte(siop.ConnectErrorPacket)))
			return nil
		}
		
		connectResponse := map[string]interface{}{"sid": socketID.String()}
		tr.Send(socketID, connectResponse, siop.WithType(siop.ConnectPacket.Byte()), siop.WithNamespace(socket.Namespace))
		tr.(rawTransport).Transport(socketID).SendBuffer()
		return nil
	}
	return doV3(v4.prev, socketID, socket, req)
}