jekyll-theme-gaeblogx 0.1.0

data/assets/market4.c

@@ -0,0 +1,509 @@
+/*
+ * cs170 -- Rich wolski
+ * producer-consumer example
+ * uses condition variables for fule/empty conditions
+ * uses condition variable to fulfill order
+ * uses separate lock for each stock
+ */
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <pthread.h>
+#include <string.h>
+
+#include "c-timer.h"
+
+#define RAND() (drand48())
+
+struct order
+{
+	int stock_id;
+	int quantity;
+	int action;	/* buy or sell */
+	int fulfilled;	
+	pthread_mutex_t lock;
+	pthread_cond_t finish;
+};
+
+struct order_que
+{
+	struct order **orders;
+	int size;
+	int head;
+	int tail;
+	pthread_mutex_t lock;
+	pthread_cond_t full;
+	pthread_cond_t empty;
+};
+
+struct stock
+{
+	pthread_mutex_t lock;
+	int quantity;
+};
+
+struct market
+{
+	struct stock *stocks;
+	int count;
+};
+
+struct order *InitOrder(int id, int quantity, int action)
+{
+	struct order *order;
+
+	order = (struct order *)malloc(sizeof(struct order));
+	if(order == NULL) {
+		return(NULL);
+	}
+	order->stock_id = id;
+	order->quantity = quantity;
+	order->action = action;
+	order->fulfilled = 0;
+	pthread_mutex_init(&order->lock,NULL);
+	pthread_cond_init(&order->finish,NULL);
+	return(order);
+}
+
+void FreeOrder(struct order *order)
+{
+	free(order);
+}
+
+struct order_que *InitOrderQue(int size)
+{
+	struct order_que *oq;
+
+	oq = (struct order_que *)malloc(sizeof(struct order_que));
+	if(oq == NULL) {
+		return(NULL);
+	}
+	memset(oq,0,sizeof(struct order_que));
+
+	oq->size = size+1; /* empty condition burns a slot */
+	oq->orders = (struct order **)malloc(oq->size*sizeof(struct order *));
+	if(oq->orders == NULL) {
+		free(oq);
+		return(NULL);
+	}
+	memset(oq->orders,0,size*sizeof(struct order *));
+
+	pthread_mutex_init(&oq->lock,NULL);
+	pthread_cond_init(&oq->full,NULL);
+	pthread_cond_init(&oq->empty,NULL);
+
+	return(oq);
+}
+
+void FreeOrderQue(struct order_que *oq)
+{
+	while(oq->head != oq->tail) {
+		FreeOrder(oq->orders[oq->tail]);
+		oq->tail = (oq->tail + 1) % oq->size;
+	}
+
+	free(oq->orders);
+	free(oq);
+	return;
+}
+
+struct market *InitMarket(int stock_count, int init_quantity)
+{
+	struct market *m;
+	int i;
+
+	m = (struct market *)malloc(sizeof(struct market));
+	if(m == NULL) {
+		return(NULL);
+	}
+	m->count = stock_count;
+
+	m->stocks = (struct stock *)malloc(stock_count*sizeof(struct stock));
+	if(m->stocks == NULL) {
+		free(m);
+		return(NULL);
+	}
+
+	for(i=0; i < stock_count; i++) {
+		m->stocks[i].quantity = init_quantity;
+		pthread_mutex_init(&(m->stocks[i].lock),NULL);
+	}
+
+	return(m);
+}
+
+void FreeMarket(struct market *m)
+{
+	free(m->stocks);
+	free(m);
+	return;
+}
+
+void PrintMarket(struct market *m)
+{
+	int i;
+	for(i=0; i < m->count; i++) {
+		printf("stock: %d, quantity: %d\n",
+			i,m->stocks[i].quantity);
+	}
+
+	return;
+}
+
+struct client_arg
+{
+	int id;
+	int order_count;
+	struct order_que *order_que;
+	int max_stock_id;
+	int max_quantity;
+	int verbose;
+};
+
+struct trader_arg
+{
+	int id;
+	struct order_que *order_que;
+	struct market *market;
+	int *done;
+	int verbose;
+};
+
+void *ClientThread(void *arg)
+{
+	struct client_arg *ca = (struct client_arg *)arg;
+	int i;
+	int next;
+	struct order *order;
+	int stock_id;
+	int quantity;
+	int action;
+	int queued;
+	double now;
+	
+
+	for(i=0; i < ca->order_count; i++) {
+		/*
+		 * create an order for a random stock
+		 */
+		stock_id = (int)(RAND() * ca->max_stock_id);
+		quantity = (int)(RAND() * ca->max_quantity);
+		if(RAND() > 0.5) {
+			action = 0; /* 0 => buy */
+		} else {
+			action = 1; /* 1 => sell */
+		}
+		order = InitOrder(stock_id,quantity,action);
+		if(order == NULL) {
+			fprintf(stderr,"no space for order\n");
+			exit(1);
+		}
+		/*
+		 * queue it for the traders
+		 */
+		queued = 0;
+		while(queued == 0) {
+			pthread_mutex_lock(&(ca->order_que->lock));
+			next = (ca->order_que->head + 1) % ca->order_que->size;
+			/*
+			 * is the queue full?
+			 */
+			while(next == ca->order_que->tail) {
+				pthread_cond_wait(&(ca->order_que->full),
+						  &(ca->order_que->lock));
+				next = (ca->order_que->head + 1) % ca->order_que->size;
+			}
+			/*
+			 * there is space in the queue, add the order and bump
+			 * the head
+			 */
+			if(ca->verbose == 1) {
+				now = CTimer();
+				printf("%10.0f client %d: ",now,ca->id);
+				printf("queued stock %d, for %d, %s\n",
+					order->stock_id,
+					order->quantity,
+					(order->action ? "SELL" : "BUY")); 
+			}
+			ca->order_que->orders[next] = order;
+			ca->order_que->head = next;
+			queued = 1;
+			pthread_cond_signal(&(ca->order_que->empty));
+			pthread_mutex_unlock(&(ca->order_que->lock));
+
+			/*
+			 * spin waiting until the order is fulfilled
+			 */
+			pthread_mutex_lock(&order->lock);
+			while(order->fulfilled == 0) {
+				pthread_cond_wait(&order->finish,&order->lock);
+			}
+			pthread_mutex_unlock(&order->lock);
+			/*
+			 * done, free the order and repeat
+			 */
+			FreeOrder(order);
+		}
+	}
+
+	return(NULL);
+}
+
+void *TraderThread(void *arg)
+{
+	struct trader_arg *ta = (struct trader_arg *)arg;
+	int dequeued;
+	struct order *order;
+	int tail;
+	double now;
+	int next;
+	struct stock *stock;
+
+	while(1) {
+		dequeued = 0;
+		while(dequeued == 0) {
+			pthread_mutex_lock(&(ta->order_que->lock));
+			/*
+			 * is the queue empty?
+			 */
+			while(ta->order_que->head == ta->order_que->tail) {
+				/*
+				 * if the queue is empty, are we done?
+				 */
+				if(*(ta->done) == 1) {
+					pthread_cond_signal(&(ta->order_que->empty));
+					pthread_mutex_unlock(&(ta->order_que->lock));
+					pthread_exit(NULL);
+				}
+				pthread_cond_wait(&(ta->order_que->empty),
+						  &(ta->order_que->lock));
+			}
+			/*
+			 * get the next order
+			 */
+			next = (ta->order_que->tail + 1) % ta->order_que->size;
+			order = ta->order_que->orders[next];
+			ta->order_que->tail = next;
+			pthread_cond_signal(&(ta->order_que->full));
+			pthread_mutex_unlock(&(ta->order_que->lock));
+			dequeued = 1;
+		}
+		/*
+		 * have an order to process
+		 */
+		stock = &(ta->market->stocks[order->stock_id]);
+		pthread_mutex_lock(&(stock->lock));
+		if(order->action == 1) { /* BUY */
+			stock->quantity -= order->quantity;
+			if(stock->quantity < 0) {
+				stock->quantity = 0;
+			}
+		} else {
+			stock->quantity += order->quantity;
+		}
+		pthread_mutex_unlock(&(stock->lock));
+		if(ta->verbose == 1) {
+			now = CTimer();
+			printf("%10.0f trader: %d ",now,ta->id);
+			printf("fulfilled stock %d for %d\n",
+				order->stock_id,
+				order->quantity);
+		}
+		/*
+		 * tell the client the order is done
+		 */
+		pthread_mutex_lock(&order->lock);
+		order->fulfilled = 1;
+		pthread_cond_signal(&order->finish);
+		pthread_mutex_unlock(&order->lock);
+	}
+
+	return(NULL);
+}
+
+#define ARGS "c:t:o:q:s:V"
+char *Usage = "market1 -c clients -t traders -o orders -q queue-size -s stock-count -V <verbose on>\n";
+
+#define INIT_COUNT 5000
+
+int main(int argc, char **argv)
+{
+	int c;
+	int client_threads;
+	int trader_threads;
+	int orders_per_client;
+	int que_size;
+	int max_stock;
+	int verbose;
+	struct client_arg *ca;
+	struct trader_arg *ta;
+	pthread_t *client_ids;
+	pthread_t *trader_ids;
+	struct order_que *order_que;
+	struct market *market;
+	int i;
+	int done;
+	int err;
+	double start;
+	double end;
+	
+
+	/*
+	 * defaults
+	 */
+	client_threads = 1;
+	trader_threads = 1;
+	orders_per_client = 1;
+	verbose = 0;
+	que_size = 1;
+	max_stock = 1;
+
+	while((c = getopt(argc,argv,ARGS)) != EOF) {
+		switch(c) {
+			case 'c':
+				client_threads = atoi(optarg);
+				break;
+			case 't':
+				trader_threads = atoi(optarg);
+				break;
+			case 'o':
+				orders_per_client = atoi(optarg);
+				break;
+			case 'q':
+				que_size = atoi(optarg);
+				break;
+			case 's':
+				max_stock = atoi(optarg);
+				break;
+			case 'V':
+				verbose = 1;
+				break;
+			default:
+				fprintf(stderr,
+					"unrecognized command %c\n",
+						(char)c);
+				fprintf(stderr,"usage: %s",Usage);
+				exit(1);
+		}
+	}
+
+	client_ids = (pthread_t *)malloc(client_threads*sizeof(pthread_t));
+	if(client_ids == NULL) {
+		exit(1);
+	}
+
+	ca = (struct client_arg *)malloc(client_threads*sizeof(struct client_arg));
+	if(ca == NULL) {
+		exit(1);
+	}
+
+	trader_ids = (pthread_t *)malloc(trader_threads*sizeof(pthread_t));
+	if(trader_ids == NULL) {
+		exit(1);
+	}
+
+	ta = (struct trader_arg *)malloc(trader_threads*sizeof(struct trader_arg));
+	if(ta == NULL) {
+		exit(1);
+	}
+
+	order_que = InitOrderQue(que_size);
+	if(order_que == NULL) {
+		exit(1);
+	}
+
+	market = InitMarket(max_stock,INIT_COUNT);
+	if(market == NULL) {
+		exit(1);
+	}
+
+	start = CTimer();
+	for(i=0; i < client_threads; i++) {
+		ca[i].id = i;
+		ca[i].order_count = orders_per_client;
+		ca[i].max_stock_id = max_stock;
+		ca[i].max_quantity = INIT_COUNT;
+		ca[i].order_que = order_que;
+		ca[i].verbose = verbose;
+		err = pthread_create(&client_ids[i],NULL,
+					ClientThread,(void *)&ca[i]);
+		if(err != 0) {
+			fprintf(stderr,"client thread create %d failed\n",i);
+			exit(1);
+		}
+	}
+
+	done = 0;
+	for(i=0; i < trader_threads; i++) {
+		ta[i].id = i;
+		ta[i].order_que = order_que;
+		ta[i].market = market;
+		ta[i].done = &done;
+		ta[i].verbose = verbose;
+		err = pthread_create(&trader_ids[i],NULL,
+			TraderThread,(void *)&ta[i]);
+		if(err != 0) {
+			fprintf(stderr,"trader thread create %d failed\n",i);
+			exit(1);
+		}
+	}
+
+	/*
+	 * wait for the clients to finish
+	 */
+	for(i=0; i < client_threads; i++) {
+		err = pthread_join(client_ids[i],NULL);
+		if(err != 0) {
+			fprintf(stderr,"client join %d failed\n",i);
+			exit(1);
+		}
+	}
+
+	/*
+	 * tell the traders we are done
+	 */
+	pthread_mutex_lock(&order_que->lock);
+	done = 1;
+	pthread_cond_signal(&order_que->empty);
+	pthread_mutex_unlock(&order_que->lock);
+
+	for(i=0; i < trader_threads; i++) {
+		err = pthread_join(trader_ids[i],NULL);
+		if(err != 0) {
+			fprintf(stderr,"trader join %d failed\n",i);
+			exit(1);
+		}
+	}
+	end = CTimer();
+
+	if(verbose == 1) {
+		PrintMarket(market);
+	}
+
+	printf("%f transactions / sec\n",
+		(double)(orders_per_client*client_threads) / (end-start));
+
+	free(ca);
+	free(ta);
+	free(client_ids);
+	free(trader_ids);
+	FreeMarket(market);
+	FreeOrderQue(order_que);
+
+	return(0);
+}
+		
+
+
+		
+		
+		
+
+	
+	
+
+
+
+	
+	
+