mirror of https://github.com/poanetwork/gecko.git
35 lines
781 B
Go
35 lines
781 B
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/proto"
|
||
|
)
|
||
|
|
||
|
// Client is an implementation of a messenger channel that talks over RPC.
|
||
|
type Client struct{ client proto.WriterClient }
|
||
|
|
||
|
// NewClient returns a database instance connected to a remote database instance
|
||
|
func NewClient(client proto.WriterClient) *Client {
|
||
|
return &Client{client: client}
|
||
|
}
|
||
|
|
||
|
// Write ...
|
||
|
func (c *Client) Write(p []byte) (int, error) {
|
||
|
resp, err := c.client.Write(context.Background(), &proto.WriteRequest{
|
||
|
Payload: p,
|
||
|
})
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
|
||
|
if resp.Errored {
|
||
|
err = errors.New(resp.Error)
|
||
|
}
|
||
|
return int(resp.Written), err
|
||
|
}
|