rpc/server/rwc/websocket/fasthttp/servlet_rwc_handlers.go
crusader e3a873a067 ing
2018-03-22 20:51:20 +09:00

74 lines
1.7 KiB
Go

package fasthttp
import (
"git.loafle.net/commons_go/rpc"
"git.loafle.net/commons_go/rpc/protocol"
cwf "git.loafle.net/commons_go/websocket_fasthttp"
"git.loafle.net/commons_go/websocket_fasthttp/websocket"
)
func New() rpc.ServletReadWriteCloseHandler {
return &ServletReadWriteCloseHandlers{}
}
type ServletReadWriteCloseHandlers struct {
rpc.ServletReadWriteCloseHandlers
}
func (srwch *ServletReadWriteCloseHandlers) ReadRequest(servletCTX rpc.ServletContext, codec protocol.ServerCodec, conn interface{}) (protocol.ServerRequestCodec, error) {
soc := conn.(cwf.Socket)
_, r, err := soc.NextReader()
requestCodec, err := codec.NewRequest(r)
return requestCodec, err
}
func (srwch *ServletReadWriteCloseHandlers) WriteResponse(servletCTX rpc.ServletContext, conn interface{}, requestCodec protocol.ServerRequestCodec, result interface{}, err error) error {
soc := conn.(cwf.Socket)
wc, wErr := soc.NextWriter(websocket.TextMessage)
if nil != wErr {
return wErr
}
defer func() {
wc.Close()
}()
if nil != err {
if wErr := requestCodec.WriteError(wc, 500, err); nil != wErr {
return wErr
}
} else {
if wErr := requestCodec.WriteResponse(wc, result); nil != wErr {
return wErr
}
}
return nil
}
func (srwch *ServletReadWriteCloseHandlers) WriteNotification(servletCTX rpc.ServletContext, conn interface{}, codec protocol.ServerCodec, method string, params []interface{}) error {
soc := conn.(cwf.Socket)
wc, wErr := soc.NextWriter(websocket.TextMessage)
if nil != wErr {
return wErr
}
defer func() {
wc.Close()
}()
if wErr := codec.WriteNotification(wc, method, params); nil != wErr {
return wErr
}
return nil
}
func (srwch *ServletReadWriteCloseHandlers) Validate() {
srwch.ServletReadWriteCloseHandlers.Validate()
}