gecko/vms/rpcchainvm/ghttp/gwriter/writer_client.go

// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package gwriter

import (
	"context"
	"errors"

	"github.com/ava-labs/gecko/vms/rpcchainvm/ghttp/gwriter/gwriterproto"
)

// Client is an implementation of a messenger channel that talks over RPC.
type Client struct{ client gwriterproto.WriterClient }

// NewClient returns a database instance connected to a remote database instance
func NewClient(client gwriterproto.WriterClient) *Client {
	return &Client{client: client}
}

// Write ...
func (c *Client) Write(p []byte) (int, error) {
	resp, err := c.client.Write(context.Background(), &gwriterproto.WriteRequest{
		Payload: p,
	})
	if err != nil {
		return 0, err
	}

	if resp.Errored {
		err = errors.New(resp.Error)
	}
	return int(resp.Written), err
}