Mirrored_Repos/wabbajack
1
0
Fork 0
mirror of https://github.com/wabbajack-tools/wabbajack.git synced 2024-08-30 18:42:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Some WorkQueue refactoring

Browse Source
This commit is contained in:
Justin Swanson 2020-01-06 23:50:04 -06:00
parent 98cb1ea28e
commit a0328fef93
1 changed files with 5 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Wabbajack.Common/WorkQueue.cs
View File

@ -12,8 +12,7 @@ namespace Wabbajack.Common
{
public class WorkQueue : IDisposable
{
internal BlockingCollection<Func<Task>>
Queue = new BlockingCollection<Func<Task>>(new ConcurrentStack<Func<Task>>());
internal BlockingCollection<Func<Task>> Queue = new BlockingCollection<Func<Task>>(new ConcurrentStack<Func<Task>>());
public const int UnassignedCpuId = 0;
@ -36,15 +35,12 @@ namespace Wabbajack.Common
// implement log messages in a non-singleton fashion, they will already be wired up properly.
public IObservable<IStatusMessage> LogMessages => Utils.LogMessages;
public WorkQueue(int threadCount = 0)
{
StartThreads(threadCount == 0 ? Environment.ProcessorCount : threadCount);
}
public int ThreadCount { get; private set; }
private void StartThreads(int threadCount)
public WorkQueue(int? threadCount = null)
{
ThreadCount = threadCount;
Threads = Enumerable.Range(1, threadCount)
ThreadCount = threadCount ?? Environment.ProcessorCount;
Threads = Enumerable.Range(1, ThreadCount)
.Select(idx =>
{
var thread = new Thread(() => ThreadBody(idx).Wait());
@ -56,8 +52,6 @@ namespace Wabbajack.Common
}).ToList();
}
public int ThreadCount { get; private set; }
private async Task ThreadBody(int idx)
{
_cpuId.Value = idx;