一个C#多线程的工作队列

 

    自己写了一个多线程的工作队列,能够实现对队列中对象的自动处理。多线程添加元素到队列中,队列根据绑定

的事件进行自动处理,可以设置WorkSequential属性来实现对队列处理的单线程(严格顺序处理)或者多线程处理(循序出队,但是

多线程处理,不保证对队列元素的处理顺利)的选择。

 

  源码:

 

代码
/***********多线程的工作队列***************
* 此工作队列保证线程安全性
*
*
*
*
* ******
*/
namespace WorkQueue
{
using System.Collections.Generic;
using System;
using System.Threading;

public delegate void UserWorkEventHandler<T>(object sender, WorkQueue<T>.EnqueueEventArgs e);
public class WorkQueue<T>
{
private bool IsWorking; //表明处理线程是否正在工作
private object lockIsWorking = new object();//对IsWorking的同步对象


private Queue<T> queue; //实际的队列
private object lockObj = new object(); //队列同步对象





/// <summary>
/// 绑定用户需要对队列中的item对象
/// 施加的操作的事件
/// </summary>
public event UserWorkEventHandler<T> UserWork;

public WorkQueue(int n)
{
queue
= new Queue<T>(n);
}

public WorkQueue()
{
queue
= new Queue<T>();
}

/// <summary>
/// 谨慎使用此函数,
/// 只保证此瞬间,队列值为空
/// </summary>
/// <returns></returns>
public bool IsEmpty()
{
lock (lockObj)
{
return queue.Count == 0;
}
}

private bool isOneThread;

/// <summary>
/// 队列处理是否需要单线程顺序执行
/// ture表示单线程处理队列的T对象
/// 默认为false,表明按照顺序出队,但是多线程处理item
/// *****注意不要频繁改变此项****
/// </summary>
public bool WorkSequential
{
get
{
return isOneThread;
}
set
{
isOneThread
= value;
}

}



/// <summary>
/// 向工作队列添加对象,
/// 对象添加以后,如果已经绑定工作的事件
/// 会触发事件处理程序,对item对象进行处理
/// </summary>
/// <param name="item">添加到队列的对象</param>
public void EnqueueItem(T item)
{
lock (lockObj)
{
queue.Enqueue(item);
}

lock (lockIsWorking)
{
if (!IsWorking)
{
IsWorking
= true;
ThreadPool.QueueUserWorkItem(doUserWork);
}
}



}


/// <summary>
/// 处理队列中对象的函数
/// </summary>
/// <param name="o"></param>
private void doUserWork(object o)
{
try
{
T item;

while (true)
{
lock (lockObj)
{
if (queue.Count > 0)
{
item
= queue.Dequeue();
}
else
{
return;
}
}
if (!item.Equals(default(T)))
{

if (isOneThread)
{
if (UserWork != null)
{
UserWork(
this, new EnqueueEventArgs(item));
}
}
else
{
ThreadPool.QueueUserWorkItem(obj
=>
{
if (UserWork != null)
{
UserWork(
this, new EnqueueEventArgs(obj));
}
}, item);
}


}

}
}
finally
{
lock (lockIsWorking)
{
IsWorking
= false;
}

}
}

/// <summary>
/// UserWork事件的参数,包含item对象
/// </summary>
public class EnqueueEventArgs : EventArgs
{
public T Item { get; private set; }
public EnqueueEventArgs(object item)
{
try
{
Item
= (T)item;
}
catch (Exception)
{

throw new InvalidCastException("object to T 转换失败");
}
}
}
}
}

 

 

 

 

  示例代码:
代码
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.IO;
using WorkQueue;
namespace Program
{
class Program
{
private static List<string> list=new List<string>(1000);
static StreamWriter sw = new StreamWriter(new FileStream("test.dat", FileMode.Create));
static void Main(string[] args)
{
WorkQueue
<int> workQueue=new WorkQueue<int>(1000);
workQueue.UserWork
+= new UserWorkEventHandler<int>(workQueue_UserWork);
// workQueue.WorkSequential = true;
ThreadPool.QueueUserWorkItem(o =>
{
for (int i = 0; i < 1000; i++)
{
workQueue.EnqueueItem(i);

}
});
Console.ReadLine();

list.ForEach(str
=>sw.WriteLine(str));
Console.WriteLine(workQueue.IsEmpty());
sw.Close();
}

static void workQueue_UserWork(object sender, WorkQueue<int>.EnqueueEventArgs e)
{

StringBuilder sb
=new StringBuilder();
sb.Append(e.Item).Append(
"\t\t").Append(DateTime.Now.ToString("u")+"\t\t").Append(Thread.CurrentThread.ManagedThreadId);
list.Add(sb.ToString());
Thread.Sleep(
15);
}
}
}

 

posted on 2010-07-09 20:11  wota  阅读(25617)  评论(5编辑  收藏  举报