Add options for attestation and warp sync requester

iotaledger · Dec 5, 2023 · f3d8ad0 · f3d8ad0
1 parent a5a57ba
commit f3d8ad0
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/pkg/protocol/options.go b/pkg/protocol/options.go
@@ -61,7 +61,9 @@ type Options struct {
 	// StorageOptions contains the options for the Storage.
 	StorageOptions []options.Option[storage.Storage]
 
-	CommitmentRequesterOptions []options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]
+	CommitmentRequesterOptions  []options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]
+	AttestationRequesterOptions []options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]
+	WarpSyncRequesterOptions    []options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]
 
 	// PreSolidFilterProvider contains the provider for the PreSolidFilter engine modules.
 	PreSolidFilterProvider module.Provider[*engine.Engine, presolidfilter.PreSolidFilter]
@@ -287,3 +289,21 @@ func WithStorageOptions(opts ...options.Option[storage.Storage]) options.Option[
 		p.Options.StorageOptions = append(p.Options.StorageOptions, opts...)
 	}
 }
+
+func WithCommitmentRequesterOptions(opts ...options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]) options.Option[Protocol] {
+	return func(p *Protocol) {
+		p.Options.CommitmentRequesterOptions = append(p.Options.CommitmentRequesterOptions, opts...)
+	}
+}
+
+func WithAttestationRequesterOptions(opts ...options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]) options.Option[Protocol] {
+	return func(p *Protocol) {
+		p.Options.AttestationRequesterOptions = append(p.Options.AttestationRequesterOptions, opts...)
+	}
+}
+
+func WithWarpSyncRequesterOptions(opts ...options.Option[eventticker.EventTicker[iotago.SlotIndex, iotago.CommitmentID]]) options.Option[Protocol] {
+	return func(p *Protocol) {
+		p.Options.WarpSyncRequesterOptions = append(p.Options.WarpSyncRequesterOptions, opts...)
+	}
+}
diff --git a/pkg/protocol/protocol_attestations.go b/pkg/protocol/protocol_attestations.go
@@ -40,7 +40,7 @@ func newAttestationsProtocol(protocol *Protocol) *AttestationsProtocol {
 		Logger:              lo.Return1(protocol.Logger.NewChildLogger("Attestations")),
 		protocol:            protocol,
 		workerPool:          protocol.Workers.CreatePool("Attestations"),
-		ticker:              eventticker.New[iotago.SlotIndex, iotago.CommitmentID](),
+		ticker:              eventticker.New[iotago.SlotIndex, iotago.CommitmentID](protocol.Options.AttestationRequesterOptions...),
 		commitmentVerifiers: shrinkingmap.New[iotago.CommitmentID, *CommitmentVerifier](),
 	}
 

diff --git a/pkg/protocol/protocol_warp_sync.go b/pkg/protocol/protocol_warp_sync.go
@@ -40,7 +40,7 @@ func newWarpSyncProtocol(protocol *Protocol) *WarpSyncProtocol {
 		Logger:     lo.Return1(protocol.Logger.NewChildLogger("WarpSync")),
 		protocol:   protocol,
 		workerPool: protocol.Workers.CreatePool("WarpSync", workerpool.WithWorkerCount(1)),
-		ticker:     eventticker.New[iotago.SlotIndex, iotago.CommitmentID](),
+		ticker:     eventticker.New[iotago.SlotIndex, iotago.CommitmentID](protocol.Options.WarpSyncRequesterOptions...),
 	}
 
 	c.ticker.Events.Tick.Hook(c.SendRequest)