mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-22 11:41:19 +00:00
c9a0b36a5f
This fixes an issue with websocket ping frame handling.
834 lines
24 KiB
Go
834 lines
24 KiB
Go
// Copyright 2016 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package rpc
|
|
|
|
import (
|
|
"bytes"
|
|
"container/list"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"net/url"
|
|
"os"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
)
|
|
|
|
var (
|
|
ErrClientQuit = errors.New("client is closed")
|
|
ErrNoResult = errors.New("no result in JSON-RPC response")
|
|
ErrSubscriptionQueueOverflow = errors.New("subscription queue overflow")
|
|
)
|
|
|
|
const (
|
|
// Timeouts
|
|
tcpKeepAliveInterval = 30 * time.Second
|
|
defaultDialTimeout = 10 * time.Second // used when dialing if the context has no deadline
|
|
defaultWriteTimeout = 10 * time.Second // used for calls if the context has no deadline
|
|
subscribeTimeout = 5 * time.Second // overall timeout eth_subscribe, rpc_modules calls
|
|
)
|
|
|
|
const (
|
|
// Subscriptions are removed when the subscriber cannot keep up.
|
|
//
|
|
// This can be worked around by supplying a channel with sufficiently sized buffer,
|
|
// but this can be inconvenient and hard to explain in the docs. Another issue with
|
|
// buffered channels is that the buffer is static even though it might not be needed
|
|
// most of the time.
|
|
//
|
|
// The approach taken here is to maintain a per-subscription linked list buffer
|
|
// shrinks on demand. If the buffer reaches the size below, the subscription is
|
|
// dropped.
|
|
maxClientSubscriptionBuffer = 20000
|
|
)
|
|
|
|
// BatchElem is an element in a batch request.
|
|
type BatchElem struct {
|
|
Method string
|
|
Args []interface{}
|
|
// The result is unmarshaled into this field. Result must be set to a
|
|
// non-nil pointer value of the desired type, otherwise the response will be
|
|
// discarded.
|
|
Result interface{}
|
|
// Error is set if the server returns an error for this request, or if
|
|
// unmarshaling into Result fails. It is not set for I/O errors.
|
|
Error error
|
|
}
|
|
|
|
// A value of this type can a JSON-RPC request, notification, successful response or
|
|
// error response. Which one it is depends on the fields.
|
|
type jsonrpcMessage struct {
|
|
Version string `json:"jsonrpc"`
|
|
ID json.RawMessage `json:"id,omitempty"`
|
|
Method string `json:"method,omitempty"`
|
|
Params json.RawMessage `json:"params,omitempty"`
|
|
Error *jsonError `json:"error,omitempty"`
|
|
Result json.RawMessage `json:"result,omitempty"`
|
|
}
|
|
|
|
func (msg *jsonrpcMessage) isNotification() bool {
|
|
return msg.ID == nil && msg.Method != ""
|
|
}
|
|
|
|
func (msg *jsonrpcMessage) isResponse() bool {
|
|
return msg.hasValidID() && msg.Method == "" && len(msg.Params) == 0
|
|
}
|
|
|
|
func (msg *jsonrpcMessage) hasValidID() bool {
|
|
return len(msg.ID) > 0 && msg.ID[0] != '{' && msg.ID[0] != '['
|
|
}
|
|
|
|
func (msg *jsonrpcMessage) String() string {
|
|
b, _ := json.Marshal(msg)
|
|
return string(b)
|
|
}
|
|
|
|
// Client represents a connection to an RPC server.
|
|
type Client struct {
|
|
idCounter uint32
|
|
connectFunc func(ctx context.Context) (net.Conn, error)
|
|
isHTTP bool
|
|
|
|
// writeConn is only safe to access outside dispatch, with the
|
|
// write lock held. The write lock is taken by sending on
|
|
// requestOp and released by sending on sendDone.
|
|
writeConn net.Conn
|
|
|
|
// for dispatch
|
|
close chan struct{}
|
|
didQuit chan struct{} // closed when client quits
|
|
reconnected chan net.Conn // where write/reconnect sends the new connection
|
|
readErr chan error // errors from read
|
|
readResp chan []*jsonrpcMessage // valid messages from read
|
|
requestOp chan *requestOp // for registering response IDs
|
|
sendDone chan error // signals write completion, releases write lock
|
|
respWait map[string]*requestOp // active requests
|
|
subs map[string]*ClientSubscription // active subscriptions
|
|
}
|
|
|
|
type requestOp struct {
|
|
ids []json.RawMessage
|
|
err error
|
|
resp chan *jsonrpcMessage // receives up to len(ids) responses
|
|
sub *ClientSubscription // only set for EthSubscribe requests
|
|
}
|
|
|
|
func (op *requestOp) wait(ctx context.Context) (*jsonrpcMessage, error) {
|
|
select {
|
|
case <-ctx.Done():
|
|
return nil, ctx.Err()
|
|
case resp := <-op.resp:
|
|
return resp, op.err
|
|
}
|
|
}
|
|
|
|
// Dial creates a new client for the given URL.
|
|
//
|
|
// The currently supported URL schemes are "http", "https", "ws" and "wss". If rawurl is a
|
|
// file name with no URL scheme, a local socket connection is established using UNIX
|
|
// domain sockets on supported platforms and named pipes on Windows. If you want to
|
|
// configure transport options, use DialHTTP, DialWebsocket or DialIPC instead.
|
|
//
|
|
// For websocket connections, the origin is set to the local host name.
|
|
//
|
|
// The client reconnects automatically if the connection is lost.
|
|
func Dial(rawurl string) (*Client, error) {
|
|
return DialContext(context.Background(), rawurl)
|
|
}
|
|
|
|
// DialContext creates a new RPC client, just like Dial.
|
|
//
|
|
// The context is used to cancel or time out the initial connection establishment. It does
|
|
// not affect subsequent interactions with the client.
|
|
func DialContext(ctx context.Context, rawurl string) (*Client, error) {
|
|
u, err := url.Parse(rawurl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
switch u.Scheme {
|
|
case "http", "https":
|
|
return DialHTTP(rawurl)
|
|
case "ws", "wss":
|
|
return DialWebsocket(ctx, rawurl, "")
|
|
case "stdio":
|
|
return DialStdIO(ctx)
|
|
case "":
|
|
return DialIPC(ctx, rawurl)
|
|
default:
|
|
return nil, fmt.Errorf("no known transport for URL scheme %q", u.Scheme)
|
|
}
|
|
}
|
|
|
|
type StdIOConn struct{}
|
|
|
|
func (io StdIOConn) Read(b []byte) (n int, err error) {
|
|
return os.Stdin.Read(b)
|
|
}
|
|
|
|
func (io StdIOConn) Write(b []byte) (n int, err error) {
|
|
return os.Stdout.Write(b)
|
|
}
|
|
|
|
func (io StdIOConn) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (io StdIOConn) LocalAddr() net.Addr {
|
|
return &net.UnixAddr{Name: "stdio", Net: "stdio"}
|
|
}
|
|
|
|
func (io StdIOConn) RemoteAddr() net.Addr {
|
|
return &net.UnixAddr{Name: "stdio", Net: "stdio"}
|
|
}
|
|
|
|
func (io StdIOConn) SetDeadline(t time.Time) error {
|
|
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
|
|
}
|
|
|
|
func (io StdIOConn) SetReadDeadline(t time.Time) error {
|
|
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
|
|
}
|
|
|
|
func (io StdIOConn) SetWriteDeadline(t time.Time) error {
|
|
return &net.OpError{Op: "set", Net: "stdio", Source: nil, Addr: nil, Err: errors.New("deadline not supported")}
|
|
}
|
|
func DialStdIO(ctx context.Context) (*Client, error) {
|
|
return newClient(ctx, func(_ context.Context) (net.Conn, error) {
|
|
return StdIOConn{}, nil
|
|
})
|
|
}
|
|
|
|
func newClient(initctx context.Context, connectFunc func(context.Context) (net.Conn, error)) (*Client, error) {
|
|
conn, err := connectFunc(initctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_, isHTTP := conn.(*httpConn)
|
|
c := &Client{
|
|
writeConn: conn,
|
|
isHTTP: isHTTP,
|
|
connectFunc: connectFunc,
|
|
close: make(chan struct{}),
|
|
didQuit: make(chan struct{}),
|
|
reconnected: make(chan net.Conn),
|
|
readErr: make(chan error),
|
|
readResp: make(chan []*jsonrpcMessage),
|
|
requestOp: make(chan *requestOp),
|
|
sendDone: make(chan error, 1),
|
|
respWait: make(map[string]*requestOp),
|
|
subs: make(map[string]*ClientSubscription),
|
|
}
|
|
if !isHTTP {
|
|
go c.dispatch(conn)
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
func (c *Client) nextID() json.RawMessage {
|
|
id := atomic.AddUint32(&c.idCounter, 1)
|
|
return []byte(strconv.FormatUint(uint64(id), 10))
|
|
}
|
|
|
|
// SupportedModules calls the rpc_modules method, retrieving the list of
|
|
// APIs that are available on the server.
|
|
func (c *Client) SupportedModules() (map[string]string, error) {
|
|
var result map[string]string
|
|
ctx, cancel := context.WithTimeout(context.Background(), subscribeTimeout)
|
|
defer cancel()
|
|
err := c.CallContext(ctx, &result, "rpc_modules")
|
|
return result, err
|
|
}
|
|
|
|
// Close closes the client, aborting any in-flight requests.
|
|
func (c *Client) Close() {
|
|
if c.isHTTP {
|
|
return
|
|
}
|
|
select {
|
|
case c.close <- struct{}{}:
|
|
<-c.didQuit
|
|
case <-c.didQuit:
|
|
}
|
|
}
|
|
|
|
// Call performs a JSON-RPC call with the given arguments and unmarshals into
|
|
// result if no error occurred.
|
|
//
|
|
// The result must be a pointer so that package json can unmarshal into it. You
|
|
// can also pass nil, in which case the result is ignored.
|
|
func (c *Client) Call(result interface{}, method string, args ...interface{}) error {
|
|
ctx := context.Background()
|
|
return c.CallContext(ctx, result, method, args...)
|
|
}
|
|
|
|
// CallContext performs a JSON-RPC call with the given arguments. If the context is
|
|
// canceled before the call has successfully returned, CallContext returns immediately.
|
|
//
|
|
// The result must be a pointer so that package json can unmarshal into it. You
|
|
// can also pass nil, in which case the result is ignored.
|
|
func (c *Client) CallContext(ctx context.Context, result interface{}, method string, args ...interface{}) error {
|
|
msg, err := c.newMessage(method, args...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
op := &requestOp{ids: []json.RawMessage{msg.ID}, resp: make(chan *jsonrpcMessage, 1)}
|
|
|
|
if c.isHTTP {
|
|
err = c.sendHTTP(ctx, op, msg)
|
|
} else {
|
|
err = c.send(ctx, op, msg)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// dispatch has accepted the request and will close the channel when it quits.
|
|
switch resp, err := op.wait(ctx); {
|
|
case err != nil:
|
|
return err
|
|
case resp.Error != nil:
|
|
return resp.Error
|
|
case len(resp.Result) == 0:
|
|
return ErrNoResult
|
|
default:
|
|
return json.Unmarshal(resp.Result, &result)
|
|
}
|
|
}
|
|
|
|
// BatchCall sends all given requests as a single batch and waits for the server
|
|
// to return a response for all of them.
|
|
//
|
|
// In contrast to Call, BatchCall only returns I/O errors. Any error specific to
|
|
// a request is reported through the Error field of the corresponding BatchElem.
|
|
//
|
|
// Note that batch calls may not be executed atomically on the server side.
|
|
func (c *Client) BatchCall(b []BatchElem) error {
|
|
ctx := context.Background()
|
|
return c.BatchCallContext(ctx, b)
|
|
}
|
|
|
|
// BatchCall sends all given requests as a single batch and waits for the server
|
|
// to return a response for all of them. The wait duration is bounded by the
|
|
// context's deadline.
|
|
//
|
|
// In contrast to CallContext, BatchCallContext only returns errors that have occurred
|
|
// while sending the request. Any error specific to a request is reported through the
|
|
// Error field of the corresponding BatchElem.
|
|
//
|
|
// Note that batch calls may not be executed atomically on the server side.
|
|
func (c *Client) BatchCallContext(ctx context.Context, b []BatchElem) error {
|
|
msgs := make([]*jsonrpcMessage, len(b))
|
|
op := &requestOp{
|
|
ids: make([]json.RawMessage, len(b)),
|
|
resp: make(chan *jsonrpcMessage, len(b)),
|
|
}
|
|
for i, elem := range b {
|
|
msg, err := c.newMessage(elem.Method, elem.Args...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
msgs[i] = msg
|
|
op.ids[i] = msg.ID
|
|
}
|
|
|
|
var err error
|
|
if c.isHTTP {
|
|
err = c.sendBatchHTTP(ctx, op, msgs)
|
|
} else {
|
|
err = c.send(ctx, op, msgs)
|
|
}
|
|
|
|
// Wait for all responses to come back.
|
|
for n := 0; n < len(b) && err == nil; n++ {
|
|
var resp *jsonrpcMessage
|
|
resp, err = op.wait(ctx)
|
|
if err != nil {
|
|
break
|
|
}
|
|
// Find the element corresponding to this response.
|
|
// The element is guaranteed to be present because dispatch
|
|
// only sends valid IDs to our channel.
|
|
var elem *BatchElem
|
|
for i := range msgs {
|
|
if bytes.Equal(msgs[i].ID, resp.ID) {
|
|
elem = &b[i]
|
|
break
|
|
}
|
|
}
|
|
if resp.Error != nil {
|
|
elem.Error = resp.Error
|
|
continue
|
|
}
|
|
if len(resp.Result) == 0 {
|
|
elem.Error = ErrNoResult
|
|
continue
|
|
}
|
|
elem.Error = json.Unmarshal(resp.Result, elem.Result)
|
|
}
|
|
return err
|
|
}
|
|
|
|
// EthSubscribe registers a subscripion under the "eth" namespace.
|
|
func (c *Client) EthSubscribe(ctx context.Context, channel interface{}, args ...interface{}) (*ClientSubscription, error) {
|
|
return c.Subscribe(ctx, "eth", channel, args...)
|
|
}
|
|
|
|
// ShhSubscribe registers a subscripion under the "shh" namespace.
|
|
func (c *Client) ShhSubscribe(ctx context.Context, channel interface{}, args ...interface{}) (*ClientSubscription, error) {
|
|
return c.Subscribe(ctx, "shh", channel, args...)
|
|
}
|
|
|
|
// Subscribe calls the "<namespace>_subscribe" method with the given arguments,
|
|
// registering a subscription. Server notifications for the subscription are
|
|
// sent to the given channel. The element type of the channel must match the
|
|
// expected type of content returned by the subscription.
|
|
//
|
|
// The context argument cancels the RPC request that sets up the subscription but has no
|
|
// effect on the subscription after Subscribe has returned.
|
|
//
|
|
// Slow subscribers will be dropped eventually. Client buffers up to 8000 notifications
|
|
// before considering the subscriber dead. The subscription Err channel will receive
|
|
// ErrSubscriptionQueueOverflow. Use a sufficiently large buffer on the channel or ensure
|
|
// that the channel usually has at least one reader to prevent this issue.
|
|
func (c *Client) Subscribe(ctx context.Context, namespace string, channel interface{}, args ...interface{}) (*ClientSubscription, error) {
|
|
// Check type of channel first.
|
|
chanVal := reflect.ValueOf(channel)
|
|
if chanVal.Kind() != reflect.Chan || chanVal.Type().ChanDir()&reflect.SendDir == 0 {
|
|
panic("first argument to Subscribe must be a writable channel")
|
|
}
|
|
if chanVal.IsNil() {
|
|
panic("channel given to Subscribe must not be nil")
|
|
}
|
|
if c.isHTTP {
|
|
return nil, ErrNotificationsUnsupported
|
|
}
|
|
|
|
msg, err := c.newMessage(namespace+subscribeMethodSuffix, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
op := &requestOp{
|
|
ids: []json.RawMessage{msg.ID},
|
|
resp: make(chan *jsonrpcMessage),
|
|
sub: newClientSubscription(c, namespace, chanVal),
|
|
}
|
|
|
|
// Send the subscription request.
|
|
// The arrival and validity of the response is signaled on sub.quit.
|
|
if err := c.send(ctx, op, msg); err != nil {
|
|
return nil, err
|
|
}
|
|
if _, err := op.wait(ctx); err != nil {
|
|
return nil, err
|
|
}
|
|
return op.sub, nil
|
|
}
|
|
|
|
func (c *Client) newMessage(method string, paramsIn ...interface{}) (*jsonrpcMessage, error) {
|
|
params, err := json.Marshal(paramsIn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &jsonrpcMessage{Version: "2.0", ID: c.nextID(), Method: method, Params: params}, nil
|
|
}
|
|
|
|
// send registers op with the dispatch loop, then sends msg on the connection.
|
|
// if sending fails, op is deregistered.
|
|
func (c *Client) send(ctx context.Context, op *requestOp, msg interface{}) error {
|
|
select {
|
|
case c.requestOp <- op:
|
|
log.Trace("", "msg", log.Lazy{Fn: func() string {
|
|
return fmt.Sprint("sending ", msg)
|
|
}})
|
|
err := c.write(ctx, msg)
|
|
c.sendDone <- err
|
|
return err
|
|
case <-ctx.Done():
|
|
// This can happen if the client is overloaded or unable to keep up with
|
|
// subscription notifications.
|
|
return ctx.Err()
|
|
case <-c.didQuit:
|
|
return ErrClientQuit
|
|
}
|
|
}
|
|
|
|
func (c *Client) write(ctx context.Context, msg interface{}) error {
|
|
deadline, ok := ctx.Deadline()
|
|
if !ok {
|
|
deadline = time.Now().Add(defaultWriteTimeout)
|
|
}
|
|
// The previous write failed. Try to establish a new connection.
|
|
if c.writeConn == nil {
|
|
if err := c.reconnect(ctx); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
c.writeConn.SetWriteDeadline(deadline)
|
|
err := json.NewEncoder(c.writeConn).Encode(msg)
|
|
c.writeConn.SetWriteDeadline(time.Time{})
|
|
if err != nil {
|
|
c.writeConn = nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (c *Client) reconnect(ctx context.Context) error {
|
|
newconn, err := c.connectFunc(ctx)
|
|
if err != nil {
|
|
log.Trace(fmt.Sprintf("reconnect failed: %v", err))
|
|
return err
|
|
}
|
|
select {
|
|
case c.reconnected <- newconn:
|
|
c.writeConn = newconn
|
|
return nil
|
|
case <-c.didQuit:
|
|
newconn.Close()
|
|
return ErrClientQuit
|
|
}
|
|
}
|
|
|
|
// dispatch is the main loop of the client.
|
|
// It sends read messages to waiting calls to Call and BatchCall
|
|
// and subscription notifications to registered subscriptions.
|
|
func (c *Client) dispatch(conn net.Conn) {
|
|
// Spawn the initial read loop.
|
|
go c.read(conn)
|
|
|
|
var (
|
|
lastOp *requestOp // tracks last send operation
|
|
requestOpLock = c.requestOp // nil while the send lock is held
|
|
reading = true // if true, a read loop is running
|
|
)
|
|
defer close(c.didQuit)
|
|
defer func() {
|
|
c.closeRequestOps(ErrClientQuit)
|
|
conn.Close()
|
|
if reading {
|
|
// Empty read channels until read is dead.
|
|
for {
|
|
select {
|
|
case <-c.readResp:
|
|
case <-c.readErr:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
|
|
for {
|
|
select {
|
|
case <-c.close:
|
|
return
|
|
|
|
// Read path.
|
|
case batch := <-c.readResp:
|
|
for _, msg := range batch {
|
|
switch {
|
|
case msg.isNotification():
|
|
log.Trace("", "msg", log.Lazy{Fn: func() string {
|
|
return fmt.Sprint("<-readResp: notification ", msg)
|
|
}})
|
|
c.handleNotification(msg)
|
|
case msg.isResponse():
|
|
log.Trace("", "msg", log.Lazy{Fn: func() string {
|
|
return fmt.Sprint("<-readResp: response ", msg)
|
|
}})
|
|
c.handleResponse(msg)
|
|
default:
|
|
log.Debug("", "msg", log.Lazy{Fn: func() string {
|
|
return fmt.Sprint("<-readResp: dropping weird message", msg)
|
|
}})
|
|
// TODO: maybe close
|
|
}
|
|
}
|
|
|
|
case err := <-c.readErr:
|
|
log.Debug("<-readErr", "err", err)
|
|
c.closeRequestOps(err)
|
|
conn.Close()
|
|
reading = false
|
|
|
|
case newconn := <-c.reconnected:
|
|
log.Debug("<-reconnected", "reading", reading, "remote", conn.RemoteAddr())
|
|
if reading {
|
|
// Wait for the previous read loop to exit. This is a rare case.
|
|
conn.Close()
|
|
<-c.readErr
|
|
}
|
|
go c.read(newconn)
|
|
reading = true
|
|
conn = newconn
|
|
|
|
// Send path.
|
|
case op := <-requestOpLock:
|
|
// Stop listening for further send ops until the current one is done.
|
|
requestOpLock = nil
|
|
lastOp = op
|
|
for _, id := range op.ids {
|
|
c.respWait[string(id)] = op
|
|
}
|
|
|
|
case err := <-c.sendDone:
|
|
if err != nil {
|
|
// Remove response handlers for the last send. We remove those here
|
|
// because the error is already handled in Call or BatchCall. When the
|
|
// read loop goes down, it will signal all other current operations.
|
|
for _, id := range lastOp.ids {
|
|
delete(c.respWait, string(id))
|
|
}
|
|
}
|
|
// Listen for send ops again.
|
|
requestOpLock = c.requestOp
|
|
lastOp = nil
|
|
}
|
|
}
|
|
}
|
|
|
|
// closeRequestOps unblocks pending send ops and active subscriptions.
|
|
func (c *Client) closeRequestOps(err error) {
|
|
didClose := make(map[*requestOp]bool)
|
|
|
|
for id, op := range c.respWait {
|
|
// Remove the op so that later calls will not close op.resp again.
|
|
delete(c.respWait, id)
|
|
|
|
if !didClose[op] {
|
|
op.err = err
|
|
close(op.resp)
|
|
didClose[op] = true
|
|
}
|
|
}
|
|
for id, sub := range c.subs {
|
|
delete(c.subs, id)
|
|
sub.quitWithError(err, false)
|
|
}
|
|
}
|
|
|
|
func (c *Client) handleNotification(msg *jsonrpcMessage) {
|
|
if !strings.HasSuffix(msg.Method, notificationMethodSuffix) {
|
|
log.Debug("dropping non-subscription message", "msg", msg)
|
|
return
|
|
}
|
|
var subResult struct {
|
|
ID string `json:"subscription"`
|
|
Result json.RawMessage `json:"result"`
|
|
}
|
|
if err := json.Unmarshal(msg.Params, &subResult); err != nil {
|
|
log.Debug("dropping invalid subscription message", "msg", msg)
|
|
return
|
|
}
|
|
if c.subs[subResult.ID] != nil {
|
|
c.subs[subResult.ID].deliver(subResult.Result)
|
|
}
|
|
}
|
|
|
|
func (c *Client) handleResponse(msg *jsonrpcMessage) {
|
|
op := c.respWait[string(msg.ID)]
|
|
if op == nil {
|
|
log.Debug("unsolicited response", "msg", msg)
|
|
return
|
|
}
|
|
delete(c.respWait, string(msg.ID))
|
|
// For normal responses, just forward the reply to Call/BatchCall.
|
|
if op.sub == nil {
|
|
op.resp <- msg
|
|
return
|
|
}
|
|
// For subscription responses, start the subscription if the server
|
|
// indicates success. EthSubscribe gets unblocked in either case through
|
|
// the op.resp channel.
|
|
defer close(op.resp)
|
|
if msg.Error != nil {
|
|
op.err = msg.Error
|
|
return
|
|
}
|
|
if op.err = json.Unmarshal(msg.Result, &op.sub.subid); op.err == nil {
|
|
go op.sub.start()
|
|
c.subs[op.sub.subid] = op.sub
|
|
}
|
|
}
|
|
|
|
// Reading happens on a dedicated goroutine.
|
|
|
|
func (c *Client) read(conn net.Conn) error {
|
|
var (
|
|
buf json.RawMessage
|
|
dec = json.NewDecoder(conn)
|
|
)
|
|
readMessage := func() (rs []*jsonrpcMessage, err error) {
|
|
buf = buf[:0]
|
|
if err = dec.Decode(&buf); err != nil {
|
|
return nil, err
|
|
}
|
|
if isBatch(buf) {
|
|
err = json.Unmarshal(buf, &rs)
|
|
} else {
|
|
rs = make([]*jsonrpcMessage, 1)
|
|
err = json.Unmarshal(buf, &rs[0])
|
|
}
|
|
return rs, err
|
|
}
|
|
|
|
for {
|
|
resp, err := readMessage()
|
|
if err != nil {
|
|
c.readErr <- err
|
|
return err
|
|
}
|
|
c.readResp <- resp
|
|
}
|
|
}
|
|
|
|
// Subscriptions.
|
|
|
|
// A ClientSubscription represents a subscription established through EthSubscribe.
|
|
type ClientSubscription struct {
|
|
client *Client
|
|
etype reflect.Type
|
|
channel reflect.Value
|
|
namespace string
|
|
subid string
|
|
in chan json.RawMessage
|
|
|
|
quitOnce sync.Once // ensures quit is closed once
|
|
quit chan struct{} // quit is closed when the subscription exits
|
|
errOnce sync.Once // ensures err is closed once
|
|
err chan error
|
|
}
|
|
|
|
func newClientSubscription(c *Client, namespace string, channel reflect.Value) *ClientSubscription {
|
|
sub := &ClientSubscription{
|
|
client: c,
|
|
namespace: namespace,
|
|
etype: channel.Type().Elem(),
|
|
channel: channel,
|
|
quit: make(chan struct{}),
|
|
err: make(chan error, 1),
|
|
in: make(chan json.RawMessage),
|
|
}
|
|
return sub
|
|
}
|
|
|
|
// Err returns the subscription error channel. The intended use of Err is to schedule
|
|
// resubscription when the client connection is closed unexpectedly.
|
|
//
|
|
// The error channel receives a value when the subscription has ended due
|
|
// to an error. The received error is nil if Close has been called
|
|
// on the underlying client and no other error has occurred.
|
|
//
|
|
// The error channel is closed when Unsubscribe is called on the subscription.
|
|
func (sub *ClientSubscription) Err() <-chan error {
|
|
return sub.err
|
|
}
|
|
|
|
// Unsubscribe unsubscribes the notification and closes the error channel.
|
|
// It can safely be called more than once.
|
|
func (sub *ClientSubscription) Unsubscribe() {
|
|
sub.quitWithError(nil, true)
|
|
sub.errOnce.Do(func() { close(sub.err) })
|
|
}
|
|
|
|
func (sub *ClientSubscription) quitWithError(err error, unsubscribeServer bool) {
|
|
sub.quitOnce.Do(func() {
|
|
// The dispatch loop won't be able to execute the unsubscribe call
|
|
// if it is blocked on deliver. Close sub.quit first because it
|
|
// unblocks deliver.
|
|
close(sub.quit)
|
|
if unsubscribeServer {
|
|
sub.requestUnsubscribe()
|
|
}
|
|
if err != nil {
|
|
if err == ErrClientQuit {
|
|
err = nil // Adhere to subscription semantics.
|
|
}
|
|
sub.err <- err
|
|
}
|
|
})
|
|
}
|
|
|
|
func (sub *ClientSubscription) deliver(result json.RawMessage) (ok bool) {
|
|
select {
|
|
case sub.in <- result:
|
|
return true
|
|
case <-sub.quit:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (sub *ClientSubscription) start() {
|
|
sub.quitWithError(sub.forward())
|
|
}
|
|
|
|
func (sub *ClientSubscription) forward() (err error, unsubscribeServer bool) {
|
|
cases := []reflect.SelectCase{
|
|
{Dir: reflect.SelectRecv, Chan: reflect.ValueOf(sub.quit)},
|
|
{Dir: reflect.SelectRecv, Chan: reflect.ValueOf(sub.in)},
|
|
{Dir: reflect.SelectSend, Chan: sub.channel},
|
|
}
|
|
buffer := list.New()
|
|
defer buffer.Init()
|
|
for {
|
|
var chosen int
|
|
var recv reflect.Value
|
|
if buffer.Len() == 0 {
|
|
// Idle, omit send case.
|
|
chosen, recv, _ = reflect.Select(cases[:2])
|
|
} else {
|
|
// Non-empty buffer, send the first queued item.
|
|
cases[2].Send = reflect.ValueOf(buffer.Front().Value)
|
|
chosen, recv, _ = reflect.Select(cases)
|
|
}
|
|
|
|
switch chosen {
|
|
case 0: // <-sub.quit
|
|
return nil, false
|
|
case 1: // <-sub.in
|
|
val, err := sub.unmarshal(recv.Interface().(json.RawMessage))
|
|
if err != nil {
|
|
return err, true
|
|
}
|
|
if buffer.Len() == maxClientSubscriptionBuffer {
|
|
return ErrSubscriptionQueueOverflow, true
|
|
}
|
|
buffer.PushBack(val)
|
|
case 2: // sub.channel<-
|
|
cases[2].Send = reflect.Value{} // Don't hold onto the value.
|
|
buffer.Remove(buffer.Front())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (sub *ClientSubscription) unmarshal(result json.RawMessage) (interface{}, error) {
|
|
val := reflect.New(sub.etype)
|
|
err := json.Unmarshal(result, val.Interface())
|
|
return val.Elem().Interface(), err
|
|
}
|
|
|
|
func (sub *ClientSubscription) requestUnsubscribe() error {
|
|
var result interface{}
|
|
return sub.client.Call(&result, sub.namespace+unsubscribeMethodSuffix, sub.subid)
|
|
}
|