一个自定义线程池的小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操作，看看结果是啥：

从图中看到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();
         }

     }

测试结果：

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