freeswitch APR-UTIL库线程池实现分析

 

概述

freeswitch的核心源代码是基于apr库开发的，在不同的系统上有很好的移植性。

APR库在之前的文章中已经介绍过了，APR-UTIL库是和APR并列的工具库，它们都是由APACHE开源出来的跨平台可移植库，不同点在于库中实现的功能接口有区别。

在应用的开发过程中，多线程并发是提高效率的常用方案，但是多线程管理并不好做。

在很多大型应用中，都会引入线程池的框架。线程池是一个线程集合，有统一的管理，当有一个新的任务下发，线程池管理会按照一定的策略将任务分配给空闲的线程。当任务积压较多时，线程池会创建新的线程来加快处理效率。

APR-UTIL库中就提供了一套线程池接口。

我对几个问题比较好奇，线程池如何管理？线程池什么情况要增加线程？什么情况会减少线程？线程池线程数目如何设置才有最优的效率？

下面我们对apr-util库的线程池实现做一个介绍。

环境

centos：CentOS  release 7.0 (Final)或以上版本

APR-UTIL：1.6.1

GCC：4.8.5

本来要使用freeswitch1.8.7中带的apr-util库源代码来梳理，但是很遗憾的是这个apr-util库版本是1.2.8，里面没有apr_thread_pool接口。。。所以从APR官网上下载了最新的1.6.1版本来做分析。

数据结构

apr线程池源文件：

apr-util-1.6.1\include\apr_thread_pool.h

apr-util-1.6.1\misc\apr_thread_pool.c

号码池数据结构定义在apr_thread_pool.c中

typedef struct apr_thread_pool_task

{

    APR_RING_ENTRY(apr_thread_pool_task) link;

    apr_thread_start_t func;

    void *param;

    void *owner;

    union

    {

        apr_byte_t priority;

        apr_time_t time;

    } dispatch;

} apr_thread_pool_task_t;

APR_RING_HEAD(apr_thread_pool_tasks, apr_thread_pool_task);

struct apr_thread_list_elt

{

    APR_RING_ENTRY(apr_thread_list_elt) link;

    apr_thread_t *thd;

    volatile void *current_owner;

    volatile enum { TH_RUN, TH_STOP, TH_PROBATION } state;

};

APR_RING_HEAD(apr_thread_list, apr_thread_list_elt);

struct apr_thread_pool

{

    apr_pool_t *pool;

    volatile apr_size_t thd_max;

    volatile apr_size_t idle_max;

    volatile apr_interval_time_t idle_wait;

    volatile apr_size_t thd_cnt;

    volatile apr_size_t idle_cnt;

    volatile apr_size_t task_cnt;

    volatile apr_size_t scheduled_task_cnt;

    volatile apr_size_t threshold;

    volatile apr_size_t tasks_run;

    volatile apr_size_t tasks_high;

    volatile apr_size_t thd_high;

    volatile apr_size_t thd_timed_out;

    struct apr_thread_pool_tasks *tasks;

    struct apr_thread_pool_tasks *scheduled_tasks;

    struct apr_thread_list *busy_thds;

    struct apr_thread_list *idle_thds;

    apr_thread_mutex_t *lock;

    apr_thread_cond_t *cond;

    volatile int terminated;

    struct apr_thread_pool_tasks *recycled_tasks;

    struct apr_thread_list *recycled_thds;

    apr_thread_pool_task_t *task_idx[TASK_PRIORITY_SEGS];

};

线程池内存模型总图，线程池，任务队列，线程队列。

 

常用函数

常用函数接口

apr_thread_pool_create       //Create a thread pool

apr_thread_pool_destroy     //Destroy the thread pool and stop all the threads

apr_thread_pool_push  //Schedule a task to the bottom of the tasks of same priority.

apr_thread_pool_schedule   //Schedule a task to be run after a delay

apr_thread_pool_top    //Schedule a task to the top of the tasks of same priority.

apr_thread_pool_tasks_cancel     //Cancel tasks submitted by the owner. If there is any task from the owner that is currently running, the function will spin until the task finished.

apr_thread_pool_tasks_count      //Get the current number of tasks waiting in the queue

apr_thread_pool_scheduled_tasks_count   //Get the current number of scheduled tasks waiting in the queue

apr_thread_pool_threads_count  //Get the current number of threads

apr_thread_pool_busy_count      //Get the current number of busy threads

apr_thread_pool_idle_count //Get the current number of idle threads

apr_thread_pool_idle_max_set    //Access function for the maximum number of idle threads. Number of current idle threads will be reduced to the new limit.

apr_thread_pool_tasks_run_count      //Get number of tasks that have run

apr_thread_pool_tasks_high_count    //Get high water mark of the number of tasks waiting to run

apr_thread_pool_threads_high_count //Get high water mark of the number of threads

apr_thread_pool_threads_idle_timeout_count   //Get the number of idle threads that were destroyed after timing out

apr_thread_pool_idle_max_get    //Access function for the maximum number of idle threads

apr_thread_pool_thread_max_set       //Access function for the maximum number of threads.

apr_thread_pool_idle_wait_set     //Access function for the maximum wait time (in microseconds) of an idling thread that exceeds the maximum number of idling threads. A non-zero value allows for the reaping of idling threads to shrink over time.  Which helps reduce thrashing.

apr_thread_pool_idle_wait_get    //Access function for the maximum wait time (in microseconds) of an idling thread that exceeds the maximum number of idling threads

apr_thread_pool_thread_max_get      //Access function for the maximum number of threads

apr_thread_pool_threshold_set   //Access function for the threshold of tasks in queue to trigger a new thread.

apr_thread_pool_threshold_get   //Access function for the threshold of tasks in queue to trigger a new thread.

apr_thread_pool_task_owner_get       //Get owner of the task currently been executed by the thread.

apr_thread_pool_create创建

APU_DECLARE(apr_status_t) apr_thread_pool_create(apr_thread_pool_t ** me,

                                                 apr_size_t init_threads,

                                                 apr_size_t max_threads,

                                                 apr_pool_t * pool)

接口逻辑：

1. 分配一块大小为apr_thread_pool_t的内存tp。
2. 在传入的内存池pool中申请一个新的内存池tp->pool。
3. 初始化线程池数据。
   1. 线程池数据初始化。
   2. 创建线程互斥锁me->lock。
   3. 创建条件变量me->cond。
   4. 在内存池pool上分配一块大小为“apr_thread_pool_tasks“的内存赋值给me->tasks。
   5. 在内存池pool上分配一块大小为“apr_thread_pool_tasks“的内存赋值给me->scheduled_tasks。
   6. 在内存池pool上分配一块大小为“apr_thread_pool_tasks“的内存赋值给me->recycled_tasks。
   7. 在内存池pool上分配一块大小为“apr_thread_list“的内存赋值给me->busy_thds。
   8. 在内存池pool上分配一块大小为“apr_thread_list“的内存赋值给me->idle_thds。
   9. 在内存池pool上分配一块大小为“apr_thread_list“的内存赋值给me->recycled_thds。
   10. 线程池数据初始化。
4. 在内存池tp->pool中注册清理回调函数。
5. 循环创建初始工作线程，并加入线程池的管理。工作线程的逻辑见“thread_pool_func工作线程”。
6. 返回创建结果。

线程池初始化成功后，内存模型如图（工作线程启动未完成时）

 

thread_pool_func工作线程

static void *APR_THREAD_FUNC thread_pool_func(apr_thread_t * t, void *param)

接口逻辑：

1. 加锁me->lock
2. 判断me->recycled_thds链表为空？为空则创建新的apr_thread_list_elt节点elt，不为空则获取recycled_thds中首节点elt并从recycled_thds中移除该节点。
3. 循环处理。
   1. 将elt节点加入me->busy_thds链表。
   2. 获取一个新任务task。TODO
   3. 循环处理。解锁me->lock。调用任务回调task->func。加锁me->lock。将task加入me->recycled_tasks链表。获取新任务task。线程状态置为TH_STOP时跳出循环。获取任务为空跳出循环。
   4. 线程从busy到stop状态，将elt加入me->recycled_thds链表尾部，解锁me->lock，退出线程。
   5. 线程从busy到idle状态，将elt节点从me->busy_thds链表中移除，将elt加入me->idle_thds链表尾部。
   6. 检查是否有定时任务并获取任务执行等待时间。
   7. 检查当前空闲线程数是否大于最大空闲数，获取空闲等待时间me->idle_wait，并设置当前线程状态为TH_PROBATION，下一轮循环中进入stop处理流程。
   8. 线程阻塞，等待条件变量me->cond的通知或超时。
4. 线程数me->thd_cnt自减。
5. 解锁me->lock。
6. 退出线程。

线程池初始化成功后，内存模型如图（工作线程启动完成时）

 

apr_thread_pool_push添加任务

APU_DECLARE(apr_status_t) apr_thread_pool_push(apr_thread_pool_t *me,

                                               apr_thread_start_t func,

                                               void *param,

                                               apr_byte_t priority,

                                               void *owner)

接口逻辑：

1. 加锁me->lock。
2. 检查me->recycled_tasks是否为空，为空则新建任务节点t，不为空则从me->recycled_tasks获取任务节点t。
3. 任务节点t数据初始化。
4. 计算任务优先级，根据优先级设置me->task_idx[seg]和me->tasks。
5. 当前工作线程数为0时，或者空闲线程数为0并且当前线程数未达到最大并且当前任务数超过阈值等条件，动态创建新的工作线程。
6. 对条件变量me->cond发通知。
7. 解锁me->lock。

线程池添加任务后的内存模型图。

 

apr_thread_pool_tasks_cancel取消任务

APU_DECLARE(apr_status_t) apr_thread_pool_tasks_cancel(apr_thread_pool_t *me,

                                                       void *owner)

接口逻辑：

1. 加锁me->lock。
2. 如果当前任务数大于0，则清空owner的所有任务。
3. 如果定时任务数大于0，则清空owner的所有定时任务。
4. 解锁me->lock。
5. 等待线程退出。

总结

APR线程池的几个关注点。

线程从busy到stop状态时，没有将elt节点从me->busy_thds链表中删除？

APR线程池没有内置的管理线程，根据当前线程数和任务数进行动态的调整，而是通过任务阈值、空闲线程最大值和超时时间等设置来控制线程数的增减，这一点和我开始想的不一样。

---

空空如常

求真得真