46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
|
package full
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"github.com/libp2p/go-libp2p/core/peer"
|
||
|
|
"go.uber.org/fx"
|
||
|
|
"golang.org/x/xerrors"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/lotus/chain/messagesigner"
|
||
|
|
consensus "github.com/filecoin-project/lotus/lib/consensus/raft"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RaftAPI struct {
|
||
|
|
fx.In
|
||
|
|
|
||
|
|
MessageSigner *messagesigner.MessageSignerConsensus `optional:"true"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *RaftAPI) GetRaftState(ctx context.Context) (*consensus.RaftState, error) {
|
||
|
|
if r.MessageSigner == nil {
|
||
|
|
return nil, xerrors.Errorf("Raft consensus not enabled. Please check your configuration")
|
||
|
|
}
|
||
|
|
return r.MessageSigner.GetRaftState(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *RaftAPI) Leader(ctx context.Context) (peer.ID, error) {
|
||
|
|
if r.MessageSigner == nil {
|
||
|
|
return "", xerrors.Errorf("Raft consensus not enabled. Please check your configuration")
|
||
|
|
}
|
||
|
|
return r.MessageSigner.Leader(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *RaftAPI) IsLeader(ctx context.Context) bool {
|
||
|
|
if r.MessageSigner == nil {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return r.MessageSigner.IsLeader(ctx)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *RaftAPI) RedirectToLeader(ctx context.Context, method string, arg interface{}, ret interface{}) (bool, error) {
|
||
|
|
if r.MessageSigner == nil {
|
||
|
|
return false, xerrors.Errorf("Raft consensus not enabled. Please check your configuration")
|
||
|
|
}
|
||
|
|
return r.MessageSigner.RedirectToLeader(ctx, method, arg, ret)
|
||
|
|
}
|