RT-Thread-rt-thread 统计CPU使用率RT-Thread问答社区

rt-thread 统计CPU使用率

发布于 2024-01-27 17:36:50 浏览：906 订阅该版

1、在位于bsp/bouffalo_lab/bl808/m0/board/board.c的头部新增

```c
#include "cpu_load.h"
```
2、在位于bsp/bouffalo_lab/bl808/m0/board/board.c文件中rt_hw_board_init()的末尾加入如下代码:

```c
#ifdef CPU_LOAD_MONITOR
    rt_thread_inited_sethook(thread_init_hook);
    rt_scheduler_sethook(sched_hook);
#endif
```
3、然后把cpu_load.h和cpu_load.c文件放入bsp/bouffalo_lab/bl808/m0/applications目录下：
cpu_load.h

```c
#ifndef _CPU_LOAD_H_
#define _CPU_LOAD_H_

#include <rtthread.h>
#include <stdio.h>
#include <rtdevice.h>
#include "bflb_mtimer.h"

#define CPU_LOAD_MONITOR

typedef struct
{
        struct rt_thread *thread;
        uint64_t count;
        uint64_t base;
}task_monitor_t;

void thread_init_hook(rt_thread_t thread);
void sched_hook(struct rt_thread* from, struct rt_thread* to);

#endif
```
cpu_load.c

```c
#include "cpu_load.h"

#ifdef CPU_LOAD_MONITOR

#define CPU_LOAD_MONITOR_SIZE 20
static task_monitor_t task_monitor_list[CPU_LOAD_MONITOR_SIZE] = {0};
static uint32_t task_count = 0;
static uint64_t monitor_base = 0;

void thread_init_hook(rt_thread_t thread)
{
        int found;
        int index;
        rt_enter_critical();
        found = 0;
        for(index = 0; index < task_count; index++)
        {
                if(thread == task_monitor_list[index].thread)
                {
                        found = 1;
                        break;
                }
        }
        if(!found)
        {
                if(task_count < CPU_LOAD_MONITOR_SIZE)
                {
                        task_monitor_list[task_count].thread = thread;
                        task_count++;
                }
        }
        else
        {
                task_monitor_list[index].base = 0;
        }
        rt_exit_critical();
}
void sched_hook(struct rt_thread* from, struct rt_thread* to)
{
        int task_found = 0;
        int index = 0;
        static uint64_t base = 0;

if(!base)
        {
                base = bflb_mtimer_get_time_us();
                monitor_base = base;
        }

uint64_t time = bflb_mtimer_get_time_us();
        for(index = 0; index < task_count; index++)
        {
                if(task_monitor_list[index].thread == from)
                {
                        task_found = 1;
                        break;
                }
        }

if(task_found != 1)
        {
                printf("error occur !!!\n");
        }
        else
        {
                if(task_monitor_list[index].base == 0)
                {
                        task_monitor_list[index].base = time;
                        task_monitor_list[index].count = time;
                }
                else
                {
                        task_monitor_list[index].count += time-base;
                }
        }
        base = time;
}

int cpu_load(int argc, const char **argv)
{
        (void)argc;
        (void)argv;
        uint64_t time = bflb_mtimer_get_time_us() - monitor_base;
        for(int i = 0; i < task_count; i++)
        {
                printf("%02d: %2d%%", i,
                                (task_monitor_list[i].count- \
                                 task_monitor_list[i].base)*100/time);
                printf(" %p ", task_monitor_list[i].thread);
                printf(" task:%s\n", task_monitor_list[i].thread->parent.name);
        }

rt_enter_critical();
        monitor_base = bflb_mtimer_get_time_us();
        for(int i = 0; i < task_count; i++)
        {
                task_monitor_list[i].base = monitor_base;
                task_monitor_list[i].count = monitor_base;
        }
        rt_exit_critical();
        return 0;
}
MSH_CMD_EXPORT(cpu_load, cpu monitor);

#endif
```
4、执行在m0目录下执行scons编译M0核固件
3、目前每敲击一次cpu_load命令，则从新开始统计各任务CPU占比，可根据自身需求选择是否重新统计及观测窗口。