RabbitEventBus.cs
2.92 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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
using Pole.Core.Exceptions;
using Pole.Core.Utils;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
namespace Pole.EventBus.RabbitMQ
{
public class RabbitEventBus
{
readonly IObserverUnitContainer observerUnitContainer;
public RabbitEventBus(
IObserverUnitContainer observerUnitContainer,
IRabbitEventBusContainer eventBusContainer,
string exchange, string routePrefix, int lBCount = 1, bool reenqueue = true, bool persistent = false)
{
if (string.IsNullOrEmpty(exchange))
throw new ArgumentNullException(nameof(exchange));
if (string.IsNullOrEmpty(routePrefix))
throw new ArgumentNullException(nameof(routePrefix));
if (lBCount < 1)
throw new ArgumentOutOfRangeException($"{nameof(lBCount)} must be greater than 1");
this.observerUnitContainer = observerUnitContainer;
Container = eventBusContainer;
Exchange = exchange;
RoutePrefix = routePrefix;
LBCount = lBCount;
Persistent = persistent;
ConsumerConfig = new ConsumerOptions
{
Reenqueue = reenqueue,
ErrorQueueSuffix = "_error",
MaxReenqueueTimes = 10
};
}
public IRabbitEventBusContainer Container { get; }
public string Exchange { get; }
public string RoutePrefix { get; }
public int LBCount { get; }
public ConsumerOptions ConsumerConfig { get; set; }
public List<string> RouteList { get; }
public Type Event { get; set; }
public string EventName { get; set; }
/// <summary>
/// 消息是否持久化
/// </summary>
public bool Persistent { get; set; }
public List<RabbitConsumer> Consumers { get; set; } = new List<RabbitConsumer>();
public RabbitEventBus BindEvent(Type eventType, string eventName)
{
Event = eventType;
EventName = eventName;
return this;
}
public Task AddGrainConsumer<PrimaryKey>()
{
var observerUnits = observerUnitContainer.GetUnits<PrimaryKey>(EventName);
foreach (var observerUnit in observerUnits)
{
string queueNameSuffix = observerUnit.EventHandlerType.FullName;
var consumer = new RabbitConsumer(
observerUnit.GetBatchEventHandler())
{
EventBus = this,
QueueInfo = new QueueInfo { RoutingKey = string.Empty, Queue = $"{RoutePrefix}_{queueNameSuffix}" },
Config = ConsumerConfig
};
Consumers.Add(consumer);
}
return Enable();
}
public Task Enable()
{
return Container.Work(this);
}
}
}