自己写了一个多线程的工作队列,能够实现对队列中对象的自动处理。多线程添加元素到队列中,队列根据绑定
的事件进行自动处理,可以设置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);
}
}
}