summaryrefslogtreecommitdiff
path: root/server/src/muse_server_workqueue.c
blob: ef90e6eaccd27db910c9ec0f5284dc8abbfe3ed2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
/**
* Multithreaded work queue.
* Copyright (c) 2012 Ronald Bennett Cemer
* This software is licensed under the BSD license.
* See the accompanying LICENSE.txt for details.
*/

#include "muse_server_private.h"

#define LL_ADD(item, list) { \
	item->prev = NULL; \
	item->next = list; \
	list = item; \
}

#define LL_REMOVE(item, list) { \
	if (item->prev) item->prev->next = item->next; \
	if (item->next) item->next->prev = item->prev; \
	if (list == item) list = item->next; \
	item->prev = item->next = NULL; \
}

static void *_ms_workqueue_worker_function(void *ptr);

static void *_ms_workqueue_worker_function(void *ptr)
{
	ms_workqueue_worker_t *worker = (ms_workqueue_worker_t *) ptr;
	ms_workqueue_job_t *job;

	muse_return_val_if_fail(worker, NULL);

	while (1) {
		pthread_mutex_lock(&worker->workqueue->jobs_mutex);

		if (worker->terminate) {
			LOGW("The value (%d) to be terminated is set", worker->terminate);
			pthread_mutex_unlock(&worker->workqueue->jobs_mutex);
			break;
		}

		while (!worker->workqueue->waiting_jobs)
			pthread_cond_wait(&worker->workqueue->jobs_cond, &worker->workqueue->jobs_mutex);

		job = worker->workqueue->waiting_jobs;

		LL_REMOVE(job, worker->workqueue->waiting_jobs);
		LOGD("remove job %p", job);

		if (ms_is_server_ready() && !job->job_function(job)) {
			pthread_mutex_unlock(&worker->workqueue->jobs_mutex);
			LOGE("Error - Execute the workqueue job, Restart muse server...");
			ms_respawn(SIGABRT);
			break;
		}
		pthread_mutex_unlock(&worker->workqueue->jobs_mutex);
	}

	free(worker);

	LOGW("thread exit...");
	pthread_exit(NULL);
}

void ms_workqueue_init(ms_workqueue_t *workqueue)
{
	ms_workqueue_worker_t *worker;
	int idx;
	pthread_attr_t attr;

	LOGD("Enter");

	muse_return_if_fail(workqueue);

	pthread_mutex_init(&workqueue->jobs_mutex, NULL);
	pthread_cond_init(&workqueue->jobs_cond, NULL);

	muse_return_if_fail(pthread_attr_init(&attr) == 0);
	if (pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED) != 0) {
		LOGE("Failed to set detach state");
		goto out;
	}

	for (idx = 0; idx < MUSE_WORK_THREAD_NUM; idx++) {
		worker = calloc(1, sizeof(ms_workqueue_worker_t));
		muse_return_if_fail(worker);

		worker->workqueue = workqueue;
		if (pthread_create(&worker->thread, &attr, _ms_workqueue_worker_function, (void *)worker)) {
			LOGE("Failed to start all worker threads");
			free(worker);
			goto out;
		}

		LL_ADD(worker, worker->workqueue->workers);
	}

out:
	muse_return_if_fail(pthread_attr_destroy(&attr) == 0);

	LOGD("Leave");
}

void ms_workqueue_deinit(ms_workqueue_t *workqueue)
{
	ms_workqueue_worker_t *worker = NULL;

	LOGD("Enter");

	muse_return_if_fail(workqueue);

	/* Set all workers to terminate. */
	for (worker = workqueue->workers; worker; worker = worker->next)
		worker->terminate = 1;

	pthread_mutex_lock(&workqueue->jobs_mutex);
	workqueue->workers = NULL;
	workqueue->waiting_jobs = NULL;
	pthread_cond_broadcast(&workqueue->jobs_cond);
	pthread_mutex_unlock(&workqueue->jobs_mutex);

	pthread_mutex_destroy(&workqueue->jobs_mutex);
	pthread_cond_destroy(&workqueue->jobs_cond);
	free(workqueue);

	LOGD("Leave");
}

void ms_workqueue_add_job(ms_workqueue_t *workqueue, ms_workqueue_job_t *job)
{
	muse_return_if_fail(workqueue);

	pthread_mutex_lock(&workqueue->jobs_mutex);
	LL_ADD(job, workqueue->waiting_jobs);
	LOGD("add job %p", job);
	pthread_cond_signal(&workqueue->jobs_cond);
	pthread_mutex_unlock(&workqueue->jobs_mutex);
}