ts_queue_test.cpp

#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <assert.h>
#include "ts_queue.hpp"

/* Global shared variables */
TSQueue<int> *q;
int num_producer;
int num_consumer;
int **result;

void *produce(void *arg)
{
	int tid = *(int *)arg;

	int from = tid * num_consumer;
	int to = tid * num_consumer + num_consumer;
	for (int i = from; i < to; i++)
	{
		q->enqueue(i);
	}

	return nullptr;
}

void *consume(void *arg)
{
	int tid = *(int *)arg;

	for (int i = 0; i < num_producer; i++)
	{
		int val = q->dequeue();
		result[tid][i] = val;
	}

	return nullptr;
}

struct Thread
{
	pthread_t t;
	int id;
};

int main(int argc, char **argv)
{
	assert(argc == 3);

	q = new TSQueue<int>(20);
	num_producer = atoi(argv[1]);
	num_consumer = atoi(argv[2]);

	result = new int *[num_consumer];
	for (int i = 0; i < num_consumer; i++)
		result[i] = new int[num_producer];

	Thread *producers = new Thread[num_producer];
	Thread *consumers = new Thread[num_consumer];

	for (int i = 0; i < num_producer; i++)
	{
		producers[i].id = i;
		pthread_create(&producers[i].t, 0, produce, (void *)&producers[i].id);
	}

	for (int i = 0; i < num_consumer; i++)
	{
		consumers[i].id = i;
		pthread_create(&consumers[i].t, 0, consume, (void *)&consumers[i].id);
	}

	for (int i = 0; i < num_producer; i++)
	{
		pthread_join(producers[i].t, 0);
	}
	for (int i = 0; i < num_consumer; i++)
	{
		pthread_join(consumers[i].t, 0);
	}

	for (int i = 0; i < num_consumer; i++)
	{
		printf("consumer %d:", i);
		for (int j = 0; j < num_producer; j++)
			printf(" %d", result[i][j]);
		printf("\n");
	}

	return 0;
}