Adds taskManager cpu usage
This commit is contained in:
@@ -13,37 +13,32 @@
|
||||
* the terms of the LGPL v3 license. See the LICENSE file for details.
|
||||
**/
|
||||
|
||||
#include <WiFi.h>
|
||||
#include <ArduinoJson.h>
|
||||
#include <ESPFS.h>
|
||||
#include <EventSocket.h>
|
||||
#include <TaskManager.h>
|
||||
#include <WiFi.h>
|
||||
|
||||
#define MAX_ESP_ANALYTICS_SIZE 1024
|
||||
#define MAX_ESP_ANALYTICS_SIZE 2024
|
||||
#define EVENT_ANALYTICS "analytics"
|
||||
#define ANALYTICS_INTERVAL 2000
|
||||
|
||||
class AnalyticsService
|
||||
{
|
||||
public:
|
||||
AnalyticsService(EventSocket *socket) : _socket(socket){};
|
||||
AnalyticsService(EventSocket *socket, TaskManager *taskManager) : _socket(socket), _taskManager(taskManager){};
|
||||
|
||||
void begin()
|
||||
{
|
||||
_socket->registerEvent(EVENT_ANALYTICS);
|
||||
void begin()
|
||||
{
|
||||
_socket->registerEvent(EVENT_ANALYTICS);
|
||||
|
||||
xTaskCreatePinnedToCore(
|
||||
this->_loopImpl, // Function that should be called
|
||||
"Analytics Service", // Name of the task (for debugging)
|
||||
5120, // Stack size (bytes)
|
||||
this, // Pass reference to this class instance
|
||||
(tskIDLE_PRIORITY), // task priority
|
||||
NULL, // Task handle
|
||||
ESP32SVELTEKIT_RUNNING_CORE // Pin to application core
|
||||
);
|
||||
};
|
||||
_taskManager->createTask(&AnalyticsService::_loopImpl, "Analytics Service", 8120, this, tskIDLE_PRIORITY, nullptr,
|
||||
ESP32SVELTEKIT_RUNNING_CORE);
|
||||
};
|
||||
|
||||
protected:
|
||||
EventSocket *_socket;
|
||||
TaskManager *_taskManager;
|
||||
|
||||
static void _loopImpl(void *_this) { static_cast<AnalyticsService *>(_this)->_loop(); }
|
||||
void _loop()
|
||||
@@ -62,6 +57,19 @@ protected:
|
||||
doc["fs_used"] = ESPFS.usedBytes();
|
||||
doc["fs_total"] = ESPFS.totalBytes();
|
||||
doc["core_temp"] = temperatureRead();
|
||||
doc["cpu0_usage"] = _taskManager->getCpuUsage(0);
|
||||
doc["cpu1_usage"] = _taskManager->getCpuUsage(1);
|
||||
doc["cpu_usage"] = _taskManager->getCpuUsage();
|
||||
// Add _taskManager->getTaskNames() as a JSON array
|
||||
JsonArray tasks = doc.createNestedArray("tasks");
|
||||
for (auto const &task : _taskManager->getTasks())
|
||||
{
|
||||
JsonObject nested = tasks.createNestedObject();
|
||||
nested["name"] = task.name;
|
||||
nested["stackSize"] = task.stackSize;
|
||||
nested["priority"] = task.priority;
|
||||
nested["coreId"] = task.coreId;
|
||||
}
|
||||
|
||||
serializeJson(doc, message);
|
||||
_socket->emit(EVENT_ANALYTICS, message);
|
||||
|
||||
@@ -107,9 +107,9 @@ void updateTask(void *param)
|
||||
|
||||
DownloadFirmwareService::DownloadFirmwareService(PsychicHttpServer *server,
|
||||
SecurityManager *securityManager,
|
||||
EventSocket *socket) : _server(server),
|
||||
EventSocket *socket, TaskManager *taskManager) : _server(server),
|
||||
_securityManager(securityManager),
|
||||
_socket(socket)
|
||||
_socket(socket), _taskManager(taskManager)
|
||||
{
|
||||
}
|
||||
|
||||
@@ -148,9 +148,9 @@ esp_err_t DownloadFirmwareService::downloadUpdate(PsychicRequest *request, JsonV
|
||||
|
||||
_socket->emit(EVENT_DOWNLOAD_OTA, output.c_str());
|
||||
|
||||
if (xTaskCreatePinnedToCore(
|
||||
if (_taskManager->createTask(
|
||||
&updateTask, // Function that should be called
|
||||
"Update", // Name of the task (for debugging)
|
||||
"Firmware download", // Name of the task (for debugging)
|
||||
OTA_TASK_STACK_SIZE, // Stack size (bytes)
|
||||
&downloadURL, // Pass reference to this class instance
|
||||
(configMAX_PRIORITIES - 1), // Pretty high task priority
|
||||
|
||||
@@ -24,6 +24,7 @@
|
||||
|
||||
#include <HTTPClient.h>
|
||||
#include <HTTPUpdate.h>
|
||||
#include <TaskManager.h>
|
||||
// #include <SSLCertBundle.h>
|
||||
|
||||
#define GITHUB_FIRMWARE_PATH "/api/downloadUpdate"
|
||||
@@ -33,7 +34,7 @@
|
||||
class DownloadFirmwareService
|
||||
{
|
||||
public:
|
||||
DownloadFirmwareService(PsychicHttpServer *server, SecurityManager *securityManager, EventSocket *socket);
|
||||
DownloadFirmwareService(PsychicHttpServer *server, SecurityManager *securityManager, EventSocket *socket, TaskManager *taskManager);
|
||||
|
||||
void begin();
|
||||
|
||||
@@ -41,5 +42,6 @@ private:
|
||||
SecurityManager *_securityManager;
|
||||
PsychicHttpServer *_server;
|
||||
EventSocket *_socket;
|
||||
TaskManager *_taskManager;
|
||||
esp_err_t downloadUpdate(PsychicRequest *request, JsonVariant &json);
|
||||
};
|
||||
|
||||
@@ -16,7 +16,7 @@
|
||||
#include <ESP32SvelteKit.h>
|
||||
|
||||
ESP32SvelteKit::ESP32SvelteKit(PsychicHttpServer *server, unsigned int numberEndpoints)
|
||||
: _server(server), _numberEndpoints(numberEndpoints),
|
||||
: _server(server), _numberEndpoints(numberEndpoints), _taskManager(),
|
||||
_featureService(server), _securitySettingsService(server, &ESPFS),
|
||||
_wifiSettingsService(server, &ESPFS, &_securitySettingsService, &_socket),
|
||||
_wifiScanner(server, &_securitySettingsService),
|
||||
@@ -32,7 +32,7 @@ ESP32SvelteKit::ESP32SvelteKit(PsychicHttpServer *server, unsigned int numberEnd
|
||||
_uploadFirmwareService(server, &_securitySettingsService),
|
||||
#endif
|
||||
#if FT_ENABLED(FT_DOWNLOAD_FIRMWARE)
|
||||
_downloadFirmwareService(server, &_securitySettingsService, &_socket),
|
||||
_downloadFirmwareService(server, &_securitySettingsService, &_socket, &_taskManager),
|
||||
#endif
|
||||
#if FT_ENABLED(FT_MQTT)
|
||||
_mqttSettingsService(server, &ESPFS, &_securitySettingsService),
|
||||
@@ -48,7 +48,7 @@ ESP32SvelteKit::ESP32SvelteKit(PsychicHttpServer *server, unsigned int numberEnd
|
||||
_batteryService(&_socket),
|
||||
#endif
|
||||
#if FT_ENABLED(FT_ANALYTICS)
|
||||
_analyticsService(&_socket),
|
||||
_analyticsService(&_socket, &_taskManager),
|
||||
#endif
|
||||
_restartService(server, &_securitySettingsService),
|
||||
_factoryResetService(server, &ESPFS, &_securitySettingsService),
|
||||
@@ -69,7 +69,7 @@ void ESP32SvelteKit::begin() {
|
||||
startServices();
|
||||
|
||||
ESP_LOGV("ESP32SvelteKit", "Starting loop task");
|
||||
xTaskCreatePinnedToCore(this->_loopImpl, "ESP32 SvelteKit Loop", 4096, this,
|
||||
_taskManager.createTask(this->_loopImpl, "Spot main", 4096, this,
|
||||
(tskIDLE_PRIORITY + 1), NULL,
|
||||
ESP32SVELTEKIT_RUNNING_CORE);
|
||||
}
|
||||
@@ -178,6 +178,8 @@ void ESP32SvelteKit::startServices() {
|
||||
#if FT_ENABLED(FT_BATTERY)
|
||||
_batteryService.begin();
|
||||
#endif
|
||||
|
||||
_taskManager.begin();
|
||||
}
|
||||
|
||||
void ESP32SvelteKit::_loop() {
|
||||
|
||||
@@ -148,6 +148,11 @@ public:
|
||||
return &_featureService;
|
||||
}
|
||||
|
||||
TaskManager *getTaskManager()
|
||||
{
|
||||
return &_taskManager;
|
||||
}
|
||||
|
||||
void factoryReset()
|
||||
{
|
||||
_factoryResetService.factoryReset();
|
||||
@@ -203,6 +208,7 @@ private:
|
||||
RestartService _restartService;
|
||||
FactoryResetService _factoryResetService;
|
||||
SystemStatus _systemStatus;
|
||||
TaskManager _taskManager;
|
||||
|
||||
String _appName = APP_NAME;
|
||||
|
||||
|
||||
@@ -3,77 +3,179 @@
|
||||
#include <esp_task_wdt.h>
|
||||
#include <map>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
#define IDLE_STACK_SIZE 2048
|
||||
#define DEFAULT_STACK_SIZE 2048+512
|
||||
|
||||
#define DELETE_TASK(handle) if (handle != nullptr) vTaskDelete(handle)
|
||||
|
||||
struct task_t
|
||||
{
|
||||
String name;
|
||||
TaskHandle_t handle;
|
||||
uint32_t stackSize;
|
||||
UBaseType_t priority;
|
||||
BaseType_t coreId;
|
||||
bool pinned;
|
||||
bool active; // blocked ('B'), ready ('R'), deleted ('D') or suspended ('S').
|
||||
};
|
||||
|
||||
class IdleTask
|
||||
{
|
||||
private:
|
||||
float _idleRatio = 0;
|
||||
unsigned long _lastMeasurement;
|
||||
|
||||
const int kMillisPerLoop = 1;
|
||||
const int kMillisPerCalc = 1000;
|
||||
|
||||
unsigned long counter = 0;
|
||||
|
||||
public:
|
||||
void ProcessIdleTime()
|
||||
{
|
||||
_lastMeasurement = millis();
|
||||
counter = 0;
|
||||
|
||||
for(;;)
|
||||
{
|
||||
int delta = millis() - _lastMeasurement;
|
||||
if (delta >= kMillisPerCalc)
|
||||
{
|
||||
_idleRatio = static_cast<float>(counter) / delta;
|
||||
_lastMeasurement = millis();
|
||||
counter = 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
esp_task_wdt_reset();
|
||||
delayMicroseconds(kMillisPerLoop * 1000);
|
||||
counter += kMillisPerLoop;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
IdleTask() : _lastMeasurement(millis())
|
||||
{
|
||||
}
|
||||
|
||||
float GetCPUUsage() const
|
||||
{
|
||||
if (millis() - _lastMeasurement > kMillisPerCalc)
|
||||
return 100.0f;
|
||||
|
||||
return 100.0f - 100 * _idleRatio;
|
||||
}
|
||||
|
||||
static void IdleTaskEntry(void *that)
|
||||
{
|
||||
static_cast<IdleTask *>(that)->ProcessIdleTime();
|
||||
}
|
||||
};
|
||||
|
||||
class TaskManager
|
||||
{
|
||||
private:
|
||||
std::map<std::string, TaskHandle_t> tasks;
|
||||
std::map<const char *, task_t> _tasks;
|
||||
IdleTask _taskIdle0;
|
||||
IdleTask _taskIdle1;
|
||||
|
||||
static void idleTask(void *pvParameters)
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
vTaskDelay(pdMS_TO_TICKS(1000)); // Delay to simulate workload
|
||||
}
|
||||
}
|
||||
TaskHandle_t _hIdle0;
|
||||
TaskHandle_t _hIdle1;
|
||||
|
||||
public:
|
||||
TaskManager()
|
||||
{
|
||||
xTaskCreatePinnedToCore(idleTask, "IdleTaskCore0", 1024, nullptr, 0, nullptr, 0);
|
||||
xTaskCreatePinnedToCore(idleTask, "IdleTaskCore1", 1024, nullptr, 0, nullptr, 1);
|
||||
}
|
||||
|
||||
void createTask(const std::string &name, void (*taskFunction)(void *), void *params = nullptr,
|
||||
uint32_t stackSize = 2048, UBaseType_t priority = 1)
|
||||
void begin()
|
||||
{
|
||||
TaskHandle_t handle;
|
||||
xTaskCreate(taskFunction, name.c_str(), stackSize, params, priority + 1, &handle);
|
||||
tasks[name] = handle;
|
||||
createTask(IdleTask::IdleTaskEntry, "Idle Core 0", IDLE_STACK_SIZE, &_taskIdle0, tskIDLE_PRIORITY-1, &_hIdle0, 0);
|
||||
createTask(IdleTask::IdleTaskEntry, "Idle Core 1", IDLE_STACK_SIZE, &_taskIdle1, tskIDLE_PRIORITY-1, &_hIdle1, 1);
|
||||
esp_task_wdt_delete(xTaskGetIdleTaskHandleForCPU(0));
|
||||
esp_task_wdt_delete(xTaskGetIdleTaskHandleForCPU(1));
|
||||
esp_task_wdt_add(_hIdle0);
|
||||
esp_task_wdt_add(_hIdle1);
|
||||
}
|
||||
|
||||
void suspendTask(const std::string &name)
|
||||
std::vector<task_t> getTasks()
|
||||
{
|
||||
if (tasks.find(name) != tasks.end())
|
||||
std::vector<task_t> tasks;
|
||||
for (auto const &task : _tasks)
|
||||
tasks.push_back(task.second);
|
||||
return tasks;
|
||||
}
|
||||
|
||||
int getTaskCount() const
|
||||
{
|
||||
return _tasks.size();
|
||||
}
|
||||
|
||||
int getKernelTaskCount() const
|
||||
{
|
||||
return uxTaskGetNumberOfTasks();
|
||||
}
|
||||
|
||||
void update()
|
||||
{
|
||||
for (auto const &task : _tasks)
|
||||
{
|
||||
vTaskSuspend(tasks[name]);
|
||||
_tasks[task.first].priority = uxTaskPriorityGet(task.second.handle);
|
||||
_tasks[task.first].coreId = xTaskGetAffinity(task.second.handle);
|
||||
}
|
||||
}
|
||||
|
||||
void resumeTask(const std::string &name)
|
||||
float getCpuUsage(int iCore = -1) const
|
||||
{
|
||||
if (tasks.find(name) != tasks.end())
|
||||
if (iCore == 0) return _taskIdle0.GetCPUUsage();
|
||||
else if (iCore == 1) return _taskIdle1.GetCPUUsage();
|
||||
return (_taskIdle0.GetCPUUsage() + _taskIdle1.GetCPUUsage()) / 2;
|
||||
}
|
||||
|
||||
BaseType_t createTask(void (*taskFunction)(void *), const char * name, uint32_t stackSize = 2048,
|
||||
void *params = nullptr, UBaseType_t priority = tskIDLE_PRIORITY + 1, TaskHandle_t* handle = nullptr,
|
||||
BaseType_t coreId = -1)
|
||||
{
|
||||
BaseType_t res = coreId == -1
|
||||
? xTaskCreate(taskFunction, name, stackSize, params, priority + 1, handle)
|
||||
: xTaskCreatePinnedToCore(taskFunction, name, stackSize, params, priority + 1, handle, coreId);
|
||||
task_t task = {name, handle, stackSize, priority + 1, coreId, coreId != -1, true};
|
||||
if (res == pdPASS)
|
||||
_tasks[name] = task;
|
||||
return res;
|
||||
}
|
||||
|
||||
void suspendTask(const char * name)
|
||||
{
|
||||
if (_tasks.find(name) != _tasks.end())
|
||||
{
|
||||
vTaskResume(tasks[name]);
|
||||
vTaskSuspend(_tasks[name].handle);
|
||||
_tasks[name].active = false;
|
||||
}
|
||||
}
|
||||
|
||||
void notifyTask(const std::string &name, uint32_t notificationValue, eNotifyAction action = eSetValueWithOverwrite)
|
||||
void resumeTask(const char * name)
|
||||
{
|
||||
if (tasks.find(name) != tasks.end())
|
||||
if (_tasks.find(name) != _tasks.end())
|
||||
{
|
||||
xTaskNotify(tasks[name], notificationValue, action);
|
||||
vTaskResume(_tasks[name].handle);
|
||||
_tasks[name].active = true;
|
||||
}
|
||||
}
|
||||
|
||||
void deleteTask(const std::string &name)
|
||||
void notifyTask(const char * name, uint32_t notificationValue, eNotifyAction action = eSetValueWithOverwrite)
|
||||
{
|
||||
if (tasks.find(name) != tasks.end())
|
||||
{
|
||||
vTaskDelete(tasks[name]);
|
||||
tasks.erase(name);
|
||||
}
|
||||
if (_tasks.find(name) != _tasks.end())
|
||||
xTaskNotify(_tasks[name].handle, notificationValue, action);
|
||||
}
|
||||
|
||||
float getCpuUsage(uint8_t coreId)
|
||||
void deleteTask(const char * name)
|
||||
{
|
||||
if (coreId > 1)
|
||||
return 0.0; // ESP32 has only core 0 and 1
|
||||
return uxTaskGetSystemState(nullptr, 0, nullptr) * 100.0 / configTICK_RATE_HZ;
|
||||
if (_tasks.find(name) != _tasks.end())
|
||||
{
|
||||
vTaskDelete(_tasks[name].handle);
|
||||
_tasks.erase(name);
|
||||
}
|
||||
}
|
||||
};
|
||||
@@ -215,8 +215,8 @@ void WiFiSettingsService::configureNetwork(wifi_settings_t &network)
|
||||
|
||||
void WiFiSettingsService::updateRSSI()
|
||||
{
|
||||
char buffer[16];
|
||||
snprintf(buffer, sizeof(buffer), WiFi.isConnected() ? "%d" : "disconnected", WiFi.RSSI());
|
||||
char buffer[4];
|
||||
snprintf(buffer, sizeof(buffer), "%d", WiFi.RSSI());
|
||||
_socket->emit(EVENT_RSSI, buffer);
|
||||
}
|
||||
|
||||
|
||||
@@ -48,7 +48,7 @@
|
||||
#define WIFI_SETTINGS_SERVICE_PATH "/api/wifiSettings"
|
||||
|
||||
#define WIFI_RECONNECTION_DELAY 1000 * 30
|
||||
#define RSSI_EVENT_DELAY 200
|
||||
#define RSSI_EVENT_DELAY 500
|
||||
|
||||
#define WIFI_SETTINGS_BUFFER_SIZE 2048
|
||||
|
||||
|
||||
Reference in New Issue
Block a user