Ignore messages from other channels

This will improve performance and also make it easier to debug.
igolaizola · Apr 24, 2023 · ea12bb4 · ea12bb4
1 parent 727df57
commit ea12bb4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pkg/ai/bluewillow/bluewillow.go b/pkg/ai/bluewillow/bluewillow.go
@@ -78,6 +78,10 @@ func New(client *discord.Client, channelID string, debug bool) (ai.Client, error
 			if err := json.Unmarshal(e.RawData, &msg); err != nil {
 				log.Println("bluewillow: couldn't unmarshal message: %w", err)
 			}
+			// Ignore messages from other channels
+			if msg.ChannelID != c.channelID {
+				return
+			}
 			c.debugLog(e.Type, e.RawData)
 
 			var key search

diff --git a/pkg/ai/midjourney/midjourney.go b/pkg/ai/midjourney/midjourney.go
@@ -82,6 +82,10 @@ func New(client *discord.Client, channelID string, debug bool) (ai.Client, error
 			if err := json.Unmarshal(e.RawData, &msg); err != nil {
 				log.Println("midjourney: couldn't unmarshal message: %w", err)
 			}
+			// Ignore messages from other channels
+			if msg.ChannelID != c.channelID {
+				return
+			}
 			c.debugLog(e.Type, e.RawData)
 
 			var key search