RPC client (#42)
* Started RPC package to allow querying balances and sending raw transactions for sc's * integrate invoke cmd in cli * bumped version * added sendrawtransaction to the rpc client.
This commit is contained in:
parent
1a1a19da7d
commit
b2a5e34aac
7 changed files with 343 additions and 4 deletions
123
pkg/rpc/client.go
Normal file
123
pkg/rpc/client.go
Normal file
|
@ -0,0 +1,123 @@
|
|||
package rpc
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
defaultDialTimeout = 4 * time.Second
|
||||
defaultRequestTimeout = 4 * time.Second
|
||||
defaultClientVersion = "2.0"
|
||||
)
|
||||
|
||||
// Client represents the middleman for executing JSON RPC calls
|
||||
// to remote NEO RPC nodes.
|
||||
type Client struct {
|
||||
// The underlying http client. It's never a good practice to use
|
||||
// the http.DefaultClient, therefore we will role our own.
|
||||
http.Client
|
||||
endpoint *url.URL
|
||||
ctx context.Context
|
||||
version string
|
||||
}
|
||||
|
||||
// ClientOptions defines options for the RPC client.
|
||||
// All Values are optional. If any duration is not specified
|
||||
// a default of 3 seconds will be used.
|
||||
type ClientOptions struct {
|
||||
Cert string
|
||||
Key string
|
||||
CACert string
|
||||
DialTimeout time.Duration
|
||||
RequestTimeout time.Duration
|
||||
// Version is the version of the client that will be send
|
||||
// along with the request body. If no version is specified
|
||||
// the default version (currently 2.0) will be used.
|
||||
Version string
|
||||
}
|
||||
|
||||
// NewClient return a new Client ready to use.
|
||||
func NewClient(ctx context.Context, endpoint string, opts ClientOptions) (*Client, error) {
|
||||
url, err := url.Parse(endpoint)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if opts.DialTimeout == 0 {
|
||||
opts.DialTimeout = defaultDialTimeout
|
||||
}
|
||||
if opts.RequestTimeout == 0 {
|
||||
opts.RequestTimeout = defaultRequestTimeout
|
||||
}
|
||||
if opts.Version == "" {
|
||||
opts.Version = defaultClientVersion
|
||||
}
|
||||
|
||||
transport := &http.Transport{
|
||||
Dial: (&net.Dialer{
|
||||
Timeout: opts.DialTimeout,
|
||||
}).Dial,
|
||||
}
|
||||
|
||||
// TODO(@antdm): Enable SSL.
|
||||
if opts.Cert != "" && opts.Key != "" {
|
||||
|
||||
}
|
||||
|
||||
return &Client{
|
||||
Client: http.Client{
|
||||
Timeout: opts.RequestTimeout,
|
||||
Transport: transport,
|
||||
},
|
||||
endpoint: url,
|
||||
ctx: ctx,
|
||||
version: opts.Version,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (c *Client) performRequest(method string, p params, v interface{}) error {
|
||||
r := request{
|
||||
JSONRPC: c.version,
|
||||
Method: method,
|
||||
Params: p.values,
|
||||
ID: 1,
|
||||
}
|
||||
|
||||
b, err := json.Marshal(r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
req, err := http.NewRequest("POST", c.endpoint.String(), bytes.NewReader(b))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
resp, err := c.Do(req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return fmt.Errorf("Remote responded with a non 200 response: %d", resp.StatusCode)
|
||||
}
|
||||
|
||||
return json.NewDecoder(resp.Body).Decode(v)
|
||||
}
|
||||
|
||||
// Ping attempts to create a connection to the endpoint.
|
||||
// and returns an error if there is one.
|
||||
func (c *Client) Ping() error {
|
||||
conn, err := net.DialTimeout("tcp", c.endpoint.Host, defaultDialTimeout)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_ = conn.Close()
|
||||
return nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue