DocSpace-client/common/ASC.Core.Common/Caching/CachedQuotaService.cs

// (c) Copyright Ascensio System SIA 2010-2022
//
// This program is a free software product.
// You can redistribute it and/or modify it under the terms
// of the GNU Affero General Public License (AGPL) version 3 as published by the Free Software
// Foundation. In accordance with Section 7(a) of the GNU AGPL its Section 15 shall be amended
// to the effect that Ascensio System SIA expressly excludes the warranty of non-infringement of
// any third-party rights.
//
// This program is distributed WITHOUT ANY WARRANTY, without even the implied warranty
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For details, see
// the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
//
// You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia, EU, LV-1021.
//
// The  interactive user interfaces in modified source and object code versions of the Program must
// display Appropriate Legal Notices, as required under Section 5 of the GNU AGPL version 3.
//
// Pursuant to Section 7(b) of the License you must retain the original Product logo when
// distributing the program. Pursuant to Section 7(e) we decline to grant you any rights under
// trademark law for use of our trademarks.
//
// All the Product's GUI elements, including illustrations and icon sets, as well as technical writing
// content are licensed under the terms of the Creative Commons Attribution-ShareAlike 4.0
// International. See the License terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode

namespace ASC.Core.Caching;

[Singletone]
class QuotaServiceCache
{
    internal const string KeyQuota = "quota";
    internal const string KeyQuotaRows = "quotarows";
    internal const string KeyUserQuotaRows = "userquotarows";
    internal readonly ICache Cache;
    internal readonly ICacheNotify<QuotaCacheItem> CacheNotify;
    internal readonly bool QuotaCacheEnabled;

    public QuotaServiceCache(IConfiguration Configuration, ICacheNotify<QuotaCacheItem> cacheNotify, ICache cache)
    {
        if (Configuration["core:enable-quota-cache"] == null)
        {
            QuotaCacheEnabled = true;
        }
        else
        {
            QuotaCacheEnabled = !bool.TryParse(Configuration["core:enable-quota-cache"], out var enabled) || enabled;
        }

        CacheNotify = cacheNotify;
        Cache = cache;

        cacheNotify.Subscribe((i) =>
        {
            if (i.Key == KeyQuota)
            {
                Cache.Remove(KeyQuota);
            }
            else
            {
                Cache.Remove(i.Key);
            }
        }, CacheNotifyAction.Any);
    }
}

[Scope]
class CachedQuotaService : IQuotaService
{
    private readonly IQuotaService Service;
    private readonly ICache Cache;
    private readonly ICacheNotify<QuotaCacheItem> CacheNotify;
    private readonly QuotaServiceCache QuotaServiceCache;

    private readonly GeolocationHelper _geolocationHelper;

    private readonly TimeSpan _cacheExpiration;

    public CachedQuotaService()
    {
        _cacheExpiration = TimeSpan.FromMinutes(10);
    }

    public CachedQuotaService(DbQuotaService service, QuotaServiceCache quotaServiceCache, GeolocationHelper geolocationHelper) : this()
    {
        Service = service ?? throw new ArgumentNullException(nameof(service));
        QuotaServiceCache = quotaServiceCache;
        Cache = quotaServiceCache.Cache;
        CacheNotify = quotaServiceCache.CacheNotify;
        _geolocationHelper = geolocationHelper;
    }

    public async Task<IEnumerable<TenantQuota>> GetTenantQuotasAsync()
    {
        var quotas = Cache.Get<IEnumerable<TenantQuota>>(QuotaServiceCache.KeyQuota + (await _geolocationHelper.GetIPGeolocationFromHttpContextAsync()).Key);
        if (quotas == null)
        {
            quotas = await Service.GetTenantQuotasAsync();
            if (QuotaServiceCache.QuotaCacheEnabled)
            {
                Cache.Insert(QuotaServiceCache.KeyQuota, quotas, DateTime.UtcNow.Add(_cacheExpiration));
            }
        }

        return quotas;
    }

    public async Task<TenantQuota> GetTenantQuotaAsync(int tenant)
    {
        return (await GetTenantQuotasAsync()).SingleOrDefault(q => q.TenantId == tenant);
    }

    public async Task<TenantQuota> SaveTenantQuotaAsync(TenantQuota quota)
    {
        var q = await Service.SaveTenantQuotaAsync(quota);
        CacheNotify.Publish(new QuotaCacheItem { Key = QuotaServiceCache.KeyQuota }, CacheNotifyAction.Any);

        return q;
    }

    public Task RemoveTenantQuotaAsync(int tenant)
    {
        throw new NotImplementedException();
    }

    public async Task SetTenantQuotaRowAsync(TenantQuotaRow row, bool exchange)
    {
        await Service.SetTenantQuotaRowAsync(row, exchange);
        CacheNotify.Publish(new QuotaCacheItem { Key = GetKey(row.TenantId) }, CacheNotifyAction.Any);

        if (row.UserId != Guid.Empty)
        {
            CacheNotify.Publish(new QuotaCacheItem { Key = GetKey(row.TenantId, row.UserId) }, CacheNotifyAction.Any);
        }
    }

    public async Task<IEnumerable<TenantQuotaRow>> FindTenantQuotaRowsAsync(int tenantId)
    {
        var key = GetKey(tenantId);
        var result = Cache.Get<IEnumerable<TenantQuotaRow>>(key);

        if (result == null)
        {
            result = await Service.FindTenantQuotaRowsAsync(tenantId);
            Cache.Insert(key, result, DateTime.UtcNow.Add(_cacheExpiration));
        }

        return result;
    }

    public async Task<IEnumerable<TenantQuotaRow>> FindUserQuotaRowsAsync(int tenantId, Guid userId)
    {
        var key = GetKey(tenantId, userId);
        var result = Cache.Get<IEnumerable<TenantQuotaRow>>(key);

        if (result == null)
        {
            result = await Service.FindUserQuotaRowsAsync(tenantId, userId);
            Cache.Insert(key, result, DateTime.UtcNow.Add(_cacheExpiration));
        }

        return result;
    }

    public string GetKey(int tenant)
    {
        return QuotaServiceCache.KeyQuotaRows + tenant;
    }

    public string GetKey(int tenant, Guid userId)
    {
        return QuotaServiceCache.KeyQuotaRows + tenant + userId;
    }
}