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
  • Client
  • ConnectionWrapper.cs
Find file
BlameHistoryPermalink
  • dingsongjie's avatar
    性能部分 改进完成 · b9422953
    dingsongjie committed 5 years ago
    b9422953
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);
        }
    }
}