Add method for handle multi topic in single method

golangid · Nov 13, 2024 · 868cd96 · 868cd96
1 parent 685f6bf
commit 868cd96
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/codebase/factory/types/worker_type.go b/codebase/factory/types/worker_type.go
@@ -26,6 +26,16 @@ type WorkerHandlerGroup struct {
 	Handlers []WorkerHandler
 }
 
+// AddMultiRoute method from WorkerHandlerGroup to handle multi topic in single method
+func (m *WorkerHandlerGroup) AddMultiRoute(patternRoutes []string, mainHandlerFunc WorkerHandlerFunc, opts ...WorkerHandlerOptionFunc) {
+	for _, route := range patternRoutes {
+		if len(route) == 0 {
+			continue
+		}
+		m.Add(route, mainHandlerFunc, opts...)
+	}
+}
+
 // Add method from WorkerHandlerGroup, patternRoute can contains unique topic name, key, or task name
 func (m *WorkerHandlerGroup) Add(patternRoute string, mainHandlerFunc WorkerHandlerFunc, opts ...WorkerHandlerOptionFunc) {
 	h := WorkerHandler{