Feat: cleanup passthrough scheduler

iotaledger · Mar 12, 2024 · 70e0272 · 70e0272
1 parent b1c8ec5
commit 70e0272
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 14 deletions.
diff --git a/pkg/protocol/engine/clock/blocktime/clock.go b/pkg/protocol/engine/clock/blocktime/clock.go
@@ -52,7 +52,6 @@ func NewProvider(opts ...options.Option[Clock]) module.Provider[*engine.Engine,
 				e.Events.Clock.ConfirmedTimeUpdated.LinkTo(c.confirmedTime.OnUpdated)
 
 				asyncOpt := event.WithWorkerPool(c.workerPool)
-
 				c.ShutdownEvent().OnTrigger(lo.Batch(
 					e.Events.BlockGadget.BlockAccepted.Hook(func(block *blocks.Block) {
 						c.acceptedTime.Advance(block.IssuingTime())

diff --git a/pkg/protocol/engine/congestioncontrol/scheduler/passthrough/scheduler.go b/pkg/protocol/engine/congestioncontrol/scheduler/passthrough/scheduler.go
@@ -30,21 +30,11 @@ func NewProvider() module.Provider[*engine.Engine, scheduler.Scheduler] {
 	})
 }
 
-func New(module module.Module) *Scheduler {
-	s := &Scheduler{
-		Module: module,
+func New(subModule module.Module) *Scheduler {
+	return module.InitSimpleLifecycle(&Scheduler{
+		Module: subModule,
 		events: scheduler.NewEvents(),
-	}
-
-	s.ConstructedEvent().Trigger()
-
-	s.ShutdownEvent().OnTrigger(func() {
-		s.StoppedEvent().Trigger()
 	})
-
-	s.InitializedEvent().Trigger()
-
-	return s
 }
 
 func (s *Scheduler) IsBlockIssuerReady(_ iotago.AccountID, _ ...iotago.WorkScore) bool {