diff --git a/pkg/consensus/consensus.go b/pkg/consensus/consensus.go index 4a17d52fc..c221c98b6 100644 --- a/pkg/consensus/consensus.go +++ b/pkg/consensus/consensus.go @@ -148,7 +148,7 @@ func NewService(cfg Config) (Service, error) { dbft.WithNewChangeView(func() payload.ChangeView { return new(changeView) }), dbft.WithNewCommit(srv.newCommit), dbft.WithNewRecoveryRequest(func() payload.RecoveryRequest { return new(recoveryRequest) }), - dbft.WithNewRecoveryMessage(func() payload.RecoveryMessage { return new(recoveryMessage) }), + dbft.WithNewRecoveryMessage(srv.newRecoveryMessage), dbft.WithVerifyPrepareRequest(srv.verifyRequest), ) @@ -267,6 +267,10 @@ func (s *service) newCommit() payload.Commit { return c } +func (s *service) newRecoveryMessage() payload.RecoveryMessage { + return &recoveryMessage{stateRootEnabled: s.stateRootEnabled()} +} + func (s *service) validatePayload(p *Payload) bool { validators := s.getValidators() if int(p.validatorIndex) >= len(validators) {