2022-03-15 18:00:53 +00:00
|
|
|
// (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
|
|
|
|
|
2022-02-15 11:52:43 +00:00
|
|
|
namespace ASC.Core.Caching;
|
|
|
|
|
|
|
|
[Singletone]
|
|
|
|
class QuotaServiceCache
|
2020-10-19 15:53:15 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
internal const string KeyQuota = "quota";
|
|
|
|
internal const string KeyQuotaRows = "quotarows";
|
|
|
|
internal readonly ICache Cache;
|
|
|
|
internal readonly ICacheNotify<QuotaCacheItem> CacheNotify;
|
|
|
|
internal readonly bool QuotaCacheEnabled;
|
2020-12-28 13:22:08 +00:00
|
|
|
|
2022-02-15 11:52:43 +00:00
|
|
|
public QuotaServiceCache(IConfiguration Configuration, ICacheNotify<QuotaCacheItem> cacheNotify, ICache cache)
|
|
|
|
{
|
|
|
|
if (Configuration["core:enable-quota-cache"] == null)
|
2020-02-20 08:05:10 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
QuotaCacheEnabled = true;
|
2020-02-20 08:05:10 +00:00
|
|
|
}
|
2022-02-15 11:52:43 +00:00
|
|
|
else
|
2020-02-20 08:05:10 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
QuotaCacheEnabled = !bool.TryParse(Configuration["core:enable-quota-cache"], out var enabled) || enabled;
|
2020-02-20 08:05:10 +00:00
|
|
|
}
|
|
|
|
|
2022-03-25 14:54:34 +00:00
|
|
|
CacheNotify = cacheNotify;
|
|
|
|
Cache = cache;
|
2020-10-19 15:53:15 +00:00
|
|
|
|
2022-02-15 11:52:43 +00:00
|
|
|
cacheNotify.Subscribe((i) =>
|
2020-02-20 14:57:48 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
if (i.Key == KeyQuota)
|
2022-02-15 11:52:43 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
Cache.Remove(KeyQuota);
|
2022-02-14 21:02:57 +00:00
|
|
|
}
|
2022-02-15 11:52:43 +00:00
|
|
|
else
|
2020-12-28 13:22:08 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
Cache.Remove(i.Key);
|
2020-12-28 13:22:08 +00:00
|
|
|
}
|
2022-02-15 11:52:43 +00:00
|
|
|
}, CacheNotifyAction.Any);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
[Scope]
|
|
|
|
class ConfigureCachedQuotaService : IConfigureNamedOptions<CachedQuotaService>
|
|
|
|
{
|
|
|
|
private readonly IOptionsSnapshot<DbQuotaService> _service;
|
|
|
|
private readonly QuotaServiceCache _quotaServiceCache;
|
|
|
|
|
|
|
|
public ConfigureCachedQuotaService(
|
|
|
|
IOptionsSnapshot<DbQuotaService> service,
|
|
|
|
QuotaServiceCache quotaServiceCache)
|
|
|
|
{
|
|
|
|
_service = service;
|
|
|
|
_quotaServiceCache = quotaServiceCache;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void Configure(string name, CachedQuotaService options)
|
|
|
|
{
|
|
|
|
Configure(options);
|
2022-03-25 14:54:34 +00:00
|
|
|
options.Service = _service.Get(name);
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public void Configure(CachedQuotaService options)
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
options.Service = _service.Value;
|
|
|
|
options.QuotaServiceCache = _quotaServiceCache;
|
|
|
|
options.Cache = _quotaServiceCache.Cache;
|
|
|
|
options.CacheNotify = _quotaServiceCache.CacheNotify;
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
}
|
2020-12-28 13:22:08 +00:00
|
|
|
|
2022-02-15 11:52:43 +00:00
|
|
|
[Scope]
|
|
|
|
class CachedQuotaService : IQuotaService
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
internal IQuotaService Service { get; set; }
|
|
|
|
internal ICache Cache { get; set; }
|
|
|
|
internal ICacheNotify<QuotaCacheItem> CacheNotify { get; set; }
|
|
|
|
internal QuotaServiceCache QuotaServiceCache { get; set; }
|
2022-02-15 11:52:43 +00:00
|
|
|
|
2022-03-25 14:54:34 +00:00
|
|
|
private readonly TimeSpan _cacheExpiration;
|
2022-02-15 11:52:43 +00:00
|
|
|
|
|
|
|
public CachedQuotaService()
|
|
|
|
{
|
2022-03-17 16:57:02 +00:00
|
|
|
_cacheExpiration = TimeSpan.FromMinutes(10);
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public CachedQuotaService(DbQuotaService service, QuotaServiceCache quotaServiceCache) : this()
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
Service = service ?? throw new ArgumentNullException(nameof(service));
|
|
|
|
QuotaServiceCache = quotaServiceCache;
|
|
|
|
Cache = quotaServiceCache.Cache;
|
|
|
|
CacheNotify = quotaServiceCache.CacheNotify;
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public IEnumerable<TenantQuota> GetTenantQuotas()
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
var quotas = Cache.Get<IEnumerable<TenantQuota>>(QuotaServiceCache.KeyQuota);
|
2022-02-15 11:52:43 +00:00
|
|
|
if (quotas == null)
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
quotas = Service.GetTenantQuotas();
|
|
|
|
if (QuotaServiceCache.QuotaCacheEnabled)
|
2022-02-15 11:52:43 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
Cache.Insert(QuotaServiceCache.KeyQuota, quotas, DateTime.UtcNow.Add(_cacheExpiration));
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
2020-12-28 13:22:08 +00:00
|
|
|
}
|
|
|
|
|
2022-02-15 11:52:43 +00:00
|
|
|
return quotas;
|
|
|
|
}
|
|
|
|
|
|
|
|
public TenantQuota GetTenantQuota(int tenant)
|
|
|
|
{
|
|
|
|
return GetTenantQuotas().SingleOrDefault(q => q.Tenant == tenant);
|
|
|
|
}
|
|
|
|
|
|
|
|
public TenantQuota SaveTenantQuota(TenantQuota quota)
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
var q = Service.SaveTenantQuota(quota);
|
|
|
|
CacheNotify.Publish(new QuotaCacheItem { Key = QuotaServiceCache.KeyQuota }, CacheNotifyAction.Any);
|
2022-02-15 11:52:43 +00:00
|
|
|
|
|
|
|
return q;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void RemoveTenantQuota(int tenant)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void SetTenantQuotaRow(TenantQuotaRow row, bool exchange)
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
Service.SetTenantQuotaRow(row, exchange);
|
|
|
|
CacheNotify.Publish(new QuotaCacheItem { Key = GetKey(row.Tenant) }, CacheNotifyAction.InsertOrUpdate);
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
public IEnumerable<TenantQuotaRow> FindTenantQuotaRows(int tenantId)
|
|
|
|
{
|
|
|
|
var key = GetKey(tenantId);
|
2022-03-25 14:54:34 +00:00
|
|
|
var result = Cache.Get<IEnumerable<TenantQuotaRow>>(key);
|
2022-02-15 11:52:43 +00:00
|
|
|
|
|
|
|
if (result == null)
|
2020-12-28 13:22:08 +00:00
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
result = Service.FindTenantQuotaRows(tenantId);
|
|
|
|
Cache.Insert(key, result, DateTime.UtcNow.Add(_cacheExpiration));
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
public string GetKey(int tenant)
|
|
|
|
{
|
2022-03-25 14:54:34 +00:00
|
|
|
return QuotaServiceCache.KeyQuotaRows + tenant;
|
2022-02-15 11:52:43 +00:00
|
|
|
}
|
2019-05-15 14:56:09 +00:00
|
|
|
}
|