253 lines
10 KiB
C#
253 lines
10 KiB
C#
/*
|
|
*
|
|
* (c) Copyright Ascensio System Limited 2010-2018
|
|
*
|
|
* This program is freeware. You can redistribute it and/or modify it under the terms of the GNU
|
|
* General Public License (GPL) version 3 as published by the Free Software Foundation (https://www.gnu.org/copyleft/gpl.html).
|
|
* In accordance with Section 7(a) of the GNU GPL 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 more details, see GNU GPL at https://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* You can contact Ascensio System SIA by email at sales@onlyoffice.com
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions of ONLYOFFICE must display
|
|
* Appropriate Legal Notices, as required under Section 5 of the GNU GPL version 3.
|
|
*
|
|
* Pursuant to Section 7 § 3(b) of the GNU GPL you must retain the original ONLYOFFICE logo which contains
|
|
* relevant author attributions when distributing the software. If the display of the logo in its graphic
|
|
* form is not reasonably feasible for technical reasons, you must include the words "Powered by ONLYOFFICE"
|
|
* in every copy of the program you distribute.
|
|
* Pursuant to Section 7 § 3(e) we decline to grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
*/
|
|
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
|
|
using ASC.Common.Caching;
|
|
using ASC.Core;
|
|
using ASC.Data.Storage.Configuration;
|
|
using ASC.Core.Common.Configuration;
|
|
using ASC.Data.Storage.DiscStorage;
|
|
using ASC.Common.DependencyInjection;
|
|
using Microsoft.AspNetCore.Routing;
|
|
|
|
namespace ASC.Data.Storage
|
|
{
|
|
public static class StorageFactory
|
|
{
|
|
private const string DefaultTenantName = "default";
|
|
private static readonly ICacheNotify Cache;
|
|
|
|
static StorageFactory()
|
|
{
|
|
Cache = AscCache.Notify;
|
|
Cache.Subscribe<DataStoreCacheItem>((r, act) => DataStoreCache.Remove(r.TenantId, r.Module));
|
|
}
|
|
|
|
public static IDataStore GetStorage(string tenant, string module)
|
|
{
|
|
return GetStorage(string.Empty, tenant, module);
|
|
}
|
|
|
|
public static IDataStore GetStorage(string configpath, string tenant, string module)
|
|
{
|
|
int tenantId;
|
|
int.TryParse(tenant, out tenantId);
|
|
return GetStorage(configpath, tenant, module, new TennantQuotaController(tenantId));
|
|
}
|
|
|
|
public static IDataStore GetStorage(string configpath, string tenant, string module, IQuotaController controller)
|
|
{
|
|
var tenantId = -2;
|
|
if (string.IsNullOrEmpty(tenant))
|
|
{
|
|
tenant = DefaultTenantName;
|
|
}
|
|
else
|
|
{
|
|
tenantId = Convert.ToInt32(tenant);
|
|
}
|
|
|
|
//Make tennant path
|
|
tenant = TennantPath.CreatePath(tenant);
|
|
|
|
var store = DataStoreCache.Get(tenant, module);
|
|
if (store == null)
|
|
{
|
|
var section = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
if (section == null)
|
|
{
|
|
throw new InvalidOperationException("config section not found");
|
|
}
|
|
|
|
var settings = StorageSettings.LoadForTenant(tenantId);
|
|
|
|
store = GetStoreAndCache(tenant, module, settings.DataStoreConsumer, controller);
|
|
}
|
|
return store;
|
|
}
|
|
|
|
public static IDataStore GetStorageFromConsumer(string configpath, string tenant, string module, DataStoreConsumer consumer)
|
|
{
|
|
if (tenant == null) tenant = DefaultTenantName;
|
|
|
|
//Make tennant path
|
|
tenant = TennantPath.CreatePath(tenant);
|
|
|
|
var section = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
if (section == null)
|
|
{
|
|
throw new InvalidOperationException("config section not found");
|
|
}
|
|
|
|
int tenantId;
|
|
int.TryParse(tenant, out tenantId);
|
|
return GetDataStore(tenant, module, consumer, new TennantQuotaController(tenantId));
|
|
}
|
|
|
|
public static IEnumerable<string> GetModuleList(string configpath, bool exceptDisabledMigration = false)
|
|
{
|
|
var section = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
return section.Module
|
|
.Where(x => x.Visible)
|
|
.Where(x=> !exceptDisabledMigration || !x.DisableMigrate)
|
|
.Select(x => x.Name);
|
|
}
|
|
|
|
public static IEnumerable<string> GetDomainList(string configpath, string modulename)
|
|
{
|
|
var section = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
if (section == null)
|
|
{
|
|
throw new ArgumentException("config section not found");
|
|
}
|
|
return
|
|
section.Module
|
|
.Single(x => x.Name.Equals(modulename, StringComparison.OrdinalIgnoreCase))
|
|
.Domain
|
|
.Where(x => x.Visible)
|
|
.Select(x => x.Name);
|
|
}
|
|
|
|
public static void InitializeHttpHandlers(this IEndpointRouteBuilder builder, string config = null)
|
|
{
|
|
//TODO:
|
|
//if (!HostingEnvironment.IsHosted)
|
|
//{
|
|
// throw new InvalidOperationException("Application not hosted.");
|
|
//}
|
|
|
|
var section = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
if (section != null)
|
|
{
|
|
//old scheme
|
|
var discHandler = section.GetHandler("disc");
|
|
if (discHandler != null && section.Module != null)
|
|
{
|
|
var props = discHandler.Property != null ? discHandler.Property.ToDictionary(r => r.Name, r => r.Value) : new Dictionary<string, string>();
|
|
foreach (var m in section.Module.Where(m => m.Type == "disc"))
|
|
{
|
|
if (m.Path.Contains(Constants.STORAGE_ROOT_PARAM))
|
|
builder.RegisterDiscDataHandler(
|
|
PathUtils.ResolveVirtualPath(m.VirtualPath),
|
|
PathUtils.ResolvePhysicalPath(m.Path, props),
|
|
m.Public);
|
|
|
|
if (m.Domain != null)
|
|
{
|
|
foreach (var d in m.Domain.Where(d => (d.Type == "disc" || string.IsNullOrEmpty(d.Type)) && d.Path.Contains(Constants.STORAGE_ROOT_PARAM)))
|
|
{
|
|
builder.RegisterDiscDataHandler(
|
|
PathUtils.ResolveVirtualPath(d.VirtualPath),
|
|
PathUtils.ResolvePhysicalPath(d.Path, props));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//new scheme
|
|
if (section.Module != null)
|
|
{
|
|
foreach (var m in section.Module)
|
|
{
|
|
//todo: add path criterion
|
|
if (m.Type == "disc" || !m.Public || m.Path.Contains(Constants.STORAGE_ROOT_PARAM))
|
|
builder.RegisterStorageHandler(
|
|
m.Name,
|
|
string.Empty,
|
|
m.Public);
|
|
|
|
//todo: add path criterion
|
|
if (m.Domain != null)
|
|
{
|
|
foreach (var d in m.Domain.Where(d => d.Path.Contains(Constants.STORAGE_ROOT_PARAM)))
|
|
{
|
|
builder.RegisterStorageHandler(
|
|
m.Name,
|
|
d.Name,
|
|
d.Public);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
internal static void ClearCache()
|
|
{
|
|
var tenantId = CoreContext.TenantManager.GetCurrentTenant().TenantId.ToString();
|
|
var path = TennantPath.CreatePath(tenantId);
|
|
foreach (var module in GetModuleList("", true))
|
|
{
|
|
Cache.Publish(DataStoreCacheItem.Create(path, module), CacheNotifyAction.Remove);
|
|
}
|
|
}
|
|
|
|
private static IDataStore GetStoreAndCache(string tenant, string module, DataStoreConsumer consumer, IQuotaController controller)
|
|
{
|
|
var store = GetDataStore(tenant, module, consumer, controller);
|
|
if (store != null)
|
|
{
|
|
DataStoreCache.Put(store, tenant, module);
|
|
}
|
|
return store;
|
|
}
|
|
|
|
private static IDataStore GetDataStore(string tenant, string module, DataStoreConsumer consumer, IQuotaController controller)
|
|
{
|
|
var storage = CommonServiceProvider.GetService<Configuration.Storage>();
|
|
var moduleElement = storage.GetModuleElement(module);
|
|
if (moduleElement == null)
|
|
{
|
|
throw new ArgumentException("no such module", module);
|
|
}
|
|
|
|
var handler = storage.GetHandler(moduleElement.Type);
|
|
Type instanceType;
|
|
IDictionary<string, string> props;
|
|
|
|
if (CoreContext.Configuration.Standalone &&
|
|
!moduleElement.DisableMigrate &&
|
|
consumer.IsSet)
|
|
{
|
|
instanceType = consumer.HandlerType;
|
|
props = consumer;
|
|
}
|
|
else
|
|
{
|
|
instanceType = Type.GetType(handler.Type, true);
|
|
props = handler.Property.ToDictionary(r=> r.Name, r=> r.Value);
|
|
}
|
|
|
|
return ((IDataStore)Activator.CreateInstance(instanceType, tenant, handler, moduleElement))
|
|
.Configure(props)
|
|
.SetQuotaController(moduleElement.Count ? controller : null
|
|
/*don't count quota if specified on module*/);
|
|
}
|
|
}
|
|
} |