Add a basic pthread mutex queue
This commit is contained in:
parent
5d4b0dc3c9
commit
e27740688a
8 changed files with 331 additions and 29 deletions
43
include/r3_queue.h
Normal file
43
include/r3_queue.h
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
/*
|
||||||
|
* r3_queue.h
|
||||||
|
* Copyright (C) 2014 c9s <c9s@c9smba.local>
|
||||||
|
*
|
||||||
|
* Distributed under terms of the MIT license.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef R3_QUEUE_H
|
||||||
|
#define R3_QUEUE_H
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
struct _queue_node {
|
||||||
|
void *data;
|
||||||
|
struct _queue_node * next;
|
||||||
|
};
|
||||||
|
typedef struct _queue_node queue_node;
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
queue_node * first;
|
||||||
|
queue_node * last;
|
||||||
|
} queue;
|
||||||
|
|
||||||
|
// create and return the queue
|
||||||
|
queue * queue_factory(void);
|
||||||
|
|
||||||
|
// destory the queue (free all the memory associate with the que even the data)
|
||||||
|
void queue_destroy(queue * que);
|
||||||
|
|
||||||
|
|
||||||
|
// enque the data into queue
|
||||||
|
// data is expected to a pointer to a heap allocated memory
|
||||||
|
int enque(queue * que, void * data);
|
||||||
|
|
||||||
|
// return the data from the que (FIFO)
|
||||||
|
// and free up all the internally allocated memory
|
||||||
|
// but the user have to free the returning data pointer
|
||||||
|
void * deque(queue * que);
|
||||||
|
|
||||||
|
#endif /* !R3_QUEUE_H */
|
|
@ -8,5 +8,19 @@
|
||||||
#ifndef R3_WORKERS_H
|
#ifndef R3_WORKERS_H
|
||||||
#define R3_WORKERS_H
|
#define R3_WORKERS_H
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
int thread_id;
|
||||||
|
node * matched_node;
|
||||||
|
} feedback_payload;
|
||||||
|
|
||||||
|
void r3_feedback_worker_init(pthread_t * t, feedback_payload * data);
|
||||||
|
|
||||||
|
void r3_worker_cancel(pthread_t t);
|
||||||
|
|
||||||
|
void r3_worker_exit();
|
||||||
|
|
||||||
#endif /* !R3_WORKERS_H */
|
#endif /* !R3_WORKERS_H */
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
lib_LTLIBRARIES = libr3.la
|
lib_LTLIBRARIES = libr3.la
|
||||||
# lib_LIBRARIES = libr3.a
|
# lib_LIBRARIES = libr3.a
|
||||||
libr3_la_SOURCES = node.c edge.c str.c token.c zmalloc.c workers.c
|
libr3_la_SOURCES = node.c edge.c str.c token.c zmalloc.c workers.c queue.c
|
||||||
# libr3_la_LDFLAGS = -export-symbols-regex '^r3_|^match_'
|
# libr3_la_LDFLAGS = -export-symbols-regex '^r3_|^match_'
|
||||||
|
|
||||||
libr3_la_LIBADD=$(DEPS_LIBS)
|
libr3_la_LIBADD=$(DEPS_LIBS)
|
||||||
|
|
121
src/queue.c
Normal file
121
src/queue.c
Normal file
|
@ -0,0 +1,121 @@
|
||||||
|
/*
|
||||||
|
* queue.c
|
||||||
|
* Copyright (C) 2014 c9s <c9s@c9smba.local>
|
||||||
|
*
|
||||||
|
* Distributed under terms of the MIT license.
|
||||||
|
*/
|
||||||
|
#include "r3_queue.h"
|
||||||
|
|
||||||
|
pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* create and return a new queue
|
||||||
|
**/
|
||||||
|
queue * queue_factory()
|
||||||
|
{
|
||||||
|
queue * new_queue = malloc(sizeof(queue));
|
||||||
|
if(new_queue == NULL) {
|
||||||
|
fprintf(stderr, "Malloc failed creating the que\n");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
new_queue->first = NULL;
|
||||||
|
new_queue->last = NULL;
|
||||||
|
return new_queue;
|
||||||
|
}
|
||||||
|
|
||||||
|
void queue_destroy(queue * que)
|
||||||
|
{
|
||||||
|
if(que == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
pthread_mutex_lock(&mutex);
|
||||||
|
if(que->first == NULL) {
|
||||||
|
// ("que->first == NULL .... \n");
|
||||||
|
free(que);
|
||||||
|
pthread_mutex_unlock(&mutex);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ("que is there lets try to free it...\n");
|
||||||
|
|
||||||
|
queue_node * _node = que->first;
|
||||||
|
|
||||||
|
while(_node != NULL) {
|
||||||
|
// freeing the data coz it's on the heap and no one to free it
|
||||||
|
// except for this one
|
||||||
|
// ("freeing : %s\n", (char *)_node->data);
|
||||||
|
free(_node->data);
|
||||||
|
queue_node *tmp = _node->next;
|
||||||
|
free(_node);
|
||||||
|
_node = tmp;
|
||||||
|
}
|
||||||
|
|
||||||
|
free(que);
|
||||||
|
|
||||||
|
pthread_mutex_unlock(&mutex);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* que is a queue pointer
|
||||||
|
* data is a heap allocated memory pointer
|
||||||
|
*/
|
||||||
|
int enque(queue * que, void * data)
|
||||||
|
{
|
||||||
|
queue_node * new_node = malloc(sizeof(queue_node));
|
||||||
|
if(new_node == NULL) {
|
||||||
|
fprintf(stderr, "Malloc failed creating a queue_node\n");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
// assumming data is in heap
|
||||||
|
new_node->data = data;
|
||||||
|
new_node->next = NULL;
|
||||||
|
|
||||||
|
pthread_mutex_lock(&mutex);
|
||||||
|
if (que->first == NULL) {
|
||||||
|
// new que
|
||||||
|
que->first = new_node;
|
||||||
|
que->last = new_node;
|
||||||
|
} else {
|
||||||
|
que->last->next = new_node;
|
||||||
|
que->last = new_node;
|
||||||
|
}
|
||||||
|
pthread_mutex_unlock(&mutex);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void * deque(queue * que)
|
||||||
|
{
|
||||||
|
// print("Entered to deque\n");
|
||||||
|
if (que == NULL) {
|
||||||
|
// print("que is null exiting...\n");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
pthread_mutex_lock(&mutex);
|
||||||
|
if (que->first == NULL) {
|
||||||
|
pthread_mutex_unlock(&mutex);
|
||||||
|
// print("que->first is null exiting...\n");
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
void * data;
|
||||||
|
queue_node * _node = que->first;
|
||||||
|
if (que->first == que->last) {
|
||||||
|
que->first = NULL;
|
||||||
|
que->last = NULL;
|
||||||
|
} else {
|
||||||
|
que->first = _node->next;
|
||||||
|
}
|
||||||
|
|
||||||
|
data = _node->data;
|
||||||
|
|
||||||
|
// print("Freeing _node@ %p", _node);
|
||||||
|
free(_node);
|
||||||
|
pthread_mutex_unlock(&mutex);
|
||||||
|
// print("Exiting deque\n");
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
|
|
@ -11,37 +11,31 @@
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <pthread.h>
|
#include <pthread.h>
|
||||||
|
|
||||||
|
|
||||||
void *PrintHello(void *threadid)
|
|
||||||
{
|
|
||||||
long tid;
|
|
||||||
tid = (long)threadid;
|
|
||||||
printf("Hello World! It's me, thread #%ld!\n", tid);
|
|
||||||
pthread_exit(NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
int thread_id;
|
|
||||||
node * matched_node;
|
|
||||||
} feedback_payload;
|
|
||||||
|
|
||||||
void *r3_feedback_worker(void * data) {
|
void *r3_feedback_worker(void * data) {
|
||||||
feedback_payload * payload = (feedback_payload*) data; // pointer cast
|
feedback_payload * payload = (feedback_payload*) data; // pointer cast
|
||||||
|
|
||||||
return NULL;
|
pthread_exit(NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
void r3_launch_feedback_worker(feedback_payload * data) {
|
void r3_feedback_worker_init(pthread_t * t, feedback_payload * data) {
|
||||||
int rc;
|
int rc;
|
||||||
pthread_t worker_thread;
|
// pthread_t worker_thread;
|
||||||
pthread_attr_t attr;
|
pthread_attr_t attr;
|
||||||
pthread_attr_init(&attr);
|
pthread_attr_init(&attr);
|
||||||
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
|
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
|
||||||
rc = pthread_create(&worker_thread, &attr, r3_feedback_worker, (void*) data);
|
rc = pthread_create(t, &attr, r3_feedback_worker, (void*) data);
|
||||||
}
|
|
||||||
|
// if the pthread is created, we may free the attr
|
||||||
|
pthread_attr_destroy(&attr);
|
||||||
void r3_worker_stop() {
|
|
||||||
|
// rc = pthread_join(t, &status);
|
||||||
|
}
|
||||||
|
|
||||||
|
void r3_worker_cancel(pthread_t t) {
|
||||||
|
pthread_cancel(t);
|
||||||
|
}
|
||||||
|
|
||||||
|
void r3_worker_exit() {
|
||||||
pthread_exit(NULL);
|
pthread_exit(NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -46,6 +46,8 @@ check_slug_SOURCES = check_slug.c
|
||||||
TESTS += check_worker
|
TESTS += check_worker
|
||||||
check_worker_SOURCES = check_worker.c
|
check_worker_SOURCES = check_worker.c
|
||||||
|
|
||||||
|
TESTS += check_queue
|
||||||
|
check_queue_SOURCES = check_queue.c
|
||||||
|
|
||||||
check_PROGRAMS = $(TESTS)
|
check_PROGRAMS = $(TESTS)
|
||||||
|
|
||||||
|
|
117
tests/check_queue.c
Normal file
117
tests/check_queue.c
Normal file
|
@ -0,0 +1,117 @@
|
||||||
|
/*
|
||||||
|
* check_queue.c
|
||||||
|
* Copyright (C) 2014 c9s <c9s@c9smba.local>
|
||||||
|
*
|
||||||
|
* Distributed under terms of the MIT license.
|
||||||
|
*/
|
||||||
|
#include "r3_queue.h"
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <pthread.h>
|
||||||
|
#include <check.h>
|
||||||
|
|
||||||
|
|
||||||
|
#define number_of_threads 6
|
||||||
|
#define number_of_threads_d 5
|
||||||
|
#define number_of_iters 1000
|
||||||
|
|
||||||
|
struct _stru {
|
||||||
|
int number;
|
||||||
|
int thread_no;
|
||||||
|
queue * q;
|
||||||
|
};
|
||||||
|
|
||||||
|
void * func(void * arg)
|
||||||
|
{
|
||||||
|
struct _stru * args = (struct _stru *) arg;
|
||||||
|
int number = args->number;
|
||||||
|
//int th_number = args->thread_no;
|
||||||
|
queue * q = args->q;
|
||||||
|
|
||||||
|
int i;
|
||||||
|
srand(time(NULL));
|
||||||
|
for(i = 0; i < number; i++) {
|
||||||
|
char * message = malloc(16);
|
||||||
|
snprintf(message, 15, "rand: %d", rand());
|
||||||
|
enque(q, (void *)message);
|
||||||
|
}
|
||||||
|
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void * func_d(void * args)
|
||||||
|
{
|
||||||
|
queue * q = (queue *) args;
|
||||||
|
|
||||||
|
void * data;
|
||||||
|
while((data = deque(q)) != NULL) {
|
||||||
|
char * string = (char *)data;
|
||||||
|
free(data);
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
START_TEST (test_queue)
|
||||||
|
{
|
||||||
|
queue * q = queue_factory();
|
||||||
|
enque(q, (void*) 1);
|
||||||
|
int i = (int) deque(q);
|
||||||
|
ck_assert_int_eq(i, 1);
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
|
START_TEST (test_queue_threads)
|
||||||
|
{
|
||||||
|
queue * q = queue_factory();
|
||||||
|
pthread_t threads[number_of_threads];
|
||||||
|
pthread_t thread_d[number_of_threads_d];
|
||||||
|
|
||||||
|
int i;
|
||||||
|
struct _stru arg[number_of_threads];
|
||||||
|
for(i = 0; i < number_of_threads; i++) {
|
||||||
|
arg[i].number = number_of_iters;
|
||||||
|
arg[i].thread_no = i;
|
||||||
|
arg[i].q = q;
|
||||||
|
pthread_create(threads+i, NULL, func, (void *)&arg[i]);
|
||||||
|
}
|
||||||
|
|
||||||
|
for(i = 0; i < number_of_threads_d; i++) {
|
||||||
|
pthread_create(thread_d+i, NULL, func_d, (void *)q);
|
||||||
|
}
|
||||||
|
|
||||||
|
for(i = 0; i < number_of_threads; i++) {
|
||||||
|
pthread_join(*(threads+i), NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
for(i = 0; i < number_of_threads_d; i++) {
|
||||||
|
pthread_join(*(thread_d+i), NULL);
|
||||||
|
}
|
||||||
|
queue_destroy(q);
|
||||||
|
}
|
||||||
|
END_TEST
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
Suite* r3_suite (void) {
|
||||||
|
Suite *suite = suite_create("queue");
|
||||||
|
TCase *tcase = tcase_create("queue_test");
|
||||||
|
tcase_add_test(tcase, test_queue_threads);
|
||||||
|
tcase_add_test(tcase, test_queue);
|
||||||
|
tcase_set_timeout(tcase, 30);
|
||||||
|
suite_add_tcase(suite, tcase);
|
||||||
|
return suite;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main (int argc, char *argv[]) {
|
||||||
|
int number_failed;
|
||||||
|
Suite *suite = r3_suite();
|
||||||
|
SRunner *runner = srunner_create(suite);
|
||||||
|
srunner_run_all(runner, CK_NORMAL);
|
||||||
|
number_failed = srunner_ntests_failed(runner);
|
||||||
|
srunner_free(runner);
|
||||||
|
return number_failed;
|
||||||
|
}
|
|
@ -27,6 +27,17 @@ START_TEST (test_feedback_worker)
|
||||||
r3_tree_insert_path(n, "/garply/grault/bar", NULL);
|
r3_tree_insert_path(n, "/garply/grault/bar", NULL);
|
||||||
r3_tree_compile(n);
|
r3_tree_compile(n);
|
||||||
|
|
||||||
|
node *matched;
|
||||||
|
matched = r3_tree_matchl(n, "/garply/grault/foo", strlen("/garply/grault/foo"), NULL);
|
||||||
|
|
||||||
|
/*
|
||||||
|
feedback_payload payload;
|
||||||
|
payload.matched_node = matched;
|
||||||
|
r3_feedback_worker_init(&payload);
|
||||||
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
r3_tree_free(n);
|
r3_tree_free(n);
|
||||||
}
|
}
|
||||||
END_TEST
|
END_TEST
|
||||||
|
|
Loading…
Reference in a new issue