2019-06-04 14:43:20 +00:00
|
|
|
/*
|
|
|
|
*
|
|
|
|
* (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.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
2022-02-10 11:24:16 +00:00
|
|
|
namespace ASC.Data.Storage;
|
2021-05-21 13:26:42 +00:00
|
|
|
|
2022-02-10 11:24:16 +00:00
|
|
|
public class CrossModuleTransferUtility
|
|
|
|
{
|
|
|
|
private readonly ILog _logger;
|
|
|
|
private readonly IDataStore _source;
|
|
|
|
private readonly IDataStore _destination;
|
|
|
|
private readonly long _maxChunkUploadSize;
|
|
|
|
private readonly int _chunkSize;
|
|
|
|
private readonly IOptionsMonitor<ILog> _option;
|
|
|
|
private readonly TempStream _tempStream;
|
|
|
|
private readonly TempPath _tempPath;
|
|
|
|
|
|
|
|
public CrossModuleTransferUtility(
|
|
|
|
IOptionsMonitor<ILog> option,
|
|
|
|
TempStream tempStream,
|
|
|
|
TempPath tempPath,
|
|
|
|
IDataStore source,
|
|
|
|
IDataStore destination)
|
|
|
|
{
|
2022-02-16 09:47:11 +00:00
|
|
|
_logger = option.Get("ASC.CrossModuleTransferUtility");
|
|
|
|
_option = option;
|
|
|
|
_tempStream = tempStream;
|
|
|
|
_tempPath = tempPath;
|
|
|
|
_source = source ?? throw new ArgumentNullException(nameof(source));
|
|
|
|
_destination = destination ?? throw new ArgumentNullException(nameof(destination));
|
|
|
|
_maxChunkUploadSize = 10 * 1024 * 1024;
|
|
|
|
_chunkSize = 5 * 1024 * 1024;
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
|
2022-02-23 19:42:34 +00:00
|
|
|
public Task CopyFileAsync(string srcDomain, string srcPath, string destDomain, string destPath)
|
2022-02-10 11:24:16 +00:00
|
|
|
{
|
2022-02-16 09:47:11 +00:00
|
|
|
if (srcDomain == null)
|
2019-06-04 14:43:20 +00:00
|
|
|
{
|
2022-02-16 09:47:11 +00:00
|
|
|
throw new ArgumentNullException(nameof(srcDomain));
|
2019-06-04 14:43:20 +00:00
|
|
|
}
|
2021-05-17 11:35:00 +00:00
|
|
|
|
2022-02-23 19:42:34 +00:00
|
|
|
if (srcPath == null)
|
2019-06-04 14:43:20 +00:00
|
|
|
{
|
2022-02-23 19:42:34 +00:00
|
|
|
throw new ArgumentNullException(nameof(srcPath));
|
2019-06-04 14:43:20 +00:00
|
|
|
}
|
2022-02-10 11:24:16 +00:00
|
|
|
|
|
|
|
if (destDomain == null)
|
|
|
|
{
|
|
|
|
throw new ArgumentNullException(nameof(destDomain));
|
|
|
|
}
|
2022-02-10 11:06:37 +00:00
|
|
|
|
2022-02-10 11:24:16 +00:00
|
|
|
if (destPath == null)
|
|
|
|
{
|
|
|
|
throw new ArgumentNullException(nameof(destPath));
|
|
|
|
}
|
|
|
|
|
2022-02-23 19:42:34 +00:00
|
|
|
return InternalCopyFileAsync(srcDomain, srcPath, destDomain, destPath);
|
|
|
|
}
|
|
|
|
|
|
|
|
private async Task InternalCopyFileAsync(string srcDomain, string srcPath, string destDomain, string destPath)
|
|
|
|
{
|
|
|
|
using var stream = await _source.GetReadStreamAsync(srcDomain, srcPath);
|
2022-02-10 11:24:16 +00:00
|
|
|
if (stream.Length < _maxChunkUploadSize)
|
|
|
|
{
|
2022-02-23 19:42:34 +00:00
|
|
|
await _destination.SaveAsync(destDomain, destPath, stream);
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
var session = new CommonChunkedUploadSession(stream.Length);
|
|
|
|
var holder = new CommonChunkedUploadSessionHolder(_tempPath, _option, _destination, destDomain);
|
2022-02-23 19:42:34 +00:00
|
|
|
await holder.InitAsync(session);
|
2022-02-10 11:24:16 +00:00
|
|
|
try
|
2022-02-10 11:06:37 +00:00
|
|
|
{
|
2022-02-10 11:24:16 +00:00
|
|
|
Stream memstream = null;
|
|
|
|
try
|
|
|
|
{
|
|
|
|
while (GetStream(stream, out memstream))
|
|
|
|
{
|
|
|
|
memstream.Seek(0, SeekOrigin.Begin);
|
2022-02-23 19:42:34 +00:00
|
|
|
await holder.UploadChunkAsync(session, memstream, _chunkSize);
|
|
|
|
await memstream.DisposeAsync();
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
finally
|
|
|
|
{
|
|
|
|
if (memstream != null)
|
|
|
|
{
|
2022-02-23 19:42:34 +00:00
|
|
|
await memstream.DisposeAsync();
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
}
|
2022-02-10 11:06:37 +00:00
|
|
|
|
2022-02-23 19:42:34 +00:00
|
|
|
await holder.FinalizeAsync(session);
|
|
|
|
await _destination.MoveAsync(destDomain, session.TempPath, destDomain, destPath);
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
catch (Exception ex)
|
2022-02-10 11:06:37 +00:00
|
|
|
{
|
2022-02-10 11:24:16 +00:00
|
|
|
_logger.Error("Copy File", ex);
|
2022-02-23 19:42:34 +00:00
|
|
|
await holder.AbortAsync(session);
|
2022-02-10 11:06:37 +00:00
|
|
|
}
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private bool GetStream(Stream stream, out Stream memstream)
|
|
|
|
{
|
|
|
|
memstream = _tempStream.Create();
|
|
|
|
var total = 0;
|
|
|
|
int readed;
|
|
|
|
const int portion = 2048;
|
|
|
|
var buffer = new byte[portion];
|
2022-02-10 11:06:37 +00:00
|
|
|
|
2022-02-10 11:24:16 +00:00
|
|
|
while ((readed = stream.Read(buffer, 0, portion)) > 0)
|
|
|
|
{
|
|
|
|
memstream.Write(buffer, 0, readed);
|
|
|
|
total += readed;
|
|
|
|
if (total >= _chunkSize)
|
2022-02-10 11:06:37 +00:00
|
|
|
{
|
2022-02-10 11:24:16 +00:00
|
|
|
break;
|
2022-02-10 11:06:37 +00:00
|
|
|
}
|
2022-02-10 11:24:16 +00:00
|
|
|
}
|
2022-02-10 11:06:37 +00:00
|
|
|
|
2022-02-10 11:24:16 +00:00
|
|
|
return total > 0;
|
|
|
|
}
|
2019-06-04 14:43:20 +00:00
|
|
|
}
|