Skip to content
  • P
    Projects
  • G
    Groups
  • S
    Snippets
  • Help

丁松杰 / Pole

  • This project
    • Loading...
  • Sign in
Go to a project
  • Project
  • Repository
  • Issues 0
  • Merge Requests 0
  • Pipelines
  • Wiki
  • Snippets
  • Members
  • Activity
  • Graph
  • Charts
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
  • Files
  • Commits
  • Branches
  • Tags
  • Contributors
  • Graph
  • Compare
  • Charts
Switch branch/tag
  • Pole
  • src
  • Pole.EventBus.Rabbitmq
  • PoleRabbitmqStartupConfigExtensions.cs
Find file
BlameHistoryPermalink
  • dingsongjie's avatar
    重构 eventbus 的 代码结构 · 5511b5b8
    dingsongjie committed 5 years ago
    5511b5b8
PoleRabbitmqStartupConfigExtensions.cs 2.37 KB
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
using System;
using System.Collections.Concurrent;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
using Pole.Core;
using Pole.EventBus;
using Pole.EventBus.RabbitMQ;

namespace Microsoft.Extensions.DependencyInjection
{
    public static class PoleRabbitmqStartupConfigExtensions
    {
        private static ConcurrentDictionary<string, ConsumerRunner> ConsumerRunners = new ConcurrentDictionary<string, ConsumerRunner>();
        public static void AddRabbitMQ(
            this StartupConfig startupOption,
            Action<RabbitOptions> rabbitConfigAction,
            Func<IRabbitEventBusContainer, Task> eventBusConfig = default)
        {
            startupOption.Services.Configure<RabbitOptions>(config => rabbitConfigAction(config));
            startupOption.Services.AddSingleton<IRabbitMQClient, RabbitMQClient>();
            //startupOption.Services.AddHostedService<ConsumerManager>();
            startupOption.Services.AddSingleton<IRabbitEventBusContainer, EventBusContainer>();
            startupOption.Services.AddSingleton<IProducer, RabbitProducer>();
            startupOption.Services.AddSingleton(serviceProvider => serviceProvider.GetService<IRabbitEventBusContainer>() as IProducerInfoContainer);
            Startup.Register(async serviceProvider =>
            {
                var container = serviceProvider.GetService<IRabbitEventBusContainer>();
                var client = serviceProvider.GetService<IRabbitMQClient>();
                var rabbitOptions = serviceProvider.GetService<IOptions<RabbitOptions>>().Value;
                if (eventBusConfig != default)
                    await eventBusConfig(container);
                else
                    await container.AutoRegister();

                var consumers = container.GetConsumers();
                foreach (var consumer in consumers)
                {
                    if (consumer is RabbitConsumer value)
                    {
                        var queue = value.QueueInfo;
                        var key = queue.Queue;

                        var runner = new ConsumerRunner(client, serviceProvider, value, queue, rabbitOptions);
                        ConsumerRunners.TryAdd(key, runner);
                        await runner.Run();
                    }
                }
            });
        }
    }
}