ConnectionWrapper.cs
1.31 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
using RabbitMQ.Client;
using System;
using System.Collections.Generic;
using System.Threading;
namespace Pole.EventBus.RabbitMQ
{
public class ConnectionWrapper
{
private readonly List<ModelWrapper> channels = new List<ModelWrapper>();
private readonly IConnection connection;
readonly SemaphoreSlim semaphoreSlim = new SemaphoreSlim(1);
public ConnectionWrapper(
IConnection connection,
RabbitOptions options)
{
this.connection = connection;
Options = options;
}
public RabbitOptions Options { get; }
public (bool success, ModelWrapper model) Get()
{
semaphoreSlim.Wait();
try
{
if (channels.Count < Options.MasChannelsPerConnection)
{
var channel = new ModelWrapper(this, connection.CreateModel());
channel.Model.ConfirmSelect();
channels.Add(channel);
return (true, channel);
}
}
finally
{
semaphoreSlim.Release();
}
return (false, default);
}
public void Return(ModelWrapper model)
{
channels.Remove(model);
}
}
}