|
@@ -2,25 +2,9 @@ package service
|
|
|
|
|
|
import (
|
|
import (
|
|
"github.com/confluentinc/confluent-kafka-go/v2/kafka"
|
|
"github.com/confluentinc/confluent-kafka-go/v2/kafka"
|
|
- "github.com/prometheus/client_golang/prometheus"
|
|
|
|
- "github.com/prometheus/client_golang/prometheus/promauto"
|
|
|
|
)
|
|
)
|
|
|
|
|
|
type ConsumerHandler func(*kafka.Message) error
|
|
type ConsumerHandler func(*kafka.Message) error
|
|
-type ConsumerOptions struct {
|
|
|
|
- Counter prometheus.Counter
|
|
|
|
-}
|
|
|
|
-
|
|
|
|
-func ConsumerOptionsWithCounter() ConsumerOptions {
|
|
|
|
- return ConsumerOptions{
|
|
|
|
- Counter: promauto.NewCounter(prometheus.CounterOpts{
|
|
|
|
- Namespace: "kafka",
|
|
|
|
- Subsystem: "consumer",
|
|
|
|
- Name: "received",
|
|
|
|
- Help: "The total number of received messages",
|
|
|
|
- }),
|
|
|
|
- }
|
|
|
|
-}
|
|
|
|
|
|
|
|
type Consumer interface {
|
|
type Consumer interface {
|
|
Service
|
|
Service
|
|
@@ -28,7 +12,7 @@ type Consumer interface {
|
|
SetPause(bool) error
|
|
SetPause(bool) error
|
|
Paused() bool
|
|
Paused() bool
|
|
|
|
|
|
- Subscribe([]string, chan *kafka.Message, ConsumerOptions) error
|
|
|
|
|
|
+ Subscribe([]string, chan *kafka.Message) error
|
|
Subscribed() bool
|
|
Subscribed() bool
|
|
|
|
|
|
RegisterHandlers(...ConsumerHandler)
|
|
RegisterHandlers(...ConsumerHandler)
|