bitwarden/src/Core/Services/Implementations/InMemoryServiceBusApplicationCacheService.cs

using System;
using System.Threading.Tasks;
using Bit.Core.Enums;
using Bit.Core.Models.Table;
using Bit.Core.Repositories;
using Bit.Core.Utilities;
using Microsoft.Azure.ServiceBus;

namespace Bit.Core.Services
{
    public class InMemoryServiceBusApplicationCacheService : InMemoryApplicationCacheService, IApplicationCacheService
    {
        private readonly TopicClient _topicClient;
        private readonly string _subName;

        public InMemoryServiceBusApplicationCacheService(
            IOrganizationRepository organizationRepository,
            GlobalSettings globalSettings)
            : base(organizationRepository)
        {
            _subName = CoreHelpers.GetApplicationCacheServiceBusSubcriptionName(globalSettings);
            _topicClient = new TopicClient(globalSettings.ServiceBus.ConnectionString,
                globalSettings.ServiceBus.ApplicationCacheTopicName);
        }

        public override async Task UpsertOrganizationAbilityAsync(Organization organization)
        {
            await base.UpsertOrganizationAbilityAsync(organization);
            var message = new Message
            {
                Label = _subName,
                UserProperties =
                {
                    { "type", (byte)ApplicationCacheMessageType.UpsertOrganizationAbility },
                    { "id", organization.Id },
                }
            };
            var task = _topicClient.SendAsync(message);
        }

        public override async Task DeleteOrganizationAbilityAsync(Guid organizationId)
        {
            await base.DeleteOrganizationAbilityAsync(organizationId);
            var message = new Message
            {
                Label = _subName,
                UserProperties =
                {
                    { "type", (byte)ApplicationCacheMessageType.DeleteOrganizationAbility },
                    { "id", organizationId },
                }
            };
            var task = _topicClient.SendAsync(message);
        }

        public async Task BaseUpsertOrganizationAbilityAsync(Organization organization)
        {
            await base.UpsertOrganizationAbilityAsync(organization);
        }

        public async Task BaseDeleteOrganizationAbilityAsync(Guid organizationId)
        {
            await base.DeleteOrganizationAbilityAsync(organizationId);
        }
    }
}