DocSpace-buildtools/products/ASC.Files/Server/Utils/ChunkedUploadSessionHolder.cs
2020-02-12 16:50:38 +03:00

132 lines
4.9 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.IO;
using ASC.Common.Logging;
using ASC.Core.ChunkedUploader;
using ASC.Files.Core;
using ASC.Web.Files.Classes;
using ASC.Web.Studio.Core;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Options;
using File = ASC.Files.Core.File;
namespace ASC.Web.Files.Utils
{
public class ChunkedUploadSessionHolder
{
public static readonly TimeSpan SlidingExpiration = TimeSpan.FromHours(12);
public IOptionsMonitor<ILog> Options { get; }
public GlobalStore GlobalStore { get; }
public SetupInfo SetupInfo { get; }
public ChunkedUploadSessionHolder(IOptionsMonitor<ILog> options, GlobalStore globalStore, SetupInfo setupInfo)
{
// clear old sessions
try
{
CommonSessionHolder(false).DeleteExpired();
}
catch (Exception err)
{
options.CurrentValue.Error(err);
}
Options = options;
GlobalStore = globalStore;
SetupInfo = setupInfo;
}
public void StoreSession(ChunkedUploadSession s)
{
CommonSessionHolder(false).Store(s);
}
public void RemoveSession(ChunkedUploadSession s)
{
CommonSessionHolder(false).Remove(s);
}
public ChunkedUploadSession GetSession(string sessionId)
{
return (ChunkedUploadSession)CommonSessionHolder(false).Get(sessionId);
}
public ChunkedUploadSession CreateUploadSession(File file, long contentLength)
{
var result = new ChunkedUploadSession(file, contentLength);
CommonSessionHolder().Init(result);
return result;
}
public void UploadChunk(ChunkedUploadSession uploadSession, Stream stream, long length)
{
CommonSessionHolder().UploadChunk(uploadSession, stream, length);
}
public void FinalizeUploadSession(ChunkedUploadSession uploadSession)
{
CommonSessionHolder().Finalize(uploadSession);
}
public void Move(ChunkedUploadSession chunkedUploadSession, string newPath)
{
CommonSessionHolder().Move(chunkedUploadSession, newPath);
}
public void AbortUploadSession(ChunkedUploadSession uploadSession)
{
CommonSessionHolder().Abort(uploadSession);
}
public Stream UploadSingleChunk(ChunkedUploadSession uploadSession, Stream stream, long chunkLength)
{
return CommonSessionHolder().UploadSingleChunk(uploadSession, stream, chunkLength);
}
private CommonChunkedUploadSessionHolder CommonSessionHolder(bool currentTenant = true)
{
return new CommonChunkedUploadSessionHolder(Options, GlobalStore.GetStore(currentTenant), FileConstant.StorageDomainTmp, SetupInfo.ChunkUploadSize);
}
}
public static class ChunkedUploadSessionHolderExtention
{
public static IServiceCollection AddChunkedUploadSessionHolderService(this IServiceCollection services)
{
services.TryAddScoped<ChunkedUploadSessionHolder>();
return services
.AddGlobalStoreService()
.AddSetupInfo();
}
}
}