462 lines
14 KiB
C#
462 lines
14 KiB
C#
using System.Collections.Concurrent;
|
||
using System.Collections.ObjectModel;
|
||
using System.ComponentModel;
|
||
using System.Diagnostics;
|
||
using System.IO;
|
||
using System.Windows;
|
||
using System.Windows.Data;
|
||
using System.Windows.Input;
|
||
using HeBianGu.Base.WpfBase;
|
||
using HeBianGu.Service.Mvc;
|
||
using Hopetry.Models;
|
||
using Hopetry.Provider;
|
||
using Hopetry.Services;
|
||
using Path = System.IO.Path;
|
||
|
||
namespace Hopetry.ViewModel.Send;
|
||
|
||
[ViewModel("Down")]
|
||
public class DownViewModel : MvcViewModelBase
|
||
{
|
||
private readonly MinioService _minioService;
|
||
|
||
public RelayCommand<MinioDownloadTask> OpenDownItemFolder { get; set; }
|
||
public RelayCommand<MinioDownloadTask> CancelDownload { get; set; }
|
||
private readonly ConcurrentQueue<MinioDownloadTask> _taskQueue = new();
|
||
private SemaphoreSlim _semaphore;
|
||
|
||
private CancellationTokenSource _processingCts = new();
|
||
public int count { get; set; }
|
||
private readonly ReaderWriterLockSlim _lock = new();
|
||
|
||
private Visibility _tab0Visibility;
|
||
|
||
public Visibility Tab0Visibility
|
||
{
|
||
get => _tab0Visibility;
|
||
set
|
||
{
|
||
_tab0Visibility = value;
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
private Visibility _tab1Visibility;
|
||
|
||
public Visibility Tab1Visibility
|
||
{
|
||
get => _tab1Visibility;
|
||
set
|
||
{
|
||
_tab1Visibility = value;
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
private int _tabIndex;
|
||
|
||
public int TabIndex
|
||
{
|
||
get => _tabIndex;
|
||
set
|
||
{
|
||
_tabIndex = value;
|
||
if (_tabIndex == 0)
|
||
{
|
||
Tab0Visibility = Visibility.Visible;
|
||
Tab1Visibility = Visibility.Hidden;
|
||
}
|
||
else if (_tabIndex == 1)
|
||
{
|
||
Tab0Visibility = Visibility.Hidden;
|
||
Tab1Visibility = Visibility.Visible;
|
||
}
|
||
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 全部任务动态标题
|
||
/// </summary>
|
||
private string _allTaskHeader;
|
||
|
||
public string AllTaskHeader
|
||
{
|
||
get => _allTaskHeader;
|
||
set
|
||
{
|
||
_allTaskHeader = value;
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
private string _runningTaskHeader;
|
||
|
||
public string RunningTaskHeader
|
||
{
|
||
get => _runningTaskHeader;
|
||
set
|
||
{
|
||
_runningTaskHeader = value;
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
private string _finishedTaskHeader;
|
||
|
||
public string FinishedTaskHeader
|
||
{
|
||
get => _finishedTaskHeader;
|
||
set
|
||
{
|
||
_finishedTaskHeader = value;
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
// 绑定属性
|
||
//public ObservableCollection<MinioDownloadTask> AllTasks { get; set; }
|
||
private ObservableCollection<MinioDownloadTask> _runningTasks;
|
||
|
||
public ObservableCollection<MinioDownloadTask> RunningTasks
|
||
{
|
||
get => _runningTasks;
|
||
set
|
||
{
|
||
_runningTasks = value;
|
||
RaisePropertyChanged(); // 触发INotifyPropertyChanged通知
|
||
}
|
||
}
|
||
|
||
private ObservableCollection<MinioDownloadTask> _finishedTasks;
|
||
|
||
public ObservableCollection<MinioDownloadTask> FinishedTasks
|
||
{
|
||
get => _finishedTasks;
|
||
set
|
||
{
|
||
_finishedTasks = value;
|
||
RaisePropertyChanged(); // 触发INotifyPropertyChanged通知
|
||
}
|
||
}
|
||
|
||
private CollectionViewSource _finishedTasksViewSource;
|
||
|
||
public CollectionViewSource FinishedTasksViewSource
|
||
{
|
||
get => _finishedTasksViewSource;
|
||
set
|
||
{
|
||
_finishedTasksViewSource = value;
|
||
RaisePropertyChanged(); // 触发INotifyProperty
|
||
}
|
||
}
|
||
|
||
private int _maxConcurrent = 3;
|
||
|
||
public int MaxConcurrent
|
||
{
|
||
get => _maxConcurrent;
|
||
set
|
||
{
|
||
if (value < 1) value = 1;
|
||
if (_maxConcurrent == value) return;
|
||
|
||
_maxConcurrent = value;
|
||
//AdjustConcurrency();
|
||
RaisePropertyChanged();
|
||
}
|
||
}
|
||
|
||
public ICommand ClearFinishedCommand { get; }
|
||
public RelayCommand<MinioDownloadTask> StartOrPauseDown { get; }
|
||
public ICommand PauseAllTask { get; }
|
||
|
||
public ICommand StopAllTask { get; }
|
||
|
||
/// <summary>
|
||
/// 构造函数
|
||
/// </summary>
|
||
/// <param name="minioService"></param>
|
||
public DownViewModel(MinioService minioService)
|
||
{
|
||
_minioService = minioService;
|
||
FinishedTasksViewSource = new CollectionViewSource { Source = FinishedTasks };
|
||
FinishedTasksViewSource.SortDescriptions.Add(new SortDescription("FinishedTime", ListSortDirection.Descending));
|
||
Console.WriteLine("初始化DownViewModel");
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
OpenDownItemFolder = new RelayCommand<MinioDownloadTask>(DoOpenDownItemFolder);
|
||
// 取消下载
|
||
CancelDownload = new RelayCommand<MinioDownloadTask>(DoCancelDownload);
|
||
// 暂停下载或者开启下载
|
||
StartOrPauseDown = new RelayCommand<MinioDownloadTask>(DoStartOrPauseDown);
|
||
// 全部暂停
|
||
PauseAllTask = new CustomCommand(DoPauseAllTask);
|
||
StopAllTask = new CustomCommand(DoStopAllTask);
|
||
// 全部取消
|
||
LoadFinishedTasks();
|
||
LoadRunningTasks();
|
||
ClearFinishedCommand = new CustomCommand(DoClearFinishedCommand);
|
||
// 启动任务处理线程
|
||
_semaphore = new SemaphoreSlim(MaxConcurrent);
|
||
new Thread(ProcessTasksLoop) { IsBackground = true }.Start();
|
||
}
|
||
|
||
// 停止所有任务
|
||
private void DoStopAllTask()
|
||
{
|
||
foreach (var minioDownloadTask in RunningTasks)
|
||
{
|
||
_taskQueue.Clear();
|
||
//Console.WriteLine("取消下载");
|
||
|
||
// 取消操作
|
||
if (minioDownloadTask.StopDownTs != null)
|
||
{
|
||
minioDownloadTask.StopDownTs.Cancel();
|
||
}
|
||
minioDownloadTask.Status = "删除中";
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
client.Deleteable<MinioDownloadTask>().Where(x => x.TaskId == minioDownloadTask.TaskId).ExecuteCommand();
|
||
}
|
||
|
||
RunningTasks.Clear();
|
||
RefreshHeader();
|
||
MessageProxy.Snacker.Show("取消下载成功");
|
||
}
|
||
|
||
// 暂停所有任务
|
||
private void DoPauseAllTask()
|
||
{
|
||
foreach (var minioDownloadTask in RunningTasks)
|
||
{
|
||
//暂停下载
|
||
if (minioDownloadTask.StopDownTs != null)
|
||
{
|
||
minioDownloadTask.StopDownTs.Cancel();
|
||
}
|
||
// 清空下载队列
|
||
_taskQueue.Clear();
|
||
minioDownloadTask.Status = "已暂停";
|
||
minioDownloadTask.StartOrPauseIcon = "\xe748";
|
||
// 速度及剩余时间（视图显示信息）
|
||
minioDownloadTask.Speed = "已暂停";
|
||
|
||
}
|
||
}
|
||
|
||
private void DoStartOrPauseDown(MinioDownloadTask item)
|
||
{
|
||
if (item.Status == "下载中" || item.Status == "等待中")
|
||
{
|
||
if (item.StopDownTs != null)
|
||
{
|
||
//暂停下载
|
||
item.StopDownTs.Cancel();
|
||
}
|
||
item.Status = "已暂停";
|
||
item.StartOrPauseIcon = "\xe748";
|
||
// 速度及剩余时间（视图显示信息）
|
||
item.Speed = "已暂停";
|
||
// 避免结构函数加载的任务报错
|
||
|
||
}
|
||
else if (item.Status == "已暂停")
|
||
{
|
||
item.Status = "等待中";
|
||
// 更新排队时，下载信息
|
||
item.Speed = "等待中";
|
||
item.StartOrPauseIcon = "\xe76e";
|
||
// 加入下载等待队列
|
||
_taskQueue.Enqueue(item);
|
||
}
|
||
}
|
||
|
||
private void DoClearFinishedCommand()
|
||
{
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
client.Deleteable<MinioDownloadTask>().Where(x => x.Status == "已完成").ExecuteCommand();
|
||
FinishedTasks.Clear();
|
||
FinishedTaskHeader = $"已完成（0）";
|
||
}
|
||
|
||
public void RefreshHeader()
|
||
{
|
||
RunningTaskHeader = $"下载中（{RunningTasks.Count}）";
|
||
FinishedTaskHeader = $"已完成（{FinishedTasks.Count}）";
|
||
}
|
||
|
||
public void LoadRunningTasks()
|
||
{
|
||
// todo 队列中有几种任务状态 数据库中应该有几种任务状态
|
||
//
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
var data = client.Ado
|
||
.SqlQuery<MinioDownloadTask>(
|
||
$"select * from download_task where status='下载中' or status='等待中' or status='已暂停' order by task_id desc")
|
||
.Select(x => new MinioDownloadTask
|
||
{
|
||
Status = "已暂停",
|
||
TaskId = x.TaskId,
|
||
FileName = x.FileName,
|
||
Bucket = x.Bucket,
|
||
ObjectKey = x.ObjectKey,
|
||
TotalSize = x.TotalSize,
|
||
Downloaded = x.Downloaded,
|
||
FilePath = x.FilePath,
|
||
CreateTime = x.CreateTime,
|
||
FinishedTime = x.FinishedTime,
|
||
FileSize = x.FileSize,
|
||
FileETag = x.FileETag,
|
||
// 初始化显示
|
||
DownloadInfo =
|
||
(x.Downloaded > 1048576 ? $"{x.Downloaded / 1048576:f2}MB" : $"{x.Downloaded / 1024:f2}KB") +
|
||
$"/{x.FileSize}",
|
||
Speed = "已暂停",
|
||
StartOrPauseIcon = "\xe748"
|
||
}).ToList();
|
||
RunningTasks = new ObservableCollection<MinioDownloadTask>(data);
|
||
RunningTaskHeader = $"下载中（{RunningTasks.Count}）";
|
||
}
|
||
|
||
public void LoadFinishedTasks()
|
||
{
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
var data = client.Ado
|
||
.SqlQuery<MinioDownloadTask>(
|
||
$"select * from download_task where status='已完成' order by datetime(finished_time) desc");
|
||
FinishedTasks = new ObservableCollection<MinioDownloadTask>(data);
|
||
FinishedTaskHeader = $"已完成（{FinishedTasks.Count}）";
|
||
}
|
||
|
||
private async void ProcessTasksLoop()
|
||
{
|
||
while (!_processingCts.IsCancellationRequested)
|
||
{
|
||
if (_taskQueue.TryDequeue(out var task))
|
||
{
|
||
// todo 这里有bug, 如何解决？？？
|
||
/*if (task.Status == "已暂停")
|
||
{
|
||
// 跳过暂停，但之前已加入队列
|
||
continue;
|
||
}*/
|
||
|
||
Console.WriteLine("存在可下载任务，正在申请许可...");
|
||
await _semaphore.WaitAsync();
|
||
Console.WriteLine("申请下载许可成功！马上开启下载");
|
||
// 启动新线程执行下载任务
|
||
// _ = Task.Run(() => ExecuteTaskAsync(task));
|
||
_ = ExecuteTaskAsync(task).ContinueWith(_ => _semaphore.Release());
|
||
}
|
||
|
||
// 无任务时，进入低功耗轮询
|
||
Thread.Sleep(500);
|
||
}
|
||
}
|
||
|
||
public void AddTask(string bucketName, string objectKey, string size, string downDir)
|
||
{
|
||
var task = new MinioDownloadTask(_minioService, bucketName, objectKey, downDir, size);
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
//todo 锁库问题 下载多线程冲突
|
||
client.Insertable(task).ExecuteCommandIdentityIntoEntity();
|
||
Application.Current.Dispatcher.Invoke(() =>
|
||
{
|
||
RunningTasks.Add(task);
|
||
RefreshHeader();
|
||
});
|
||
Console.WriteLine($"文件大小：{size} ");
|
||
_taskQueue.Enqueue(task);
|
||
}
|
||
|
||
|
||
private async Task ExecuteTaskAsync(MinioDownloadTask task)
|
||
{
|
||
try
|
||
{
|
||
// minio中好像自带重试机制
|
||
// 补充
|
||
if (task.Minio == null)
|
||
{
|
||
task.Minio = _minioService;
|
||
}
|
||
|
||
await task.StartDownload();
|
||
Application.Current.Dispatcher.Invoke(() =>
|
||
{
|
||
RunningTasks.Remove(task);
|
||
FinishedTasks.Add(task);
|
||
RefreshHeader();
|
||
});
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
// 异常释放信号源
|
||
//_semaphore.Release();
|
||
Console.WriteLine($"下载失败或取消：{ex.Message}");
|
||
}
|
||
finally
|
||
{
|
||
if (task.Status.Equals("删除中"))
|
||
{
|
||
Console.WriteLine($"删除文件：{task.FileName}");
|
||
File.Delete(Path.Combine(task.FilePath, task.FileName));
|
||
Console.WriteLine($"{task.FileName}删除成功");
|
||
}
|
||
}
|
||
}
|
||
|
||
private void AdjustConcurrency()
|
||
{
|
||
lock (_semaphore)
|
||
{
|
||
// 调整信号量容量
|
||
var delta = _maxConcurrent - _semaphore.CurrentCount;
|
||
if (delta > 0)
|
||
{
|
||
for (int i = 0; i < delta; i++)
|
||
{
|
||
_semaphore.Release();
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
public void DoOpenDownItemFolder(MinioDownloadTask para)
|
||
{
|
||
//Console.WriteLine($"点击item值：{JsonConvert.SerializeObject(para)}");
|
||
//Process.Start("explorer.exe", para.FilePath);
|
||
var file = Path.Combine(para.FilePath, para.FileName);
|
||
Process.Start("explorer.exe", $"/select,\"{file}\"");
|
||
}
|
||
|
||
public void DoCancelDownload(MinioDownloadTask item)
|
||
{
|
||
Console.WriteLine("取消下载");
|
||
|
||
// 取消操作
|
||
if (item.StopDownTs != null)
|
||
{
|
||
item.StopDownTs.Cancel();
|
||
}
|
||
item.Status = "删除中";
|
||
using var client = SqlSugarConfig.GetSqlSugarScope();
|
||
client.Deleteable<MinioDownloadTask>().Where(x => x.TaskId == item.TaskId).ExecuteCommand();
|
||
RunningTasks.Remove(item);
|
||
RefreshHeader();
|
||
// 删除下载文件 放到下载里面了
|
||
//File.Delete(Path.Combine(item.FilePath, item.FileName));
|
||
MessageProxy.Snacker.Show($"<{item.FileName}>取消下载");
|
||
}
|
||
|
||
|
||
protected override void Init()
|
||
{
|
||
}
|
||
|
||
protected override void Loaded(string args)
|
||
{
|
||
}
|
||
} |