Make this bih multithreaded
This commit is contained in:
72
main.c
72
main.c
@ -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 (;;) {
|
||||
|
Reference in New Issue
Block a user