nng协议nni_taskq_sys_init(void) 对nni_taskq_systq 初始化

函数调用关系:nni_init(void) --> nni_plat_init(nni_init_helper) 今天分析的函数位于 nni_init_helper()函数中的 nni_taskq_sys_init() 。 这个函数主要用于针对 全局变量 nni_taskq_systq进行申请空间并初始化。

  1. 确定线程数量 num_thr

这个通过宏定义或者函数获得。 没定义则取CPU数量 * 2; 最大值为16. 如果小于2, 则指定num_thr 的值为2. 最少2个,最多16个。

2.执行初始化 nni_taskq_init**(&nni_taskq_systq,num_thr)**

nni_taskq 的定义:

cpp 复制代码
struct nni_taskq_thr{
    nni_taskq *tqt_tq;
    nni_thr    tqt_thread;
}  

struct    nni_taskq {
    nni_list       tq_tasks;
    nni_mtx        tq_mtx;
    nni_cv         tq_sched_cv;
    nni_cv         tq_wait_cv;
    nni_taskq_thr  *tq_threads;
    int            tq_nthreads;
    bool           tq_run;
}

基本步骤:

  1. tq申请空间, 为tq->tq_threads 申请内存空间

  2. 对线程数量赋值,tq->tq_nthreads = nthr .

  3. 链表初始化 nni_taskq_systq->tq-tasks互斥锁初始化, 条件变量初始化

cpp 复制代码
//请详细分析并给出使用范例
nni_mtx_init(&tq->mtx);
nni_cv_init(&tq->tq_sched_cv, &tq->tq_mtx);
nni_cv_init(&tq->tq_wait_cv, &tq->tq_mtx);
  1. 循环对全局变量nni_taskq_systq的成员变量 所有的 **tq_threads[i]**进行初始化。

  2. th_threads[i] 的初始化流程:

tq->th_threads[i].tqt_tq = tq;

执行nni_thr_inittqt_thread[i].tqt_thread 进行初始化。 这个函数和其他的初始化类似,首先对第一层变量进行赋值,然后调用**nni_plat_thr_init,**初始化后的值为:

cpp 复制代码
//tq的类型是  nni_taskq  *tq;
nni_thr_init(&tq->tq_threads[i].tqt_thread, nni_taskq_thread, &tq->tq_threads[i]);


tq->tq_threads[i].tqt_tq = tq;

&tq->tq_threads[i].tqt_thread 的类型是nni_thr
//nni_thr_init
tq->tq_threads[i].tqt_thread->done  = 0;
tq->tq_threads[i].tqt_thread->start = 0;
tq->tq_threads[i].tqt_thread->stop  = 0;
tq->tq_threads[i].tqt_thread->fn    = nni_taskq_thread;
tq->tq_threads[i].tqt_thread->arg   = &tq->tq_threads[i];

//tq->tq_threads[i].tqt_thread->thr的类型是 nni_plat_thr

//nni_plat_thr_init(&thr->thr, nni_thr_wrap, thr)
tq->tq_threads[i].tqt_thread->thr->func = nni_thr_wrap;
tq->tq_threads[i].tqt_thread->thr->arg  = tq->tq_threads[i].tqt_thread;

在nni_plat_thr_init(nni_plat_thr *thr, void (*fn)(void *), void *arg)中创建线程,执行线程函数:nni_plat_thr_main(void *arg);

在线程函数中调用函数指针:thr->func(thr->arg); 也就是nni_plat_init 初始化的thr->func 亦即:nni_thr_wrap,参数是nni_plat_init 中设置的参数,这里是 &tq->tq_threads[i];由于初始化时,start==0, stop==0 所以函数将等待在条件变量上。

cpp 复制代码
static void
nni_thr_wrap(void *arg)
{
	nni_thr *thr = arg;
	int      start;

	nni_plat_mtx_lock(&thr->mtx);
	while (((start = thr->start) == 0) && (thr->stop == 0)) { //初始化条件,函数将等待在条件变量
		nni_plat_cv_wait(&thr->cv);
	}
	nni_plat_mtx_unlock(&thr->mtx);
	if ((start) && (thr->fn != NULL)) {
		thr->fn(thr->arg);
	}
	nni_plat_mtx_lock(&thr->mtx);
	thr->done = 1;
	nni_plat_cv_wake(&thr->cv);
	nni_plat_mtx_unlock(&thr->mtx);
}

在 nni_taskq_init 的最后,这里将执行nni_thr_run 函数,设置条件 start == 1 并唤醒等待在条件变量上的线程。

cpp 复制代码
for (int i = 0; i < tq->tq_nthreads; i++) {
		nni_thr_run(&tq->tq_threads[i].tqt_thread);
}

函数继续执行,调用 nni_taskq_thread函数,参数是 &tq->tq_threads[i] ;

执行简图请参考resolv_thr 的初始化

cpp 复制代码
static void
nni_taskq_thread(void *self)
{
	nni_taskq_thr *thr = self;
	nni_taskq     *tq  = thr->tqt_tq;
	nni_task      *task;

	nni_thr_set_name(NULL, "nng:task");

	nni_mtx_lock(&tq->tq_mtx);
	for (;;) {
		if ((task = nni_list_first(&tq->tq_tasks)) != NULL) {

			nni_list_remove(&tq->tq_tasks, task);

			nni_mtx_unlock(&tq->tq_mtx);

			task->task_cb(task->task_arg);

			nni_mtx_lock(&task->task_mtx);
			task->task_busy--;
			if (task->task_busy == 0) {
				nni_cv_wake(&task->task_cv);
			}
			nni_mtx_unlock(&task->task_mtx);

			nni_mtx_lock(&tq->tq_mtx);

			continue;
		}

		if (!tq->tq_run) {
			break;
		}
		nni_cv_wait(&tq->tq_sched_cv);
	}
	nni_mtx_unlock(&tq->tq_mtx);
}
相关推荐
.似水37 分钟前
2025.4.22_C_可变参数列表
java·c语言·算法
小鹿鹿啊2 小时前
C语言编程--14.电话号码的字母组合
c语言·开发语言·算法
一只鱼^_3 小时前
第十六届蓝桥杯大赛软件赛省赛 C/C++ 大学B组 [京津冀]
c语言·数据结构·c++·算法·贪心算法·蓝桥杯·动态规划
Kisorge3 小时前
【电机仿真】MPC模型预测转速、电流双闭环控制器——PMSM有感FOC控制
c语言
几点才到啊4 小时前
C语言实现冒泡排序:算法原理与代码解析
c语言·算法·排序算法
binary思维4 小时前
C语言实现贪心算法
c语言·算法·贪心算法
努力做小白5 小时前
Linux扩展
linux·c语言·笔记
minji...7 小时前
C语言 函数递归
c语言·开发语言·算法
CHTXRT7 小时前
2025第十六届蓝桥杯大赛(软件赛)网络安全赛 Writeup
c语言·网络·web安全·网络安全·蓝桥杯·wireshark
小智学长 | 嵌入式9 小时前
单片机-89C51部分:4、固件烧录
c语言·单片机·嵌入式硬件