The consumer interface is changed to a minimum subset

Signed-off-by: axfor <aixiaoxiang2009@hotmail.com>
axfor · Mar 22, 2023 · e74b66b · e74b66b
1 parent f8863b5
commit e74b66b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pkg/kafka/consumer/config.go b/pkg/kafka/consumer/config.go
@@ -15,6 +15,7 @@
 package consumer
 
 import (
+	"context"
 	"fmt"
 
 	"github.com/Shopify/sarama"
@@ -25,7 +26,9 @@ import (
 
 // Consumer is an interface to features of Sarama that are necessary for the consumer
 type Consumer interface {
-	sarama.ConsumerGroup
+	Consume(ctx context.Context, topics []string, handler sarama.ConsumerGroupHandler) error
+	Errors() <-chan error
+	Close() error
 }
 
 // Builder builds a new kafka consumer