-
Notifications
You must be signed in to change notification settings - Fork 1.7k
/
jsonrpccodec.go
101 lines (90 loc) · 2.49 KB
/
jsonrpccodec.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package api
import (
"encoding/json"
"errors"
"fmt"
)
// Wrapping/unwrapping Message objects into JSON RPC ones folllowing https://www.jsonrpc.org/specification
type JsonRPCRequest struct {
Version string `json:"jsonrpc"`
Id string `json:"id"`
Method string `json:"method"`
Params *Message `json:"params,omitempty"`
}
type JsonRPCResponse struct {
Version string `json:"jsonrpc"`
Id string `json:"id"`
Result *Message `json:"result,omitempty"`
Error *JsonRPCError `json:"error,omitempty"`
}
// JSON-RPC error can only be sent to users. It is not used for messages between Gateways and Nodes.
type JsonRPCError struct {
Code int `json:"code"`
Message string `json:"message"`
Data json.RawMessage `json:"data,omitempty"`
}
type JsonRPCCodec struct {
}
var _ Codec = (*JsonRPCCodec)(nil)
func (*JsonRPCCodec) DecodeRequest(msgBytes []byte) (*Message, error) {
var request JsonRPCRequest
err := json.Unmarshal(msgBytes, &request)
if err != nil {
return nil, err
}
if request.Version != "2.0" {
return nil, errors.New("incorrect jsonrpc version")
}
if request.Method == "" {
return nil, errors.New("empty method field")
}
if request.Params == nil {
return nil, errors.New("missing params attribute")
}
request.Params.Body.MessageId = request.Id
request.Params.Body.Method = request.Method
return request.Params, nil
}
func (*JsonRPCCodec) EncodeRequest(msg *Message) ([]byte, error) {
request := JsonRPCRequest{
Version: "2.0",
Id: msg.Body.MessageId,
Method: msg.Body.Method,
Params: msg,
}
return json.Marshal(request)
}
func (*JsonRPCCodec) DecodeResponse(msgBytes []byte) (*Message, error) {
var response JsonRPCResponse
err := json.Unmarshal(msgBytes, &response)
if err != nil {
return nil, err
}
if response.Error != nil {
return nil, fmt.Errorf("received non-empty error field: %v", response.Error)
}
if response.Result != nil {
response.Result.Body.MessageId = response.Id
}
return response.Result, nil
}
func (*JsonRPCCodec) EncodeResponse(msg *Message) ([]byte, error) {
response := JsonRPCResponse{
Version: "2.0",
Id: msg.Body.MessageId,
Result: msg,
}
return json.Marshal(response)
}
func (*JsonRPCCodec) EncodeNewErrorResponse(id string, code int, message string, data []byte) ([]byte, error) {
response := JsonRPCResponse{
Version: "2.0",
Id: id,
Error: &JsonRPCError{
Code: code,
Message: message,
Data: data,
},
}
return json.Marshal(response)
}