rpc-go/protocol/json/client.go
2018-08-22 18:04:25 +09:00

60 lines
1.4 KiB
Go

package json
import (
"encoding/json"
"git.loafle.net/overflow/rpc-go/codec"
"git.loafle.net/overflow/rpc-go/protocol"
)
// ----------------------------------------------------------------------------
// Codec
// ----------------------------------------------------------------------------
// NewClientCodec returns a new JSON Codec.
func NewCustomClientCodec(codecSelector codec.CodecSelector) protocol.ClientCodec {
return &ClientCodec{
codecSelector: codecSelector,
}
}
// NewClientCodec returns a new JSON Codec.
func NewClientCodec() protocol.ClientCodec {
return NewCustomClientCodec(codec.DefaultCodecSelector)
}
// ClientCodec creates a ClientCodecRequest to process each request.
type ClientCodec struct {
codecSelector codec.CodecSelector
}
func (cc *ClientCodec) NewRequest(method string, args []interface{}, id interface{}) (messageType int, message []byte, err error) {
params, err := convertParamsToStringArray(args)
if nil != err {
return 0, nil, err
}
req := &clientRequest{
Version: Version,
Method: method,
Params: params,
ID: id,
}
buf, err := json.Marshal(req)
if nil != err {
return 0, nil, err
}
return cc.codecSelector.Encode(buf)
}
func (cc *ClientCodec) NewResponse(messageType int, message []byte) (protocol.ClientResponseCodec, error) {
buf, err := cc.codecSelector.Decode(messageType, message)
if nil != err {
return nil, err
}
return newClientResponseCodec(buf)
}