package kafkaclient

import (
	"context"
	"time"

	"github.com/segmentio/kafka-go"
)

func Write(ctx context.Context, writer *kafka.Writer, message kafka.Message) error {
	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
	defer cancel()

	err := writer.WriteMessages(ctx, message)
	return err
}