1

我在下面有一个简单的程序，它有 2 个线程执行某些任务。

Thread1 是数据馈送器。Thread2 是数据处理器。

到目前为止，通过我的方法完成的工作正在发挥作用，但我希望有更好的方式在工作完成时获得通知

这是代码

class Program 
{
    private static BlockingCollection<int> _samples = new BlockingCollection<int>();
    private static CancellationTokenSource _cancellationTokenSource = new CancellationTokenSource();
    private static bool _cancel;


    static void Main(string[] args)
    {
        ThreadStart thread1 = delegate
        {
            ProcessThread1();
        };
        new Thread(thread1).Start();

        ThreadStart thread2 = delegate
        {
            ProcessThread2();
        };
        new Thread(thread2).Start();

        Console.WriteLine("Press any key to cancel..");

        Console.Read();

        _cancel = true;
        _cancellationTokenSource.Cancel();

        Console.Read();

    }

    private static void ProcessThread1()
    {
        for (int i = 0; i < 10; i++)
        {
            if (_cancel)
            {
                break;
            }

            Console.WriteLine("Adding data..");

            _samples.TryAdd(i,100);

            Thread.Sleep(1000);
        }

        // I dont like this. Instead can I get notified in the UI thread that this thread is complete.
        _cancel = true;
        _cancellationTokenSource.Cancel();

    }

    private static void ProcessThread2()
    {
        while (!_cancellationTokenSource.IsCancellationRequested)
        {
            int data;

            if (_samples.TryTake(out data, 100)) 
            {
                // Do some work.    
                Console.WriteLine("Processing data..");
            }

        }

        Console.WriteLine("Cancelled.");
    }


}

如果用户请求取消或工作完成，我希望程序退出。

我不确定当 ProcessThread1 用完时如何得到通知。目前我在工作完成时设置cancel = true，但似乎不对。任何帮助表示赞赏。

4

2 回答 2

2

如果您使用Task而不是手动创建线程，则可以在任务上附加一个延续，以通知您的 UI 工作已完成。

Task workOne = Task.Factory.StartNew( () => ProcessThread1());

workOne.ContinueWith(t =>
{
     // Update UI here
}, TaskScheduler.FromCurrentSynchronizationContext());

使用 .NET 4.5，这变得更加容易，因为您可以使用新的async语言支持：

var workOne = Task.Run(ProcessThread1);
var workTwo = Task.Run(ProcessThread2);

// asynchronously wait for both tasks to complete...
await Task.WhenAll(workOne, workTwo);

// Update UI here.

请注意，这两者都是在设计时考虑到用户界面 - 并且在控制台应用程序中会表现异常，因为控制台应用程序中没有当前的同步上下文。当您将其移至真正的用户界面时，它将正常运行。

于 2013-08-23T18:37:13.463 回答
1

再启动一个线程，它的唯一工作就是等待控制台输入：

private void ConsoleInputProc()
{
    Console.Write("Press Enter to cancel:");
    Console.ReadLine();
    _cancellationTokenSource.Cancel();
}

然后您的主线程启动两个处理线程和输入线程。

// create and start the processing threads
Thread t1 = new Thread(thread1);
Thread t2 = new Thread(thread2);
t1.Start();
t2.Start();

// create and start the input thread
Thread inputThread = new Thread(ConsoleInputProc);
inputThread.Start();

然后，您等待两个处理线程：

t1.Join();
// first thread finished. Request cancellation.
_cancellationTokenSource.Cancel();
t2.Join();

因此，如果用户按下 Enter，则输入线程会设置取消标志。thread1 和 thread2 都看到取消请求并退出。

如果thread1完成了它的工作，那么主线程设置取消标志，thread2将取消。

在任何一种情况下，程序都不会退出，直到线程 2 退出。

无需显式终止输入线程。程序退出时它会死掉。

顺便说一句，我会从线程 1 proc 中删除这些行：

    // I dont like this. Instead can I get notified in the UI thread that this thread is complete.
    _cancel = true;
    _cancellationTokenSource.Cancel();

我将_cancel完全删除该变量，并IsCancellationRequested像第二个线程一样让第一个线程检查。

不幸的是，您必须启动一个专用线程来等待控制台输入，但这是我所知道的完成此操作的唯一方法。Windows 控制台似乎没有可等待事件。

请注意，您可以使用 来做同样的事情Task，总体上更容易使用。任务执行的代码是相同的。

更新

纵观全局，我发现您有一个典型的生产者/消费者设置，带有BlockingCollection. 您可以使您的生产者和消费者线程更清洁：

private static void ProcessThread1()
{
    for (int i = 0; i < 10; i++)
    {
        Console.WriteLine("Adding data..");
        _samples.TryAdd(i, Timeout.Infinite, _cancellationTokenSource.Token);
        // not sure why the sleep is here
        Thread.Sleep(1000);
    }
    // Marks the queue as complete for adding.
    // When the queue goes empty, the consumer will know that
    // no more data is forthcoming.
    _samples.CompleteAdding();
}

private static void ProcessThread2()
{
    int data;
    while (_samples.TryTake(out data, TimeSpan.Infinite, _cancellationTokenSource.Token))
    {
        // Do some work.    
        Console.WriteLine("Processing data..");
    }

    Console.WriteLine("Cancelled.");
}

你仍然需要那个输入线程（除非你想在 上旋转一个循环Console.KeyAvailable），但这大大简化了你的生产者和消费者。

于 2013-08-23T20:33:45.490 回答