aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src/server.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/server.c')
-rw-r--r--src/server.c113
1 files changed, 82 insertions, 31 deletions
diff --git a/src/server.c b/src/server.c
index 509d144..b5c7463 100644
--- a/src/server.c
+++ b/src/server.c
@@ -2,6 +2,7 @@
#include "ci_queue.h"
#include "log.h"
#include "msg.h"
+#include "signal.h"
#include "tcp_server.h"
#include "worker_queue.h"
@@ -64,28 +65,31 @@ static void *server_scheduler(void *_server)
struct server *server = (struct server *)_server;
int ret = 0;
- ret = pthread_mutex_lock(&server->scheduler_mtx);
+ ret = pthread_mutex_lock(&server->server_mtx);
if (ret) {
pthread_print_errno(ret, "pthread_mutex_lock");
goto exit;
}
while (1) {
- while (!server_has_runs_and_workers(server)) {
- ret = pthread_cond_wait(&server->scheduler_cv, &server->scheduler_mtx);
+ while (!server->stopping && !server_has_runs_and_workers(server)) {
+ ret = pthread_cond_wait(&server->server_cv, &server->server_mtx);
if (ret) {
pthread_print_errno(ret, "pthread_cond_wait");
goto unlock;
}
}
+ if (server->stopping)
+ goto unlock;
+
ret = server_scheduler_iteration(server);
if (ret < 0)
goto unlock;
}
unlock:
- pthread_check(pthread_mutex_unlock(&server->scheduler_mtx), "pthread_mutex_unlock");
+ pthread_check(pthread_mutex_unlock(&server->server_mtx), "pthread_mutex_unlock");
exit:
return NULL;
@@ -93,60 +97,81 @@ exit:
int server_create(struct server *server, const struct settings *settings)
{
+ pthread_attr_t scheduler_attr;
int ret = 0;
+ ret = pthread_mutex_init(&server->server_mtx, NULL);
+ if (ret) {
+ pthread_print_errno(ret, "pthread_mutex_init");
+ goto fail;
+ }
+
+ ret = pthread_cond_init(&server->server_cv, NULL);
+ if (ret) {
+ pthread_print_errno(ret, "pthread_cond_init");
+ goto destroy_mtx;
+ }
+
+ server->stopping = 0;
+
ret = tcp_server_create(&server->tcp_server, settings->port);
if (ret < 0)
- return ret;
+ goto destroy_cv;
worker_queue_create(&server->worker_queue);
ci_queue_create(&server->ci_queue);
- ret = pthread_mutex_init(&server->scheduler_mtx, NULL);
+ ret = pthread_attr_init(&scheduler_attr);
if (ret) {
- pthread_print_errno(ret, "pthread_mutex_init");
- goto destroy_tcp_server;
+ pthread_print_errno(ret, "pthread_attr_init");
+ goto destroy_ci_queue;
}
- ret = pthread_cond_init(&server->scheduler_cv, NULL);
- if (ret) {
- pthread_print_errno(ret, "pthread_cond_init");
- goto destroy_scheduler_mtx;
- }
+ ret = signal_set_thread_attr(&scheduler_attr);
+ if (ret)
+ goto destroy_attr;
- ret = pthread_create(&server->scheduler, NULL, server_scheduler, server);
+ ret = pthread_create(&server->scheduler, &scheduler_attr, server_scheduler, server);
if (ret) {
pthread_print_errno(ret, "pthread_create");
- goto destroy_scheduler_cv;
+ goto destroy_attr;
}
- return ret;
+ pthread_check(pthread_attr_destroy(&scheduler_attr), "pthread_attr_destroy");
-destroy_scheduler_cv:
- pthread_check(pthread_cond_destroy(&server->scheduler_cv), "pthread_cond_destroy");
+ return ret;
-destroy_scheduler_mtx:
- pthread_check(pthread_mutex_destroy(&server->scheduler_mtx), "pthread_mutex_destroy");
+destroy_attr:
+ pthread_check(pthread_attr_destroy(&scheduler_attr), "pthread_attr_destroy");
+destroy_ci_queue:
ci_queue_destroy(&server->ci_queue);
worker_queue_destroy(&server->worker_queue);
-destroy_tcp_server:
tcp_server_destroy(&server->tcp_server);
+destroy_cv:
+ pthread_check(pthread_cond_destroy(&server->server_cv), "pthread_cond_destroy");
+
+destroy_mtx:
+ pthread_check(pthread_mutex_destroy(&server->server_mtx), "pthread_mutex_destroy");
+
+fail:
return ret;
}
void server_destroy(struct server *server)
{
+ print_log("Shutting down\n");
+
pthread_check(pthread_join(server->scheduler, NULL), "pthread_join");
- pthread_check(pthread_cond_destroy(&server->scheduler_cv), "pthread_cond_destroy");
- pthread_check(pthread_mutex_destroy(&server->scheduler_mtx), "pthread_mutex_destroy");
ci_queue_destroy(&server->ci_queue);
worker_queue_destroy(&server->worker_queue);
tcp_server_destroy(&server->tcp_server);
+ pthread_check(pthread_cond_destroy(&server->server_cv), "pthread_cond_destroy");
+ pthread_check(pthread_mutex_destroy(&server->server_mtx), "pthread_mutex_destroy");
}
struct msg_context {
@@ -193,15 +218,41 @@ static int server_conn_handler(int fd, void *server)
return msg_recv_and_handle(fd, server_msg_handler, &ctx);
}
+static int server_set_stopping(struct server *server)
+{
+ int ret = 0;
+
+ ret = pthread_mutex_lock(&server->server_mtx);
+ if (ret) {
+ pthread_print_errno(ret, "pthread_mutex_lock");
+ return ret;
+ }
+
+ server->stopping = 1;
+
+ ret = pthread_cond_signal(&server->server_cv);
+ if (ret) {
+ pthread_print_errno(ret, "pthread_cond_signal");
+ goto unlock;
+ }
+
+unlock:
+ pthread_check(pthread_mutex_unlock(&server->server_mtx), "pthread_mutex_unlock");
+
+ return ret;
+}
+
int server_main(struct server *server)
{
int ret = 0;
- while (1) {
+ while (!global_stop_flag) {
ret = tcp_server_accept(&server->tcp_server, server_conn_handler, server);
if (ret < 0)
- return ret;
+ break;
}
+
+ return server_set_stopping(server);
}
int server_new_worker(struct server *server, int fd)
@@ -211,7 +262,7 @@ int server_new_worker(struct server *server, int fd)
print_log("Registering a new worker\n");
- ret = pthread_mutex_lock(&server->scheduler_mtx);
+ ret = pthread_mutex_lock(&server->server_mtx);
if (ret) {
pthread_print_errno(ret, "pthread_mutex_lock");
return ret;
@@ -223,14 +274,14 @@ int server_new_worker(struct server *server, int fd)
worker_queue_push(&server->worker_queue, entry);
- ret = pthread_cond_signal(&server->scheduler_cv);
+ ret = pthread_cond_signal(&server->server_cv);
if (ret) {
pthread_print_errno(ret, "pthread_cond_signal");
goto unlock;
}
unlock:
- pthread_check(pthread_mutex_unlock(&server->scheduler_mtx), "pthread_mutex_unlock");
+ pthread_check(pthread_mutex_unlock(&server->server_mtx), "pthread_mutex_unlock");
return ret;
}
@@ -242,7 +293,7 @@ int server_ci_run(struct server *server, const char *url, const char *rev)
print_log("Scheduling a new CI run for repository %s\n", url);
- ret = pthread_mutex_lock(&server->scheduler_mtx);
+ ret = pthread_mutex_lock(&server->server_mtx);
if (ret) {
pthread_print_errno(ret, "pthread_mutex_lock");
return ret;
@@ -254,14 +305,14 @@ int server_ci_run(struct server *server, const char *url, const char *rev)
ci_queue_push(&server->ci_queue, entry);
- ret = pthread_cond_signal(&server->scheduler_cv);
+ ret = pthread_cond_signal(&server->server_cv);
if (ret) {
pthread_print_errno(ret, "pthread_cond_signal");
goto unlock;
}
unlock:
- pthread_check(pthread_mutex_unlock(&server->scheduler_mtx), "pthread_mutex_unlock");
+ pthread_check(pthread_mutex_unlock(&server->server_mtx), "pthread_mutex_unlock");
return ret;
}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-22 11:36:14 +0000