【Linux 内核】线程调度示例一 ③ ( 获取线程优先级 | 设置线程调度策略 | 代码示例 )

一、获取线程优先级

---

1、pthread_attr_setschedparam 和 pthread_attr_getschedparam 函数

设置、获取 线程 优先级的 核心 函数 :

① 设置 " 创建线程 " 的优先级 :

int pthread_attr_setschedparam(pthread_attr_t *attr, const struct sched_param *param)

② 获取 " 创建线程 " 的优先级 :

int pthread_attr_getschedparam(pthread_attr_t *attr, const struct sched_param *param)

设置 struct sched_param *param 结构体的 sched_priority 字段 , 即可设置 " 优先级 " 属性 ;

上述 $2$ 个函数 , 如果执行成功 , 返回 $0$ ;

如果执行失败 , 则返回错误代码 :

• EINVAL : 属性设置无效 ;
• ENOTSUP : 设置的属性值不合法 ;

2、获取线程优先级代码示例

代码示例 :

/**
 * @brief 获取线程优先级
 * 
 * @return int 
 */
static int get_thread_priority(pthread_attr_t *p_attr){
	// 获取调度参数
	struct sched_param param;

    // 获取线程调度优先级
    int ret = pthread_attr_getschedparam(p_attr, &param);

    // 确保获取优先级操作执行成功，如果执行失败，则退出程序
    assert(ret == 0);

    printf("获取的线程优先级为 %d\n", param.sched_priority);

    return param.sched_priority;
}

二、设置线程调度策略

---

1、pthread_attr_setschedpolicy 函数

创建 pthread 线程时 , 默认的线程时 SCHED_OTHHER 调度策略 , 可以通过下面的函数 , 修改调度策略 ;

修改线程 " 调度策略 " 函数 :

int pthread_attr_setschedpolicy(pthread_attr_t *attr, int policy);

如果函数执行成功 , 返回 $0$ ;

如果函数执行失败 , 返回错误代码 :

• EINVALEINVAL : pthread_attr_t *attr 线程未初始化 , 或者 , int policy 不是有效调度策略 ;
• EFAULTEFAULT : pthread_attr_t *attr 指针无效 ;
• ENOTSUPENOTSUP : 尝试将调度策略修改为 SCHED_FIFO 或 SCHED_RR 实时调度策略 ;

2、设置线程调度策略代码示例

代码示例 :

/**
 * @brief 设置线程调度策略
 * 
 * @return int 
 */
static int set_thread_policy(pthread_attr_t *p_attr, int policy){
    // 设置线程调度策略
    int ret = pthread_attr_setschedpolicy(p_attr, policy);

    // 确保获取优先级操作执行成功，如果执行失败，则退出程序
    assert(ret == 0);

    // 验证线程是否设置调度策略成功， 打印调度策略
    get_thread_policy(p_attr);

    return 0;
}