rpc/connection/websocket/fasthttp/servlet_handlers.go

package fasthttp

import (
	"fmt"

	"git.loafle.net/commons_go/rpc"
	"git.loafle.net/commons_go/rpc/protocol"
	"git.loafle.net/commons_go/websocket_fasthttp/websocket"
)

type ServletHandlers struct {
}

func (sh *ServletHandlers) GetRequest(servletCTX rpc.ServletContext, codec protocol.ServerCodec, reader interface{}) (protocol.ServerRequestCodec, error) {
	conn := reader.(*websocket.Conn)
	_, r, err := conn.NextReader()

	requestCodec, err := codec.NewRequest(r)

	return requestCodec, err
}

func (sh *ServletHandlers) SendResponse(servletCTX rpc.ServletContext, requestCodec protocol.ServerRequestCodec, writer interface{}, result interface{}, err error) error {
	conn := writer.(*websocket.Conn)

	wc, lerr := conn.NextWriter(websocket.TextMessage)
	if nil != lerr {

	}

	if nil != err {
		if lerr := requestCodec.WriteError(wc, 500, err); nil != lerr {

		}
	} else {
		if err := requestCodec.WriteResponse(wc, result); nil != err {

		}
	}

	return fmt.Errorf("Servlet Handler: SendResponse is not implemented")
}

func (sh *ServletHandlers) SendNotification(servletCTX rpc.ServletContext, codec protocol.ServerCodec, writer interface{}, method string, args ...interface{}) error {
	conn := writer.(*websocket.Conn)

	wc, lerr := conn.NextWriter(websocket.TextMessage)
	if nil != lerr {

	}

	if err := codec.WriteNotification(wc, method, args); nil != err {

	}

	return fmt.Errorf("Servlet Handler: SendNotification is not implemented")
}