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-03-09 17:15:51 +00:00
|
|
|
ArgumentNullException.ThrowIfNull(srcDomain);
|
|
|
|
ArgumentNullException.ThrowIfNull(srcPath);
|
|
|
|
ArgumentNullException.ThrowIfNull(destDomain);
|
|
|
|
ArgumentNullException.ThrowIfNull(destPath);
|
2022-02-10 11:24:16 +00:00
|
|
|
|
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
|
|
|
}
|