JiShe.CollectBus/src/JiShe.CollectBus.KafkaProducer/Consumer/ConsumerService.cs

using Confluent.Kafka;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;
using JiShe.CollectBus.Kafka.Attributes;
using Volo.Abp.DependencyInjection;
using JiShe.CollectBus.Kafka.AdminClient;

namespace JiShe.CollectBus.Kafka.Consumer
{
    public abstract class ConsumerService : IConsumerService, IDisposable, ISingletonDependency
    {
        private readonly ILogger<ConsumerService> _logger;
        private readonly IConsumer<string, string> _consumer;

        public ConsumerService(
            ILogger<ConsumerService> logger,
            IConfiguration configuration)
        {
            _logger = logger;

            var consumerConfig = configuration.GetSection("Kafka:ConsumerConfig")
                .Get<ConsumerConfig>();

            _consumer = new ConsumerBuilder<string, string>(consumerConfig)
                .SetErrorHandler(OnConsumeError)
                .Build();
        }

        public void Subscribe(string topic)
        {
            _consumer.Subscribe(topic);
            _logger.LogInformation($"Subscribed to topic: {topic}");
        }

        public ConsumeResult<string, string> Consume(CancellationToken cancellationToken)
        {
            return _consumer.Consume(cancellationToken);
        }

        public void Commit(ConsumeResult<string, string> result)
        {
            _consumer.Commit(result);
            _logger.LogDebug($"Committed offset: {result.TopicPartitionOffset}");
        }

        private void OnConsumeError(IConsumer<string, string> consumer, Error error)
        {
            _logger.LogError($"Kafka consumer error: {error.Reason}");
        }

        public void Dispose()
        {
            _consumer?.Close();
            _consumer?.Dispose();
        }
    }
}