Skip to content

Commit

Permalink
feat: batch producer implemented to konsumer (#16) (#24)
Browse files Browse the repository at this point in the history
* feat: batch producer implemented to konsumer (#16)

* feat: batch producer example added (#16)
  • Loading branch information
ugurcanerdogan committed Aug 24, 2023
1 parent 013b756 commit 04af6d0
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 1 deletion.
17 changes: 16 additions & 1 deletion examples/with-kafka-producer/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,24 @@ func main() {
},
})

const topicName = "standart-topic"

_ = producer.Produce(context.Background(), kafka.Message{
Topic: "standart-topic",
Topic: topicName,
Key: []byte("1"),
Value: []byte(`{ "foo": "bar" }`),
})

_ = producer.ProduceBatch(context.Background(), []kafka.Message{
{
Topic: topicName,
Key: []byte("1"),
Value: []byte(`{ "foo": "bar" }`),
},
{
Topic: topicName,
Key: []byte("2"),
Value: []byte(`{ "foo2": "bar2" }`),
},
})
}
10 changes: 10 additions & 0 deletions producer.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@ import (

type Producer interface {
Produce(ctx context.Context, message Message) error
ProduceBatch(ctx context.Context, messages []Message) error
Close() error
}

Expand All @@ -28,6 +29,15 @@ func (c *producer) Produce(ctx context.Context, message Message) error {
return c.w.WriteMessages(ctx, kafka.Message(message))
}

func (c *producer) ProduceBatch(ctx context.Context, messages []Message) error {
kafkaMessages := make([]kafka.Message, 0, len(messages))
for i := range messages {
convertedMessage := kafka.Message(messages[i])
kafkaMessages = append(kafkaMessages, convertedMessage)
}
return c.w.WriteMessages(ctx, kafkaMessages...)
}

func (c *producer) Close() error {
return c.w.Close()
}

0 comments on commit 04af6d0

Please sign in to comment.