mirror of
https://github.com/duplicati/duplicati.git
synced 2025-11-27 19:10:29 +08:00
118 lines
4.7 KiB
C#
118 lines
4.7 KiB
C#
// Copyright (C) 2025, The Duplicati Team
|
|
// https://duplicati.com, hello@duplicati.com
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a
|
|
// copy of this software and associated documentation files (the "Software"),
|
|
// to deal in the Software without restriction, including without limitation
|
|
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
// and/or sell copies of the Software, and to permit persons to whom the
|
|
// Software is furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
// OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
// DEALINGS IN THE SOFTWARE.
|
|
|
|
using System;
|
|
using CoCoL;
|
|
using System.Threading.Tasks;
|
|
using System.IO;
|
|
using System.Net.Http;
|
|
using Duplicati.Library.Utility;
|
|
using System.Threading;
|
|
using Timeout = System.Threading.Timeout;
|
|
|
|
namespace Duplicati.Library.UsageReporter
|
|
{
|
|
public static class ReportSetUploader
|
|
{
|
|
/// <summary>
|
|
/// The tag used for logging
|
|
/// </summary>
|
|
private static readonly string LOGTAG = Logging.Log.LogTagFromType(typeof(ReportSetUploader));
|
|
|
|
/// <summary>
|
|
/// The maximum number of pending uploads
|
|
/// </summary>
|
|
private const int MAX_PENDING_UPLOADS = 50;
|
|
|
|
/// <summary>
|
|
/// The target upload url
|
|
/// </summary>
|
|
private const string UPLOAD_URL = "https://usage-reporter.duplicati.com/api/v1/report";
|
|
|
|
/// <summary>
|
|
/// The default timeout in seconds for report uploads
|
|
/// </summary>
|
|
private const int UPLOAD_OPERATION_TIMEOUT_SECONDS = 60;
|
|
|
|
|
|
/// <summary>
|
|
/// Runs the upload process
|
|
/// </summary>
|
|
/// <returns>A tuple with the completion task and the channel to use</returns>
|
|
public static Tuple<Task, IWriteChannel<string>> Run()
|
|
{
|
|
var channel = ChannelManager.CreateChannel<string>(
|
|
buffersize: MAX_PENDING_UPLOADS,
|
|
pendingWritersOverflowStrategy: QueueOverflowStrategy.LIFO
|
|
);
|
|
|
|
var task = AutomationExtensions.RunTask(
|
|
channel.AsRead(),
|
|
|
|
async (chan) =>
|
|
{
|
|
while (true)
|
|
{
|
|
var f = await chan.ReadAsync();
|
|
|
|
try
|
|
{
|
|
if (File.Exists(f))
|
|
{
|
|
int rc;
|
|
using (var fs = File.OpenRead(f))
|
|
{
|
|
if (fs.Length > 0)
|
|
{
|
|
using var request = new HttpRequestMessage(HttpMethod.Post, UPLOAD_URL);
|
|
|
|
request.Content = new StreamContent(fs);
|
|
|
|
using var timeoutToken = new CancellationTokenSource();
|
|
timeoutToken.CancelAfter(TimeSpan.FromSeconds(UPLOAD_OPERATION_TIMEOUT_SECONDS));
|
|
|
|
using var client = HttpClientHelper.CreateClient();
|
|
// Explicitly keeping the default 100s timeout
|
|
using var response = await client.UploadStream(request, timeoutToken.Token);
|
|
rc = (int)response.StatusCode;
|
|
}
|
|
else
|
|
rc = 200;
|
|
}
|
|
|
|
if (rc >= 200 && rc <= 299)
|
|
File.Delete(f);
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Logging.Log.WriteErrorMessage(LOGTAG, "UploadFailed", ex, "UsageReporter failed");
|
|
}
|
|
}
|
|
}
|
|
);
|
|
|
|
return new Tuple<Task, IWriteChannel<string>>(task, channel);
|
|
}
|
|
|
|
}
|
|
}
|
|
|