mirror of
https://github.com/lxsang/ant-http
synced 2024-12-25 16:28:21 +01:00
use single thread scheduler and worker
This commit is contained in:
parent
6b83f363f8
commit
90d44d782f
2
Makefile
2
Makefile
@ -33,7 +33,7 @@ httpd: lib $(SERVER_O)
|
|||||||
|
|
||||||
relay: lib $(SERVER_O)
|
relay: lib $(SERVER_O)
|
||||||
$(CC) $(CFLAGS) $(SERVER_O) -o $(BUILDIRD)/relay relay.c $(SERVERLIB)
|
$(CC) $(CFLAGS) $(SERVER_O) -o $(BUILDIRD)/relay relay.c $(SERVERLIB)
|
||||||
cp relayd $(BUILDIRD)
|
cp forward $(BUILDIRD)
|
||||||
lib: $(LIBOBJS)
|
lib: $(LIBOBJS)
|
||||||
$(CC) $(CFLAGS) $(DB_LIB) $(SSL_LIB) -shared -o $(LIB_NAME).$(EXT) $(LIBOBJS)
|
$(CC) $(CFLAGS) $(DB_LIB) $(SSL_LIB) -shared -o $(LIB_NAME).$(EXT) $(LIBOBJS)
|
||||||
cp $(LIB_NAME).$(EXT) $(LIB_PATH$)/
|
cp $(LIB_NAME).$(EXT) $(LIB_PATH$)/
|
||||||
|
@ -80,28 +80,6 @@ static int available_workers()
|
|||||||
pthread_mutex_unlock(&scheduler.server_lock);
|
pthread_mutex_unlock(&scheduler.server_lock);
|
||||||
return n;
|
return n;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void work(void* data)
|
|
||||||
{
|
|
||||||
antd_task_item_t it;
|
|
||||||
antd_worker_t* worker = (antd_worker_t*)data;
|
|
||||||
while(working())
|
|
||||||
{
|
|
||||||
pthread_mutex_lock(&scheduler.server_lock);
|
|
||||||
worker->status = 0;
|
|
||||||
pthread_mutex_unlock(&scheduler.server_lock);
|
|
||||||
// fetch the next in queue
|
|
||||||
it = next_task();
|
|
||||||
if(!it) continue;
|
|
||||||
//LOG("worker processing \n");
|
|
||||||
pthread_mutex_lock(&scheduler.server_lock);
|
|
||||||
worker->status = 1;
|
|
||||||
pthread_mutex_unlock(&scheduler.server_lock);
|
|
||||||
// execute the task
|
|
||||||
antd_execute_task(it);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static antd_callback_t* callback_of( void* (*callback)(void*) )
|
static antd_callback_t* callback_of( void* (*callback)(void*) )
|
||||||
{
|
{
|
||||||
antd_callback_t* cb = NULL;
|
antd_callback_t* cb = NULL;
|
||||||
@ -151,33 +129,64 @@ static void execute_callback(antd_task_t* task)
|
|||||||
free(task);
|
free(task);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
static void work(void* data)
|
||||||
|
{
|
||||||
|
antd_worker_t* worker = (antd_worker_t*)data;
|
||||||
|
while(working())
|
||||||
|
{
|
||||||
|
antd_attach_task(worker);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
Main API methods
|
Main API methods
|
||||||
init the main scheduler
|
init the main scheduler
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
/*
|
||||||
|
* assign task to a worker
|
||||||
|
*/
|
||||||
|
void antd_attach_task(antd_worker_t* worker)
|
||||||
|
{
|
||||||
|
antd_task_item_t it;
|
||||||
|
pthread_mutex_lock(&scheduler.server_lock);
|
||||||
|
worker->status = 0;
|
||||||
|
pthread_mutex_unlock(&scheduler.server_lock);
|
||||||
|
// fetch the next in queue
|
||||||
|
it = next_task();
|
||||||
|
if(!it) return;
|
||||||
|
//LOG("worker processing \n");
|
||||||
|
pthread_mutex_lock(&scheduler.server_lock);
|
||||||
|
worker->status = 1;
|
||||||
|
pthread_mutex_unlock(&scheduler.server_lock);
|
||||||
|
// execute the task
|
||||||
|
antd_execute_task(it);
|
||||||
|
}
|
||||||
|
|
||||||
void antd_scheduler_init(int n)
|
void antd_scheduler_init(int n)
|
||||||
{
|
{
|
||||||
time_t t;
|
time_t t;
|
||||||
srand((unsigned) time(&t));
|
srand((unsigned) time(&t));
|
||||||
scheduler.n_workers = n;
|
scheduler.n_workers = n;
|
||||||
scheduler.status = 1;
|
scheduler.status = 1;
|
||||||
scheduler.workers = (antd_worker_t*)malloc(n*(sizeof(antd_worker_t)));
|
|
||||||
if(!scheduler.workers)
|
|
||||||
{
|
|
||||||
LOG("Cannot allocate memory for worker\n");
|
|
||||||
exit(-1);
|
|
||||||
}
|
|
||||||
for(int i = 0; i < N_PRIORITY; i++) scheduler.task_queue[i] = NULL;
|
for(int i = 0; i < N_PRIORITY; i++) scheduler.task_queue[i] = NULL;
|
||||||
// create scheduler.workers
|
// create scheduler.workers
|
||||||
for(int i = 0; i < scheduler.n_workers;i++)
|
if(n > 0)
|
||||||
{
|
{
|
||||||
scheduler.workers[i].status = 0;
|
scheduler.workers = (antd_worker_t*)malloc(n*(sizeof(antd_worker_t)));
|
||||||
if (pthread_create(&scheduler.workers[i].pid , NULL,(void *(*)(void *))work, (void*)&scheduler.workers[i]) != 0)
|
if(!scheduler.workers)
|
||||||
{
|
{
|
||||||
scheduler.workers[i].status = -1;
|
LOG("Cannot allocate memory for worker\n");
|
||||||
perror("pthread_create: cannot create worker\n");
|
exit(-1);
|
||||||
|
}
|
||||||
|
for(int i = 0; i < scheduler.n_workers;i++)
|
||||||
|
{
|
||||||
|
scheduler.workers[i].status = 0;
|
||||||
|
if (pthread_create(&scheduler.workers[i].pid , NULL,(void *(*)(void *))work, (void*)&scheduler.workers[i]) != 0)
|
||||||
|
{
|
||||||
|
scheduler.workers[i].status = -1;
|
||||||
|
perror("pthread_create: cannot create worker\n");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
LOG("Antd scheduler initialized with %d worker\n", scheduler.n_workers);
|
LOG("Antd scheduler initialized with %d worker\n", scheduler.n_workers);
|
||||||
@ -197,6 +206,7 @@ void antd_scheduler_destroy()
|
|||||||
{
|
{
|
||||||
// free all the chains
|
// free all the chains
|
||||||
antd_task_item_t it, curr;
|
antd_task_item_t it, curr;
|
||||||
|
stop();
|
||||||
for(int i=0; i < N_PRIORITY; i++)
|
for(int i=0; i < N_PRIORITY; i++)
|
||||||
{
|
{
|
||||||
it = scheduler.task_queue[i];
|
it = scheduler.task_queue[i];
|
||||||
@ -211,7 +221,6 @@ void antd_scheduler_destroy()
|
|||||||
free(curr);
|
free(curr);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
stop();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
|
@ -94,4 +94,5 @@ int antd_scheduler_status();
|
|||||||
void antd_execute_task(antd_task_item_t);
|
void antd_execute_task(antd_task_item_t);
|
||||||
|
|
||||||
int antd_scheduler_busy();
|
int antd_scheduler_busy();
|
||||||
|
void antd_attach_task(antd_worker_t* worker);
|
||||||
#endif
|
#endif
|
22
relay.c
22
relay.c
@ -1,5 +1,6 @@
|
|||||||
#include "http_server.h"
|
#include "http_server.h"
|
||||||
#include "libs/scheduler.h"
|
#include "libs/scheduler.h"
|
||||||
|
#include <fcntl.h>
|
||||||
/*
|
/*
|
||||||
this node is a relay from the http
|
this node is a relay from the http
|
||||||
to https
|
to https
|
||||||
@ -65,7 +66,7 @@ void* antd_get_host(void * client)
|
|||||||
void** data = (void**)malloc(2*(sizeof *data));
|
void** data = (void**)malloc(2*(sizeof *data));
|
||||||
data[0] = client;
|
data[0] = client;
|
||||||
data[1] = (void*)host;
|
data[1] = (void*)host;
|
||||||
LOG("Host is %s\n", host);
|
LOG("[%s] Request for: %s --> https://%s\n", ((antd_client_t*)client)->ip, host, host);
|
||||||
return antd_create_task(antd_redirect,data, NULL);
|
return antd_create_task(antd_redirect,data, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -84,23 +85,28 @@ int main(int argc, char* argv[])
|
|||||||
signal(SIGABRT, SIG_IGN);
|
signal(SIGABRT, SIG_IGN);
|
||||||
signal(SIGINT, stop_serve);
|
signal(SIGINT, stop_serve);
|
||||||
server_sock = startup(&port);
|
server_sock = startup(&port);
|
||||||
// 1 worker is
|
struct timeval timeout;
|
||||||
antd_scheduler_init(1);
|
timeout.tv_sec = 0;
|
||||||
LOG("httpd running on port %d\n", port);
|
timeout.tv_usec = 500;
|
||||||
|
// 0 worker
|
||||||
|
antd_scheduler_init(0);
|
||||||
|
antd_worker_t worker;
|
||||||
|
worker.status = 0;
|
||||||
|
// set server socket to non blocking
|
||||||
|
fcntl(server_sock, F_SETFL, O_NONBLOCK); /* Change the socket into non-blocking state */
|
||||||
|
LOG("relayd running on port %d\n", port);
|
||||||
|
|
||||||
while (antd_scheduler_status())
|
while (antd_scheduler_status())
|
||||||
{
|
{
|
||||||
|
// execute task
|
||||||
|
antd_attach_task(&worker);
|
||||||
client_sock = accept(server_sock,(struct sockaddr *)&client_name,&client_name_len);
|
client_sock = accept(server_sock,(struct sockaddr *)&client_name,&client_name_len);
|
||||||
if (client_sock == -1)
|
if (client_sock == -1)
|
||||||
{
|
{
|
||||||
perror("Cannot accept client request\n");
|
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
antd_client_t* client = (antd_client_t*)malloc(sizeof(antd_client_t));
|
antd_client_t* client = (antd_client_t*)malloc(sizeof(antd_client_t));
|
||||||
// set timeout to socket
|
// set timeout to socket
|
||||||
struct timeval timeout;
|
|
||||||
timeout.tv_sec = 0;
|
|
||||||
timeout.tv_usec = 500;
|
|
||||||
|
|
||||||
if (setsockopt (client_sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout,sizeof(timeout)) < 0)
|
if (setsockopt (client_sock, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout,sizeof(timeout)) < 0)
|
||||||
perror("setsockopt failed\n");
|
perror("setsockopt failed\n");
|
||||||
|
Loading…
Reference in New Issue
Block a user