Kingo.XxlJob.Core/src/DotXxlJob.Core/TaskExecutorFactory.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Microsoft.Extensions.DependencyInjection;

namespace DotXxlJob.Core
{
    /// <summary>
    /// 负责响应RPC请求，调度任务执行器的工厂类
    /// </summary>
    public class TaskExecutorFactory
    {
        private readonly IServiceProvider _provider;

        private readonly Dictionary<string, ITaskExecutor> _cache = new Dictionary<string, ITaskExecutor>();
        public TaskExecutorFactory(IServiceProvider provider)
        {
            this._provider = provider;
            Initialize();
        }

        private void Initialize()
        {
            var executors =  this._provider.GetServices<ITaskExecutor>();

            var taskExecutors = executors as ITaskExecutor[] ?? executors.ToArray();
            if (executors == null || !taskExecutors.Any()) return;
            
            foreach (var item in taskExecutors)
            {
                this._cache.Add(item.GlueType,item);
            }
        }

        public ITaskExecutor GetTaskExecutor(string glueType)
        {
            return this._cache.TryGetValue(glueType, out var executor) ? executor : null;
        }
    }
}