chore: use different consumer for notification_send

flanksource · Jul 27, 2023 · 16ba49e · 16ba49e
1 parent 6045e54
commit 16ba49e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/events/events.go b/events/events.go
@@ -52,6 +52,7 @@ func StartConsumers(gormDB *gorm.DB, config Config) {
 	allConsumers := []EventConsumer{
 		NewTeamConsumer(gormDB),
 		NewNotificationConsumer(gormDB),
+		NewNotificationSendConsumer(gormDB),
 		NewResponderConsumer(gormDB),
 	}
 	if config.UpstreamPush.Valid() {

diff --git a/events/notifications.go b/events/notifications.go
@@ -34,6 +34,16 @@ func NewNotificationConsumer(db *gorm.DB) EventConsumer {
 	}
 }
 
+func NewNotificationSendConsumer(db *gorm.DB) EventConsumer {
+	return EventConsumer{
+		WatchEvents:      []string{EventNotificationSend},
+		ProcessBatchFunc: processNotificationEvents,
+		BatchSize:        1,
+		Consumers:        5,
+		DB:               db,
+	}
+}
+
 func processNotificationEvents(ctx *api.Context, events []api.Event) []*api.Event {
 	var failedEvents []*api.Event
 	for _, e := range events {