Make this bih multithreaded

This commit is contained in:
kamkow1
2025-06-16 01:20:11 +02:00
parent bfcc66ff2a
commit a532d5b03d
4 changed files with 162 additions and 41 deletions

72
main.c
View File

@ -1,4 +1,5 @@
#include <libgen.h>
#include <pthread.h>
#define GEBS_IMPLEMENTATION
#include "gebs/gebs.h"
@ -9,24 +10,72 @@
#include "routes.h"
#include "baked.h"
typedef void (*Route_Handler)(struct mg_connection *conn, struct mg_http_message *msg);
typedef struct {
char *key; // path
Route_Handler value;
} Route;
Route *route_hashtable = NULL;
void run_in_thread(void *(*f)(void *), void *p)
{
pthread_t tid = 0;
pthread_attr_t attr;
pthread_attr_init(&attr);
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
pthread_create(&tid, &attr, f, p);
pthread_attr_destroy(&attr);
}
void *route_thread_function(void *param)
{
Route_Thread_Data *data = (Route_Thread_Data *)param;
struct mg_http_message http_msg = {0};
int r = mg_http_parse(data->message.buf, data->message.len, &http_msg);
if (r <= 0) {
Route_Result result = {0};
result.status_code = 400;
mg_wakeup(data->mgr, data->conn_id, &result, sizeof(result));
return nil;
}
char key[MG_PATH_MAX] = {0};
strncpy(key, http_msg.uri.buf, http_msg.uri.len);
Route_Handler handler = shget(route_hashtable, key);
Route_Result result = {0};
handler(&http_msg, &result);
mg_wakeup(data->mgr, data->conn_id, &result, sizeof(result));
free(data->message.buf);
free(data);
return nil;
}
void event_handler(struct mg_connection *conn, int ev, void *ev_data)
{
if (ev == MG_EV_HTTP_MSG) {
struct mg_http_message *msg = (struct mg_http_message *)ev_data;
char key[MG_PATH_MAX] = {0};
strncpy(key, msg->uri.buf, msg->uri.len);
Route_Handler handler = shget(route_hashtable, key);
handler(conn, msg);
Route_Thread_Data *data = calloc(1, sizeof(*data));
data->message = mg_strdup(msg->message);
data->conn_id = conn->id;
data->mgr = conn->mgr;
run_in_thread(&route_thread_function, data);
} else if (ev == MG_EV_WAKEUP) {
struct mg_str *data = (struct mg_str *)ev_data;
Route_Result *result = (Route_Result *)data->buf;
defer {
list_free(&result->headers);
sb_free(&result->body);
}
Gebs_String_Builder sb = {0};
defer { sb_free(&sb); }
nsl_join(&result->headers, &sb, "\r\n");
if (result->headers.count > 0) {
sb_append_nstr(&sb, "\r\n");
}
sb_finish(&sb);
mg_http_reply(conn, result->status_code, sb.items, "%s", result->body.items);
}
}
@ -50,6 +99,7 @@ int main(int argc, char ** argv)
mg_mgr_init(&mgr);
init_route_hashtable();
mg_wakeup_init(&mgr);
mg_http_listen(&mgr, "http://localhost:8080", &event_handler, NULL);
for (;;) {