linux 负载均衡 cpu迁移 cpu亲和性设置之间的联系

在做cpu迁移和负载均衡的最后一步时，将会将源地址进程移除出rq队列，并重新加载到目标cpu队列中去，其实现函数分别是

__migrate_task 和 pull_task 这两个函数

这两个函数实际的执行操作大同小异：

Ø deactivate_task()从最忙runquene中移除进程

Ø set_task_cpu()更新进程CPU指向

Ø activate_task()把进程重新插入添加到本地CPU

Ø check_preempt_curr()检查是否拥有更高优先权，可以调用resched_task()

cpu migrate 调度的逻辑图

扫描二维码关注公众号，回复： 3765044 查看本文章

migrate_call --> migrate_tasks--> __migrate_task --> deactive|set_task_cpu|activate

负载均衡的逻辑图

active_load_balance_cpu_stop-->move_one_task--> can_migrate_task |pull_task --> deactive|set_task_cpu|activate

cpu重新指向的逻辑图

deactive_task --> dequeue_task

set_task_cpu --> __set_task_cpu --> |set_task_rq

|task_thread_info(p)->cpu = cpu;

pull_task和__migrate的代码

static void pull_task(struct rq *src_rq, struct task_struct *p,

struct rq *this_rq, int this_cpu)

{

deactivate_task(src_rq, p, 0);

set_task_cpu(p, this_cpu);

activate_task(this_rq, p, 0);

check_preempt_curr(this_rq, p, 0);

}

static int __migrate_task(struct task_struct *p, int src_cpu, int dest_cpu)

{

……

if (p->on_rq) {

deactivate_task(rq_src, p, 0);

set_task_cpu(p, dest_cpu);

activate_task(rq_dest, p, 0);

check_preempt_curr(rq_dest, p, 0);

}

……

}

无论是pull_task还是 __migrate_task 都调用了set_task_cpu 实现指向目标cpu，两者核心功能和实现都一样，

但是__migrate_task加强了验错机制，pull_task在之前调用负载函数时已经完成了验错任务，所以pull_task只管执行即可

进程的cpu亲和性设置后，将会比较进程是否在新的cpu亲和性位图内，如果是的话，就保持原样，否则 立刻进行迁移

cpu亲和性设置的系统调用逻辑图

SYSCALL_DEFINE3(sched_setaffinity, pid_t, pid, unsigned int, len,unsigned long __user *, user_mask_ptr) -->sys_sched_setaffinity

sys_sched_setaffinity--> sched_setaffinity -->set_cpus_allowed_ptr-->do_set_cpus_allowed--> migration_cpu_stop --> __migrate_task

linux 负载均衡 cpu迁移 cpu亲和性设置 之间的联系