#include "BufferPool.h"

#include <stdio.h>
#include <string.h>


#define BLOCK_SIZE 2048
#define NUM_THREADS 10
#define NUM_LOOPS 50


int32 gID;


int32
allocator(void *_pool)
{
	BufferPool &pool = *(BufferPool *)_pool;
	int32 id = atomic_add(&gID, 1);

	for (int32 i = 0; i < NUM_LOOPS; i++) {
		snooze(50000);

		printf("%ld. Get buffer...\n", id);
		status_t status;
		void *buffer;
		if ((status = pool.GetBuffer(&buffer)) != B_OK) {
			printf("\t%ld. Couldn't get buffer: %s\n", id, strerror(status));
			continue;
		}
		printf("\t%ld. got buffer\n", id);
		snooze(50000);
		pool.PutBuffer(buffer);
		printf("\t%ld. released buffer\n", id);
	}
	return 0;
}


int
main(int argc, char **argv)
{
	BufferPool pool;
	thread_id thread[NUM_THREADS];

	if (pool.RequestBuffers(BLOCK_SIZE) == B_OK) {
		for (int i = 0; i < NUM_THREADS; i++) {
			thread[i] = spawn_thread(allocator, "", B_NORMAL_PRIORITY, (void *)&pool);
			if (thread[i] < B_OK)
				fprintf(stderr, "Couldn't spawn thread %d\n", i);
			resume_thread(thread[i]);
		}

		for (int32 i = 0; i < NUM_THREADS; i++) {
			status_t status;
			wait_for_thread(thread[i], &status);
		}

		pool.ReleaseBuffers();
	}

	return 0;
}