67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
|
package modules
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
rpc "github.com/libp2p/go-libp2p-gorpc"
|
||
|
|
"github.com/libp2p/go-libp2p/core/host"
|
||
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
||
|
|
"github.com/libp2p/go-libp2p/core/protocol"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/lotus/api"
|
||
|
|
"github.com/filecoin-project/lotus/chain/types"
|
||
|
|
consensus "github.com/filecoin-project/lotus/lib/consensus/raft"
|
||
|
|
"github.com/filecoin-project/lotus/node/impl/full"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RPCHandler struct {
|
||
|
|
mpoolAPI full.MpoolAPI
|
||
|
|
cons *consensus.Consensus
|
||
|
|
}
|
||
|
|
|
||
|
|
//type ConsensusRPCAPI struct {
|
||
|
|
// cons *consensus.Consensus
|
||
|
|
// rpcHandler *RPCHandler
|
||
|
|
//}
|
||
|
|
|
||
|
|
func NewRPCHandler(mpoolAPI full.MpoolAPI, cons *consensus.Consensus) *RPCHandler {
|
||
|
|
return &RPCHandler{mpoolAPI, cons}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *RPCHandler) MpoolPushMessage(ctx context.Context, msgWhole *api.MpoolMessageWhole, ret *types.SignedMessage) error {
|
||
|
|
signedMsg, err := h.mpoolAPI.MpoolPushMessage(ctx, msgWhole.Msg, msgWhole.Spec)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
*ret = *signedMsg
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *RPCHandler) AddPeer(ctx context.Context, pid peer.ID, ret *struct{}) error {
|
||
|
|
return h.cons.AddPeer(ctx, pid)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add other consensus RPC calls here
|
||
|
|
|
||
|
|
func NewRPCClient(host host.Host) *rpc.Client {
|
||
|
|
protocolID := protocol.ID("/p2p/rpc/ping")
|
||
|
|
return rpc.NewClient(host, protocolID)
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRPCServer(host host.Host, rpcHandler *RPCHandler) error {
|
||
|
|
protocolID := protocol.ID("/p2p/rpc/ping")
|
||
|
|
rpcServer := rpc.NewServer(host, protocolID)
|
||
|
|
return rpcServer.RegisterName("Consensus", rpcHandler)
|
||
|
|
//return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// contructorsfor rpc client and rpc server
|
||
|
|
// rpc handler
|
||
|
|
|
||
|
|
// rpcClient
|
||
|
|
// Consensus
|
||
|
|
// MessageSigner
|
||
|
|
// MpoolAPI
|
||
|
|
// RPC handler
|
||
|
|
// RPC server
|