You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.3 KiB
45 lines
1.3 KiB
package blockproposal
|
|
|
|
import (
|
|
msg_pb "github.com/harmony-one/harmony/api/proto/message"
|
|
"github.com/harmony-one/harmony/consensus"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
)
|
|
|
|
// Service is a block proposal service.
|
|
type Service struct {
|
|
stopChan chan struct{}
|
|
stoppedChan chan struct{}
|
|
c *consensus.Consensus
|
|
messageChan chan *msg_pb.Message
|
|
waitForConsensusReady func(c *consensus.Consensus, stopChan chan struct{}, stoppedChan chan struct{})
|
|
}
|
|
|
|
// New returns a block proposal service.
|
|
func New(c *consensus.Consensus, waitForConsensusReady func(c *consensus.Consensus, stopChan chan struct{}, stoppedChan chan struct{})) *Service {
|
|
return &Service{
|
|
c: c,
|
|
waitForConsensusReady: waitForConsensusReady,
|
|
stopChan: make(chan struct{}),
|
|
stoppedChan: make(chan struct{}),
|
|
}
|
|
}
|
|
|
|
// Start starts block proposal service.
|
|
func (s *Service) Start() error {
|
|
s.run()
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) run() {
|
|
s.waitForConsensusReady(s.c, s.stopChan, s.stoppedChan)
|
|
}
|
|
|
|
// Stop stops block proposal service.
|
|
func (s *Service) Stop() error {
|
|
utils.Logger().Info().Msg("Stopping block proposal service.")
|
|
s.stopChan <- struct{}{}
|
|
<-s.stoppedChan
|
|
utils.Logger().Info().Msg("Role conversion stopped.")
|
|
return nil
|
|
}
|
|
|