一个自定义线程池的小Demo

时间:2023-03-09 06:25:53
一个自定义线程池的小Demo

在项目中如果是web请求时候,IIS会自动分配一个线程来进行处理,如果很多个应用程序共享公用一个IIS的时候,线程分配可能会出现一个问题(当然也是我的需求造成的)

之前在做项目的时候,有一个需求,就是当程序启动的时候,希望能够启动一定数目的线程,然后每一个线程始终都是在运行的状态,不进行释放,然后循环去做一些事情。那么IIS的线程管理可能就不是我想要的,因为我想我的一些程序,只用我开启的线程来做工作。也就是说我想模拟一个线程池,每次有一个调用的时候从自定义线程池中取出一个,用完再放回去。

谈谈我的思路:

1.程序一启动就通过for循环来创建,一定数目的线程(这个数目是可以配置的)

2.至少要有三个容器来存储线程,分别是工作线程队列和空闲线程队列以及等待队列

3.使用线程中的AutoResetEvent类,初始每一个线程都是unsignaled状态,线程一启动就一直在循环调用WaitOne()方法,那么每次外部调用的时候,都调用一次这个类实例对象的set,线程然后就可以继续做下面的工作了。

4.至少两个方法:

第一个开放给外部,让外部的方法能够被传入执行,然后这个方法能够判断空闲队列,等待队列,以及工作队列的状态,如果传入的时候发现,空闲队列有空闲的线程就直接,将任务委托给空闲队列的一个线程执行,否则把它放到等待队列。

第二个方法,需要能够将工作完成的线程从工作队列移动到空闲队列,然后判断一下等待队列是不是有任务,有的话就交给空闲队列里面的线程来执行。

体思路如上,可以试试先写一下。

1.因为每个线程都有一个AutoResetEvent的实例,所以最好把Thread进行封装,变成我们自己的Thread。

  public class Task
{
#region Variable
//一个AutoResetEvent实例
private AutoResetEvent _locks = new AutoResetEvent(false);
//一个Thread实例
private Thread _thread;
// 绑定回调方法,就是外部实际执行的任务
public Action _taskAction; //定义一个事件用来绑定工作完成后的操作,也就是4中所说的工作队列向空闲队列移动
public event Action<Task> WorkComplete; /// <summary>
///设置线程拥有的Key
/// </summary>
public string Key { get; set; } #endregion //线程需要做的工作
private void Work()
{
while (true)
{
//判断信号状态,如果有set那么 _locks.WaitOne()后的程序就继续执行
_locks.WaitOne();
_taskAction();
//执行事件
WorkComplete(this);
}
} #region event
//构造函数
public Task()
{
//スレッドオブジェクトを初期化する
_thread = new Thread(Work);
_thread.IsBackground = true;
Key = Guid.NewGuid().ToString();
//线程开始执行
_thread.Start();
} //Set开起信号
public void Active()
{
_locks.Set();
} #endregion
}

解释:上面那个Key的作用,因为多个线程同时进行的时候,我们并不知道哪一个线程的工作先执行完,所以说上面的工作队列,实际上应该用一个字典来保存,这样我们就能在一个线程结束工作之后,通过这  里的KEY(每个线程不一样),来进行定位了。

2.线程封装好了,然后就可以实现线程池了

   public class TaskPool
{ #region Variable
//创建的线程数
private int _threadCount;
//空闲线程队列
private Queue<Task> _freeQueue;
//工作线程字典(为什么?)
private Dictionary<string, Task> _workingDictionary;
//空闲队列,存放需要被执行的外部函数
private Queue<Action> _waitQueue;
#endregion #region Event
//自定义线程池的构造函数
public TaskPool()
{
_workingDictionary = new Dictionary<string, Task>();
_freeQueue = new Queue<Task>();
_waitQueue = new Queue<Action>();
_threadCount = ; Task task = null;
//产生固定数目的线程
for (int i = ; i < _threadCount; i++)
{
task = new Task();
//给每一个任务绑定事件
task.WorkComplete += new Action<Task>(WorkComplete);
//将每一个新创建的线程放入空闲队列中
_freeQueue.Enqueue(task);
}
} //线程任务完成之后的工作
void WorkComplete(Task obj)
{
lock (this)
{
//将线程从字典中排除
_workingDictionary.Remove(obj.Key);
//将该线程放入空闲队列
_freeQueue.Enqueue(obj); //判断是否等待队列中有任务未完成
if (_waitQueue.Count > )
{
//取出一个任务
Action item = _waitQueue.Dequeue();
Task newTask = null;
//空闲队列中取出一个线程
newTask = _freeQueue.Dequeue();
// 线程执行任务
newTask._taskAction = item;
//把线程放入到工作队列当中
_workingDictionary.Add(newTask.Key, newTask);
//设置信号量
newTask.Active();
return;
}
else
{
return;
}
}
} //添加任务到线程池
public void AddTaskItem(Action taskItem)
{
lock (this)
{
Task task = null;
//判断空闲队列是否存在线程
if (_freeQueue.Count > )
{
//存在线程,取出一个线程
task = _freeQueue.Dequeue();
//将该线程放入工作队列
_workingDictionary.Add(task.Key, task);
//执行传入的任务
task._taskAction = taskItem;
//设置信号量
task.Active();
return;
}
else
{
//空闲队列中没有空闲线程,就把任务放到等待队列中
_waitQueue.Enqueue(taskItem);
return;
}
}
}
#endregion }

解释:这里的两个方法,基本符合我的设想,注意每一个方法里面都有lock操作,这就保证了,多个线程进行操作相同的队列对象的时候,能够进行互斥。保证一个时间只有一个线程在操作。

测试代码:

    class Program
{
static void Main(string[] args)
{
TaskPool _taskPool = new TaskPool(); Console.WriteLine(Thread.CurrentThread.ManagedThreadId);
for (var i = ; i < ; i++)
{
_taskPool.AddTaskItem(Print);
}
Console.Read();
} public static void Print()
{
Console.WriteLine("Do Something!");
}
}
}

这里我执行了20次print操作,看看结果是啥:

一个自定义线程池的小Demo

从图中看到20次确实执行了,但是看不到线程是哪些,稍微修改一下自定义的线程池。

1.在自定义线程的构造函数中添加:如下代码,查看哪些线程被创建了

         public Task()
{
_thread = new Thread(Work);
_thread.IsBackground = true;
Key = Guid.NewGuid().ToString();
//线程开始执行
_thread.Start();
Console.WriteLine("Thread:"+_thread.ManagedThreadId+" has been created!");
}

2.在线程完成工作方法之后添加如下代码,查看哪些线程参与执行任务

         private void Work()
{
while (true)
{
//判断信号状态,如果有set那么 _locks.WaitOne()后的程序就继续执行
_locks.WaitOne();
_taskAction();
Console.WriteLine("Thread:" + Thread.CurrentThread.ManagedThreadId+"workComplete");
//执行事件
WorkComplete(this);
}
}

3.修改客户端程序

     class Program
{
static void Main(string[] args)
{
TaskPool _taskPool = new TaskPool(); for (var i = ; i < ; i++)
{
_taskPool.AddTaskItem(Print);
}
Console.Read();
} public static void Print()
{
Thread.Sleep();
} }

测试结果:

一个自定义线程池的小Demo

从结果可以看到,开始和执行的线程都是固定的那10个,所以这个程序是可用的。