98 lines
1.8 KiB
Go
98 lines
1.8 KiB
Go
|
package json
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/micro/go-micro/codec"
|
||
|
|
)
|
||
|
|
|
||
|
|
type clientCodec struct {
|
||
|
|
dec *json.Decoder // for reading JSON values
|
||
|
|
enc *json.Encoder // for writing JSON values
|
||
|
|
c io.Closer
|
||
|
|
|
||
|
|
// temporary work space
|
||
|
|
req clientRequest
|
||
|
|
resp clientResponse
|
||
|
|
|
||
|
|
sync.Mutex
|
||
|
|
pending map[uint64]string
|
||
|
|
}
|
||
|
|
|
||
|
|
type clientRequest struct {
|
||
|
|
Method string `json:"method"`
|
||
|
|
Params [1]interface{} `json:"params"`
|
||
|
|
ID uint64 `json:"id"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type clientResponse struct {
|
||
|
|
ID uint64 `json:"id"`
|
||
|
|
Result *json.RawMessage `json:"result"`
|
||
|
|
Error interface{} `json:"error"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func newClientCodec(conn io.ReadWriteCloser) *clientCodec {
|
||
|
|
return &clientCodec{
|
||
|
|
dec: json.NewDecoder(conn),
|
||
|
|
enc: json.NewEncoder(conn),
|
||
|
|
c: conn,
|
||
|
|
pending: make(map[uint64]string),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *clientCodec) Write(m *codec.Message, b interface{}) error {
|
||
|
|
c.Lock()
|
||
|
|
c.pending[m.Id] = m.Method
|
||
|
|
c.Unlock()
|
||
|
|
c.req.Method = m.Method
|
||
|
|
c.req.Params[0] = b
|
||
|
|
c.req.ID = m.Id
|
||
|
|
return c.enc.Encode(&c.req)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *clientResponse) reset() {
|
||
|
|
r.ID = 0
|
||
|
|
r.Result = nil
|
||
|
|
r.Error = nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *clientCodec) ReadHeader(m *codec.Message) error {
|
||
|
|
c.resp.reset()
|
||
|
|
if err := c.dec.Decode(&c.resp); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
c.Lock()
|
||
|
|
m.Method = c.pending[c.resp.ID]
|
||
|
|
delete(c.pending, c.resp.ID)
|
||
|
|
c.Unlock()
|
||
|
|
|
||
|
|
m.Error = ""
|
||
|
|
m.Id = c.resp.ID
|
||
|
|
if c.resp.Error != nil {
|
||
|
|
x, ok := c.resp.Error.(string)
|
||
|
|
if !ok {
|
||
|
|
return fmt.Errorf("invalid error %v", c.resp.Error)
|
||
|
|
}
|
||
|
|
if x == "" {
|
||
|
|
x = "unspecified error"
|
||
|
|
}
|
||
|
|
m.Error = x
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *clientCodec) ReadBody(x interface{}) error {
|
||
|
|
if x == nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return json.Unmarshal(*c.resp.Result, x)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (c *clientCodec) Close() error {
|
||
|
|
return c.c.Close()
|
||
|
|
}
|