163 lines
5.5 KiB
C#
163 lines
5.5 KiB
C#
|
/*
|
|||
|
*
|
|||
|
* (c) Copyright Ascensio System Limited 2010-2021
|
|||
|
*
|
|||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|||
|
* you may not use this file except in compliance with the License.
|
|||
|
* You may obtain a copy of the License at
|
|||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
|
* Unless required by applicable law or agreed to in writing, software
|
|||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|||
|
* See the License for the specific language governing permissions and
|
|||
|
* limitations under the License.
|
|||
|
*
|
|||
|
*/
|
|||
|
|
|||
|
|
|||
|
using System;
|
|||
|
using System.Collections.Generic;
|
|||
|
using System.Linq;
|
|||
|
using System.Linq.Expressions;
|
|||
|
|
|||
|
using ASC.Common;
|
|||
|
using ASC.Common.Caching;
|
|||
|
using ASC.Core.Common.EF;
|
|||
|
using ASC.Core.Tenants;
|
|||
|
using ASC.Files.Core;
|
|||
|
using ASC.Files.Core.EF;
|
|||
|
using ASC.Web.Files.Core.Search;
|
|||
|
|
|||
|
namespace ASC.Files.ThumbnailBuilder
|
|||
|
{
|
|||
|
[Scope]
|
|||
|
internal class FileDataProvider
|
|||
|
{
|
|||
|
private readonly ThumbnailSettings thumbnailSettings;
|
|||
|
private readonly ICache cache;
|
|||
|
private readonly FilesDbContext filesDbContext;
|
|||
|
private readonly CoreDbContext coreDbContext;
|
|||
|
private readonly string cacheKey;
|
|||
|
|
|||
|
public FileDataProvider(
|
|||
|
ThumbnailSettings thumbnailSettings,
|
|||
|
ICache ascCache,
|
|||
|
DbContextManager<FilesDbContext> dbContextManager,
|
|||
|
DbContextManager<CoreDbContext> coredbContextManager
|
|||
|
)
|
|||
|
{
|
|||
|
this.thumbnailSettings = thumbnailSettings;
|
|||
|
cache = ascCache;
|
|||
|
filesDbContext = dbContextManager.Get(thumbnailSettings.ConnectionStringName);
|
|||
|
coreDbContext = coredbContextManager.Get(thumbnailSettings.ConnectionStringName);
|
|||
|
cacheKey = "PremiumTenants";
|
|||
|
}
|
|||
|
|
|||
|
public int[] GetPremiumTenants()
|
|||
|
{
|
|||
|
var result = cache.Get<int[]>(cacheKey);
|
|||
|
|
|||
|
if (result != null)
|
|||
|
{
|
|||
|
return result;
|
|||
|
}
|
|||
|
|
|||
|
/*
|
|||
|
// v_premium_tenants view:
|
|||
|
select
|
|||
|
t.tenant,
|
|||
|
(now() < max(t.stamp)) AS premium
|
|||
|
|
|||
|
from tenants_tariff t
|
|||
|
left join tenants_quota q ON (t.tariff = q.tenant)
|
|||
|
|
|||
|
where
|
|||
|
(
|
|||
|
(
|
|||
|
isnull(t.comment) or
|
|||
|
(
|
|||
|
(not((t.comment like '%non-profit%'))) and
|
|||
|
(not((t.comment like '%test%'))) and
|
|||
|
(not((t.comment like '%translate%'))) and
|
|||
|
(not((t.comment like '%trial%')))
|
|||
|
)
|
|||
|
) and
|
|||
|
(not((q.features like '%free%'))) and
|
|||
|
(not((q.features like '%non-profit%'))) and
|
|||
|
(not((q.features like '%trial%')))
|
|||
|
)
|
|||
|
group by t.tenant
|
|||
|
*/
|
|||
|
|
|||
|
var search =
|
|||
|
coreDbContext.Tariffs
|
|||
|
.Join(coreDbContext.Quotas.DefaultIfEmpty(), a => a.Tariff, b => b.Tenant, (tariff, quota) => new { tariff, quota })
|
|||
|
.Where(r =>
|
|||
|
(
|
|||
|
r.tariff.Comment == null ||
|
|||
|
(
|
|||
|
!r.tariff.Comment.Contains("non-profit") &&
|
|||
|
!r.tariff.Comment.Contains("test") &&
|
|||
|
!r.tariff.Comment.Contains("translate") &&
|
|||
|
!r.tariff.Comment.Contains("trial")
|
|||
|
)
|
|||
|
) &&
|
|||
|
(
|
|||
|
!r.quota.Features.Contains("free") &&
|
|||
|
!r.quota.Features.Contains("non-profit") &&
|
|||
|
!r.quota.Features.Contains("trial")
|
|||
|
)
|
|||
|
)
|
|||
|
.GroupBy(r => r.tariff.Tenant)
|
|||
|
.Select(r => new { tenant = r.Key, stamp = r.Max(b => b.tariff.Stamp) })
|
|||
|
.Where(r=> r.stamp > DateTime.UtcNow);
|
|||
|
|
|||
|
result = search.Select(r=> r.tenant).ToArray();
|
|||
|
|
|||
|
cache.Insert(cacheKey, result, DateTime.UtcNow.AddHours(1));
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
|
|||
|
private IEnumerable<FileData<int>> GetFileData(Expression<Func<DbFile, bool>> where)
|
|||
|
{
|
|||
|
var search = filesDbContext.Files
|
|||
|
.Where(r => r.CurrentVersion && r.Thumb == Thumbnail.Waiting && !r.Encrypted)
|
|||
|
.OrderByDescending(r => r.ModifiedOn)
|
|||
|
.Take(thumbnailSettings.SqlMaxResults);
|
|||
|
|
|||
|
if (where != null)
|
|||
|
{
|
|||
|
search = search.Where(where);
|
|||
|
}
|
|||
|
|
|||
|
return search
|
|||
|
.Join(filesDbContext.Tenants, r => r.TenantId, r => r.Id, (f, t) => new FileTenant { DbFile = f, DbTenant = t })
|
|||
|
.Where(r => r.DbTenant.Status == TenantStatus.Active)
|
|||
|
.Select(r => new FileData<int>(r.DbFile.TenantId, r.DbFile.Id, ""))
|
|||
|
.ToList();
|
|||
|
}
|
|||
|
|
|||
|
public IEnumerable<FileData<int>> GetFilesWithoutThumbnails()
|
|||
|
{
|
|||
|
IEnumerable<FileData<int>> result;
|
|||
|
|
|||
|
var premiumTenants = GetPremiumTenants();
|
|||
|
|
|||
|
if (premiumTenants.Any())
|
|||
|
{
|
|||
|
result = GetFileData(r => premiumTenants.Contains(r.TenantId));
|
|||
|
|
|||
|
if (result.Any())
|
|||
|
{
|
|||
|
return result;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
result = GetFileData(null);
|
|||
|
|
|||
|
return result;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|