Skip to content

Commit

Permalink
[Examples] [Async Handlers] simplify code
Browse files Browse the repository at this point in the history
  • Loading branch information
chipweinberger authored Nov 5, 2024
1 parent ce60853 commit a89a954
Showing 1 changed file with 32 additions and 45 deletions.
77 changes: 32 additions & 45 deletions examples/protocols/http_server/async_handlers/main/main.c
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ static const char *TAG = "example";

// Async requests are queued here while they wait to
// be processed by the workers
static QueueHandle_t async_req_queue;
static QueueHandle_t request_queue;

// Track the number of free workers at any given time
static SemaphoreHandle_t worker_ready_count;
Expand All @@ -53,22 +53,8 @@ typedef struct {
httpd_req_handler_t handler;
} httpd_async_req_t;


static bool is_on_async_worker_thread(void)
{
// is our handle one of the known async handles?
TaskHandle_t handle = xTaskGetCurrentTaskHandle();
for (int i = 0; i < CONFIG_EXAMPLE_MAX_ASYNC_REQUESTS; i++) {
if (worker_handles[i] == handle) {
return true;
}
}
return false;
}


// Submit an HTTP req to the async worker queue
static esp_err_t submit_async_req(httpd_req_t *req, httpd_req_handler_t handler)
// queue an HTTP req to the worker queue
static esp_err_t queue_request(httpd_req_t *req, httpd_req_handler_t handler)
{
// must create a copy of the request that we own
httpd_req_t* copy = NULL;
Expand Down Expand Up @@ -97,7 +83,7 @@ static esp_err_t submit_async_req(httpd_req_t *req, httpd_req_handler_t handler)

// Since worker_ready_count > 0 the queue should already have space.
// But lets wait up to 100ms just to be safe.
if (xQueueSend(async_req_queue, &async_req, pdMS_TO_TICKS(100)) == false) {
if (xQueueSend(request_queue, &async_req, pdMS_TO_TICKS(100)) == false) {
ESP_LOGE(TAG, "worker queue is full");
httpd_req_async_handler_complete(copy); // cleanup
return ESP_FAIL;
Expand All @@ -106,25 +92,10 @@ static esp_err_t submit_async_req(httpd_req_t *req, httpd_req_handler_t handler)
return ESP_OK;
}


/* A long running HTTP GET handler */
static esp_err_t long_async_handler(httpd_req_t *req)
/* handle long request (on async thread) */
static esp_err_t long_async(httpd_req_t *req)
{
ESP_LOGI(TAG, "uri: /long");
// This handler is first invoked on the httpd thread.
// In order to free the httpd thread to handle other requests,
// we must resubmit our request to be handled on an async worker thread.
if (is_on_async_worker_thread() == false) {

// submit
if (submit_async_req(req, long_async_handler) == ESP_OK) {
return ESP_OK;
} else {
httpd_resp_set_status(req, "503 Busy");
httpd_resp_sendstr(req, "<div> no workers available. server busy.</div>");
return ESP_OK;
}
}
ESP_LOGI(TAG, "running: /long");

// track the number of long requests
static uint8_t req_count = 0;
Expand Down Expand Up @@ -154,7 +125,8 @@ static esp_err_t long_async_handler(httpd_req_t *req)
return ESP_OK;
}

static void async_req_worker_task(void *p)
// each worker thread loops forever, processing requests
static void worker_task(void *p)
{
ESP_LOGI(TAG, "starting async req task worker");

Expand All @@ -166,7 +138,7 @@ static void async_req_worker_task(void *p)

// wait for a request
httpd_async_req_t async_req;
if (xQueueReceive(async_req_queue, &async_req, portMAX_DELAY)) {
if (xQueueReceive(request_queue, &async_req, portMAX_DELAY)) {

ESP_LOGI(TAG, "invoking %s", async_req.req->uri);

Expand All @@ -185,7 +157,8 @@ static void async_req_worker_task(void *p)
vTaskDelete(NULL);
}

static void start_async_req_workers(void)
// start worker threads
static void start_workers(void)
{

// counting semaphore keeps track of available workers
Expand All @@ -198,17 +171,17 @@ static void start_async_req_workers(void)
}

// create queue
async_req_queue = xQueueCreate(1, sizeof(httpd_async_req_t));
if (async_req_queue == NULL){
ESP_LOGE(TAG, "Failed to create async_req_queue");
request_queue = xQueueCreate(1, sizeof(httpd_async_req_t));
if (request_queue == NULL){
ESP_LOGE(TAG, "Failed to create request_queue");
vSemaphoreDelete(worker_ready_count);
return;
}

// start worker tasks
for (int i = 0; i < CONFIG_EXAMPLE_MAX_ASYNC_REQUESTS; i++) {

bool success = xTaskCreate(async_req_worker_task, "async_req_worker",
bool success = xTaskCreate(worker_task, "async_req_worker",
ASYNC_WORKER_TASK_STACK_SIZE, // stack size
(void *)0, // argument
ASYNC_WORKER_TASK_PRIORITY, // priority
Expand All @@ -221,6 +194,20 @@ static void start_async_req_workers(void)
}
}

/* adds /long request to the request queue */
static esp_err_t long_handler(httpd_req_t *req)
{
ESP_LOGI(TAG, "uri: /long");

// add to the async request queue
if (queue_request(req, long_async) == ESP_OK) {
return ESP_OK;
} else {
httpd_resp_set_status(req, "503 Busy");
httpd_resp_sendstr(req, "<div> no workers available. server busy.</div>");
return ESP_OK;
}
}

/* A quick HTTP GET handler, which does not
use any asynchronous features */
Expand Down Expand Up @@ -271,7 +258,7 @@ static httpd_handle_t start_webserver(void)
const httpd_uri_t long_uri = {
.uri = "/long",
.method = HTTP_GET,
.handler = long_async_handler,
.handler = long_handler,
};

const httpd_uri_t quick_uri = {
Expand Down Expand Up @@ -346,7 +333,7 @@ void app_main(void)
#endif // CONFIG_EXAMPLE_CONNECT_ETHERNET

// start workers
start_async_req_workers();
start_workers();

/* Start the server for the first time */
server = start_webserver();
Expand Down

0 comments on commit a89a954

Please sign in to comment.