mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-03 17:44:29 +00:00
119 lines
4.0 KiB
Go
119 lines
4.0 KiB
Go
package direct
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces/remote"
|
|
"github.com/ledgerwatch/erigon-lib/gointerfaces/types"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/protobuf/types/known/emptypb"
|
|
)
|
|
|
|
type EthBackendClientDirect struct {
|
|
server remote.ETHBACKENDServer
|
|
}
|
|
|
|
func NewEthBackendClientDirect(server remote.ETHBACKENDServer) *EthBackendClientDirect {
|
|
return &EthBackendClientDirect{server: server}
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) Etherbase(ctx context.Context, in *remote.EtherbaseRequest, opts ...grpc.CallOption) (*remote.EtherbaseReply, error) {
|
|
return s.server.Etherbase(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) NetVersion(ctx context.Context, in *remote.NetVersionRequest, opts ...grpc.CallOption) (*remote.NetVersionReply, error) {
|
|
return s.server.NetVersion(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) NetPeerCount(ctx context.Context, in *remote.NetPeerCountRequest, opts ...grpc.CallOption) (*remote.NetPeerCountReply, error) {
|
|
return s.server.NetPeerCount(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) EngineGetPayloadV1(ctx context.Context, in *remote.EngineGetPayloadRequest, opts ...grpc.CallOption) (*types.ExecutionPayload, error) {
|
|
return s.server.EngineGetPayloadV1(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) EngineNewPayloadV1(ctx context.Context, in *types.ExecutionPayload, opts ...grpc.CallOption) (*remote.EnginePayloadStatus, error) {
|
|
return s.server.EngineNewPayloadV1(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) EngineForkChoiceUpdatedV1(ctx context.Context, in *remote.EngineForkChoiceUpdatedRequest, opts ...grpc.CallOption) (*remote.EngineForkChoiceUpdatedReply, error) {
|
|
return s.server.EngineForkChoiceUpdatedV1(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) Version(ctx context.Context, in *emptypb.Empty, opts ...grpc.CallOption) (*types.VersionReply, error) {
|
|
return s.server.Version(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) ProtocolVersion(ctx context.Context, in *remote.ProtocolVersionRequest, opts ...grpc.CallOption) (*remote.ProtocolVersionReply, error) {
|
|
return s.server.ProtocolVersion(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) ClientVersion(ctx context.Context, in *remote.ClientVersionRequest, opts ...grpc.CallOption) (*remote.ClientVersionReply, error) {
|
|
return s.server.ClientVersion(ctx, in)
|
|
}
|
|
|
|
// -- start Subscribe
|
|
|
|
func (s *EthBackendClientDirect) Subscribe(ctx context.Context, in *remote.SubscribeRequest, opts ...grpc.CallOption) (remote.ETHBACKEND_SubscribeClient, error) {
|
|
ch := make(chan *subscribeReply, 16384)
|
|
streamServer := &SubscribeStreamS{ch: ch, ctx: ctx}
|
|
go func() {
|
|
defer close(ch)
|
|
streamServer.Err(s.server.Subscribe(in, streamServer))
|
|
}()
|
|
return &SubscribeStreamC{ch: ch, ctx: ctx}, nil
|
|
}
|
|
|
|
type subscribeReply struct {
|
|
r *remote.SubscribeReply
|
|
err error
|
|
}
|
|
type SubscribeStreamS struct {
|
|
ch chan *subscribeReply
|
|
ctx context.Context
|
|
grpc.ServerStream
|
|
}
|
|
|
|
func (s *SubscribeStreamS) Send(m *remote.SubscribeReply) error {
|
|
s.ch <- &subscribeReply{r: m}
|
|
return nil
|
|
}
|
|
func (s *SubscribeStreamS) Context() context.Context { return s.ctx }
|
|
func (s *SubscribeStreamS) Err(err error) {
|
|
if err == nil {
|
|
return
|
|
}
|
|
s.ch <- &subscribeReply{err: err}
|
|
}
|
|
|
|
type SubscribeStreamC struct {
|
|
ch chan *subscribeReply
|
|
ctx context.Context
|
|
grpc.ClientStream
|
|
}
|
|
|
|
func (c *SubscribeStreamC) Recv() (*remote.SubscribeReply, error) {
|
|
m, ok := <-c.ch
|
|
if !ok || m == nil {
|
|
return nil, io.EOF
|
|
}
|
|
return m.r, m.err
|
|
}
|
|
func (c *SubscribeStreamC) Context() context.Context { return c.ctx }
|
|
|
|
// -- end Subscribe
|
|
|
|
func (s *EthBackendClientDirect) Block(ctx context.Context, in *remote.BlockRequest, opts ...grpc.CallOption) (*remote.BlockReply, error) {
|
|
return s.server.Block(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) TxnLookup(ctx context.Context, in *remote.TxnLookupRequest, opts ...grpc.CallOption) (*remote.TxnLookupReply, error) {
|
|
return s.server.TxnLookup(ctx, in)
|
|
}
|
|
|
|
func (s *EthBackendClientDirect) NodeInfo(ctx context.Context, in *remote.NodesInfoRequest, opts ...grpc.CallOption) (*remote.NodesInfoReply, error) {
|
|
return s.server.NodeInfo(ctx, in)
|
|
}
|