using System.Configuration; using System.Windows; using System.Windows.Input; using HeBianGu.Base.WpfBase; using Hopetry.Models; using Hopetry.Provider; using SqlSugar; namespace Hopetry.Services; using System; using System.IO; using System.Net.Http; using System.Threading; using System.Threading.Tasks; using System.ComponentModel; using Minio; [SugarTable(TableName = "download_task")] public class MinioDownloadTask : INotifyPropertyChanged { private readonly MinioService _minio; /// /// /// private CancellationTokenSource _cts; /// /// /// private ManualResetEventSlim _pauseEvent = new(true); // 任务属性(绑定到UI) [SugarColumn(ColumnName = "task_id", IsPrimaryKey = true, IsIdentity = true)] public int TaskId { get; set; } [SugarColumn(ColumnName = "file_name")] public string FileName { get; set; } [SugarColumn(ColumnName = "bucket_name")] public string Bucket { get; set; } [SugarColumn(ColumnName = "object_key")] public string ObjectKey { get; set; } [SugarColumn(ColumnName = "total_size")] public long TotalSize { get; private set; } [SugarColumn(ColumnName = "downloaded")] public long Downloaded { get; private set; } [SugarColumn(ColumnName = "file_path")] public string FilePath { get; set; } [SugarColumn(ColumnName = "status")] public string Status { get; set; } = "等待中"; [SugarColumn(IsIgnore = true)] public string Progress { get { return TotalSize == 0 ? "0%" : $"{(Downloaded * 100 / TotalSize):0.0}%"; } } [SugarColumn(IsIgnore = true)] public ICommand PauseCommand { get; } [SugarColumn(IsIgnore = true)] public ICommand CancelCommand { get; } public MinioDownloadTask() { PauseCommand = new CustomCommand(OnPause); CancelCommand = new CustomCommand(OnCancel); } public MinioDownloadTask(MinioService minio, string bucket, string objectKey, string downDir) { Status = "等待中"; _minio = minio; Bucket = bucket; ObjectKey = objectKey; FileName = Path.GetFileName(objectKey); FilePath = downDir; PauseCommand = new CustomCommand(OnPause); CancelCommand = new CustomCommand(OnCancel); } /// /// 下载 /// /// public async Task StartDownload() { _cts = new CancellationTokenSource(); try { Status = "下载中"; var updateTask = new MinioDownloadTask { TaskId = TaskId, Status = Status }; using (var client = SqlSugarConfig.GetSqlSugarScope()) { await client.Updateable(updateTask).IgnoreNullColumns().ExecuteCommandAsync(); } Console.WriteLine($"id {TaskId} path: {FilePath} key: {ObjectKey}文件下载中..."); updateTask.Status = "下载中"; await _minio.DownLoadObject(Bucket, ObjectKey, FilePath, ""); Status = "已完成"; updateTask.Status = Status; using (var client = SqlSugarConfig.GetSqlSugarScope()) { await client.Updateable(updateTask).IgnoreNullColumns().ExecuteCommandAsync(); } Console.WriteLine($"文件{ObjectKey}下载完成"); } catch (OperationCanceledException) { Status = "已取消"; } catch (Exception ex) { Status = $"错误:{ex.Message}"; } finally { _pauseEvent.Dispose(); OnPropertyChanged(nameof(Status)); } } private void OnPause() { if (Status == "下载中") { _pauseEvent.Reset(); Status = "已暂停"; } else if (Status == "已暂停") { _pauseEvent.Set(); Status = "下载中"; } OnPropertyChanged(nameof(Status)); } private void OnCancel() { _cts?.Cancel(); Status = "已取消"; OnPropertyChanged(nameof(Status)); } public event PropertyChangedEventHandler PropertyChanged; protected void OnPropertyChanged(string propertyName) => PropertyChanged?.Invoke(this, new PropertyChangedEventArgs(propertyName)); }