乐趣区

程序员修仙之路-数据结构之设计一个高性能线程池

原因排查
经过一个多小时的代码排查终于查明了线上程序线程数过多的原因:这是一个接收 mq 消息的一个服务,程序大体思路是这样的,监听的线程每次收到一条消息,就启动一个线程去执行,每次启动的线程都是新的。说到这里,咱们就谈一谈这个程序有哪些弊端呢:

每次收到一条消息都创建一个新的线程,要知道线程的资源对于系统来说是很昂贵的,消息处理完成还要销毁这个线程。
这个程序用到的线程数量是没有限制的。当线程到达一定数量,程序反而因线程在 cpu 切换开销的原因处理效率降低。无论的你的服务器 cpu 是多少核心,这个现象都有发生的可能。

解决问题
线程多的问题该怎么解决呢,增加 cpu 核心数?治标不治本。对于开发者而言,最为常用也最为有效的是线程池化,也就是说线程池。
线程池是一种多线程处理形式,处理过程中将任务添加到队列,然后在创建线程后自动启动这些任务。这避免了在处理短时间任务时创建与销毁线程的代价。线程池不仅能够保证内核的充分利用,还能防止过分调度。可用线程数量应该取决于可用的并发处理器、处理器内核、内存、网络 sockets 等的数量。例如,线程数一般取 cpu 数量 + 2 比较合适,线程数过多会导致额外的线程切换开销。
线程池其中一项很重要的技术点就是任务的队列,队列虽然属于一种基础的数据结构,但是发挥了举足轻重的作用。
队列
队列是一种特殊的线性表,特殊之处在于它只允许在表的前端(front)进行删除操作,而在表的后端(rear)进行插入操作,和栈一样,队列是一种操作受限制的线性表。进行插入操作的端称为队尾,进行删除操作的端称为队头。
队列是一种采用的 FIFO(first in first out) 方式的线性表,也就是经常说的先进先出策略。
实现
数组
队列可以用数组 Q[1…m] 来存储,数组的上界 m 即是队列所容许的最大容量。在队列的运算中需设两个指针:head,队头指针,指向实际队头元素 + 1 的位置;tail,队尾指针,指向实际队尾元素位置。一般情况下,两个指针的初值设为 0,这时队列为空,没有元素。以下为一个简单的实例(生产环境需要优化):
public class QueueArray<T>
{
// 队列元素的数组容器
T[] container = null;
int IndexHeader, IndexTail;
public QueueArray(int size)
{
container = new T[size];
IndexHeader = 0;
IndexTail = 0;
}
public void Enqueue(T item)
{
// 入队的元素放在头指针的指向位置,然后头指针前移
container[IndexHeader] = item;
IndexHeader++;
}
public T Dequeue()
{
// 出队:把尾元素指针指向的元素取出并清空(不清空也可以)对应的位置,尾指针前移
T item = container[IndexTail];
container[IndexTail] = default(T);
IndexTail++;
return item;
}

}
链表
队列采用的 FIFO(first in first out),新元素总是被插入到链表的尾部,而读取的时候总是从链表的头部开始读取。每次读取一个元素,释放一个元素。所谓的动态创建,动态释放。因而也不存在溢出等问题。由于链表由元素连接而成,遍历也方便。以下是一个实例仅供参考:
public class QueueLinkList<T>
{
LinkedList<T> contianer = null;
public QueueLinkList()
{
contianer = new LinkedList<T>();
}
public void Enqueue(T item)
{
// 入队的元素其实就是加入到队尾
contianer.AddLast(item);
}
public T Dequeue()
{
// 出队:取链表第一个元素,然后把这个元素删除
T item = contianer.First.Value;
contianer.RemoveFirst();
return item;
}

}
队列扩展阅读

队列通过数组来实现的话有什么问题吗?是的。首先基于数组不可变本质的因素(具体可参考菜菜之前的文章),当一个队列的元素把数组沾满的时候,数组扩容是有性能问题的,数组的扩容过程不只是开辟新空间分配内存那么简单,还要有数组元素的 copy 过程,更可怕的是会给 GC 造成极大的压力。如果数组比较小可能影响比较小,但是当一个数组比较大的时候,比如占用 500M 内存的一个数组,数据 copy 其实会造成比较大的性能损失。
队列通过数组来实现,随着头指针和尾指针的位置移动,尾指针最终会指向第一个元素的位置,也就是说没有元素可以出队了,其实要解决这个问题有两种方式,其一:在出队或者入队的过程中不断的移动所有元素的位置,避免上边所说的极端情况发生;其二:可以把数组的首尾元素连接起来,使其成为一个环状,也就是经常说的循环队列。
队列在一些特殊场景下其实还有一些变种,比如说循环队列,阻塞队列,并发队列等,有兴趣的同学可以去研究一下,这里不在展开讨论。这里说到阻塞队列就多说一句,其实用阻塞队列可以实现一个最基本的生产者消费者模式。
当队列用链表方式实现的时候,由于链表的首尾操作时间复杂度都是 O(1),而且没有空间大小的限制,所以一般的队列用链表实现更简单。
当队列中无元素可出队或者没有空间可入队的时候,是阻塞当前的操作还是返回错误信息,取决于在座各位队列的设计者了。

简单实用的线程池
// 线程池
public class ThreadPool
{
bool PoolEnable = false; // 线程池是否可用
List<Thread> ThreadContainer = null; // 线程的容器
ConcurrentQueue<ActionData> JobContainer = null; // 任务的容器
public ThreadPool(int threadNumber)
{
PoolEnable = true;
ThreadContainer = new List<Thread>(threadNumber);
JobContainer = new ConcurrentQueue<ActionData>();
for (int i = 0; i < threadNumber; i++)
{
var t = new Thread(RunJob);
ThreadContainer.Add(t);
t.Start();
}
}
// 向线程池添加一个任务
public void AddTask(Action<object> job,object obj, Action<Exception> errorCallBack=null)
{
if (JobContainer != null)
{
JobContainer.Enqueue(new ActionData { Job = job, Data = obj , ErrorCallBack= errorCallBack});
}

}
// 终止线程池
public void FinalPool()
{
PoolEnable = false;
JobContainer = null;
if (ThreadContainer != null)
{
foreach (var t in ThreadContainer)
{
// 强制线程退出并不好,会有异常
//t.Abort();
t.Join();
}
ThreadContainer = null;
}

}
private void RunJob()
{
while (true&& JobContainer!=null&& PoolEnable)
{
// 任务列表取任务
ActionData job=null;
JobContainer?.TryDequeue(out job);
if (job == null)
{
// 如果没有任务则休眠
Thread.Sleep(10);
continue;
}
try
{
// 执行任务
job.Job.Invoke(job.Data);
}
catch(Exception error)
{
// 异常回调
job?.ErrorCallBack(error);
}
}
}
}

public class ActionData
{
// 执行任务的参数
public object Data {get; set;}
// 执行的任务
public Action<object> Job {get; set;}
// 发生异常时候的回调方法
public Action<Exception> ErrorCallBack {get; set;}
}
使用
ThreadPool pool = new ThreadPool(100);
for (int i = 0; i < 5000; i++)
{
pool.AddTask((obj) =>
{
Console.WriteLine($”{obj}__{System.Threading.Thread.CurrentThread.ManagedThreadId}”);
}, i, (e) =>
{
Console.WriteLine(e.Message);
});
}
pool.FinalPool();
Console.Read();

添加关注,查看更精美版本,收获更多精彩

退出移动版