c#多线程同步之EventWaitHandle再次使用

 1     /// <summary>
 2     /// 文件传输器,用来获取全文文件,自动根据全文文件数量,开启一定数量的线程,采用生产者消费模式
 3     /// </summary>
 4     public class FileTranser
 5     {
 6         private static IFileTranser Transer = new RealFileTranser();
 7         // 文件队列
 8         static Queue<FullTextListViewModel> FileTaskQueue = new Queue<FullTextListViewModel>();
 9         // 为保证线程安全,使用一个锁来保护队列的访问
10         readonly static object locker = new object();
11         // 通过 autoResetEvent 给工作线程发信号
12         static EventWaitHandle autoResetEvent = new AutoResetEvent(false);
13         public static CancellationTokenSource cancel = null;
14 
15         static int MaxThreadCount = 5;
16         static int MinThreadCount = 1;
17         static int PageSize = 100;
18 
19         static List<Thread> threads = new List<Thread>();
20         static bool IsRunning = false;
21         public static void Start(FullTextListViewModel model, int total)
22         {
23             if (cancel != null && cancel.IsCancellationRequested) return;
24             if (IsRunning == false || threadCount > threads.Count)
25             {
26                 int startCount = threadCount - threads.Count;
27                 IsRunning = true;
28                 // 任务开始,启动工作线程
29                 for (int i = 0; i < startCount; i++)
30                 {
31                     Thread t = new Thread(Work);
32                     t.Name = "f_" + (i + 1);
33                     t.IsBackground = false;
34                     t.Start();
35                     threads.Add(t);
36                 }
37             }
38             lock (locker)
39             {
40                 FileTaskQueue.Enqueue(model);
41             }
42             autoResetEvent.Set();
43         }
44         public static void Cancel()
45         {
46             IsRunning = false;
47             threads.Clear();
48             if (cancel != null)
49             {
50                 cancel.Cancel();
51             }
52             autoResetEvent.Set();
53             Logger.Log("取消获取全文的线程执行");
54         }
55         /// <summary>执行工作</summary>
56         static void Work()
57         {
58             while (true)
59             {
60                 if (cancel.IsCancellationRequested)
61                 {
62                     Logger.Log("线程已取消,当前线程:" + Thread.CurrentThread.Name);
63                     autoResetEvent.Set();
64                     break;
65                 }
66                 else
67                 {
68                     FullTextListViewModel model = null;
69                     lock (locker)
70                     {
71                         if (FileTaskQueue.Count > 0)
72                         {
73                             model = FileTaskQueue.Dequeue();
74                         }
75                     }
76                     if (model != null)
77                     {
78                         Logger.Log("全文传输文件已经开始,当前Id:" + model.ClientTaskId + ",当前线程:" + Thread.CurrentThread.Name);
79                         FileTranser.Transer.DoWork(model);
80                     }
81                     else
82                     {
83                         Logger.Log("线程" + Thread.CurrentThread.Name + ",已被阻塞,等待任务");
84                         autoResetEvent.WaitOne();
85                     }
86                 }
87             }
88         }
89     }

     这段不到100行的代码,采用的思想是,生产者消费模式,其中应用了AutoResetEvent ,从字面上看,是自动重置事件,它是EventWaitHandle的一个子类。

     我们还是先来看看这段代码所要表达的意思。第8行,定义了一个文件传输队列FileTaskQueue,它用来接收生产者生产的实体,即FullTextListViewModel类型的对象。29-35行,开启了若干个线程,40行,给队列加锁,把model放到队列里,紧接着,通过Set方法,释放被阻塞的线程。此时,如果有三个线程,队列里只有1个model,那么三个线程消费一个model,必然有两个线程直接被阻塞了,那个得到model的线程执行完毕后,也被阻塞了,一直到下一个model进队,再次运行Set方法, 它们当中只有一个线程被解除了阻塞,接着干活,其它线程的继续等待。如果有足够多的model进队了,意味着活多了,这三个线程都会忙碌起来,不会阻塞。

     autoResetEvent,每次调用set方法时,只有一个线程被释放。这点很关键,否则在设计多线程的程序时,会有意想不到的结果。

 

posted @ 2017-05-18 18:43  micDavid  阅读(1264)  评论(0编辑  收藏  举报