Zxd.Core/code/DG.Kafka.Worker/KafkaWorkerManager.cs

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace DG.Kafka.Worker
{
    internal class KafkaWorkerManager : IKafkaWorkerManager
    {
        private TaskConfig _taskConfig;
        private List<Consumer> _consumers;
        private readonly ILogger<KafkaWorkerManager> _logger;
        private readonly IServiceProvider _serviceProvider;

        public KafkaWorkerManager(ILogger<KafkaWorkerManager> logger, IServiceProvider serviceProvider)
        {
            _logger = logger;
            // 读取任务配置
            _taskConfig = KafkaClient.Default.ConfigurationManager.GetSection("TaskConfig").Get<TaskConfig>();
            _consumers = KafkaClient.Default.ConfigurationManager.GetSection("Consumers").Get<List<Consumer>>();
            if (_taskConfig == null)
            {
                throw new ArgumentNullException(nameof(_taskConfig));
            }
            if (_consumers == null)
            {
                throw new ArgumentNullException(nameof(_consumers));
            }
            _serviceProvider = serviceProvider;
        }

        public async Task RegisterWorker<TWorker, T>(string? topic) 
            where TWorker : KafkaWorkerBase<T>
        {
            var t = _serviceProvider.GetRequiredService<TWorker>();
            _logger.LogDebug($"注册worker: {typeof(TWorker).Name}");
            var consumer = _consumers.FirstOrDefault(c => c.Topic == topic);
            if (consumer == null)
            {
                throw new ArgumentNullException(nameof(consumer));
            }
            await KafkaClient.Builder<T>(consumer, t.Start, t.ShopAsync);
        }

        public async Task RegisterBatchWorker<TWorker, T>(string? topic, int batchsize = 1000) 
            where TWorker : BatchKafkaWorkerBase<T>
        {
            var t = _serviceProvider.GetRequiredService<TWorker>();
            _logger.LogDebug($"注册worker: {typeof(TWorker).Name}");
            var consumer = _consumers.FirstOrDefault(c => c.Topic == topic);
            if (consumer == null)
            {
                throw new ArgumentNullException(nameof(consumer));
            }
            await KafkaClient.BatchBuilder<T>(consumer, t.Start, t.ShopAsync, batchsize);
        }
    }
}