fix tier2 max concurrent request configuration

streamingfast · Mar 12, 2024 · 152e2bd · 152e2bd
1 parent 4999381
commit 152e2bd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/app/tier2.go b/app/tier2.go
@@ -88,6 +88,9 @@ func (a *Tier2App) Run() error {
 		opts = append(opts, service.WithModuleExecutionTracing())
 	}
 
+	if a.config.MaximumConcurrentRequests > 0 {
+		opts = append(opts, service.WithMaxConcurrentRequests(a.config.MaximumConcurrentRequests))
+	}
 	opts = append(opts, service.WithReadinessFunc(a.setReadiness))
 
 	svc, err := service.NewTier2(

diff --git a/service/options.go b/service/options.go
@@ -56,6 +56,18 @@ func WithModuleExecutionTracing() Option {
 	}
 }
 
+func WithMaxConcurrentRequests(max uint64) Option {
+	return func(a anyTierService) {
+		switch s := a.(type) {
+		case *Tier1Service:
+			// not used
+		case *Tier2Service:
+			s.runtimeConfig.MaxConcurrentRequests = int64(max)
+		}
+	}
+
+}
+
 func WithReadinessFunc(f func(bool)) Option {
 	return func(a anyTierService) {
 		switch s := a.(type) {