#include<stdio.h>
#include<string.h>
#include<unistd.h>
#include<stdlib.h>
#include<pthread.h> #define ONE_SECOND 1000000
#define RANGE 10
#define PERIOD 2
#define NUM_THREADS 4 typedef struct
{
int *carpark; //用一个数组来模拟停车场停车位
int capacity; //停车场的车辆容量
int occupied; //停车场现有的车辆数目
int nextin; //下一个进来的车的位置(用carpark数组代表的下标表示)
int nextout; //下一个取走的车的停车位置
int cars_in; //记录停车场进入车辆的总和
int cars_out; //记录从停车场开出去的车辆总和
pthread_mutex_t lock; //互斥量,保护该结构中的数据被线程互斥的方式使用
pthread_cond_t space; //条件变量 描述停车场是否有空位置
pthread_cond_t car; //条件变量 描述停车场是否有车
pthread_barrier_t bar; //线程屏障
}cp_t; static void initialise(cp_t *cp,int size)
{
cp->occupied = cp->nextin = cp->nextout = cp->cars_in = cp->cars_out = ;
cp->capacity = size;
cp->carpark = (int *)malloc(cp->capacity * sizeof(*cp->carpark)); //初始化线程屏障,NUM_THREADS表示等待 NUM_THREADS=4个线程同步执行
pthread_barrier_init(&cp->bar,NULL,NUM_THREADS); if(cp->carpark == NULL)
{
perror("malloc()");
exit();
} srand((unsigned int )getpid()); pthread_mutex_init(&cp->lock,NULL); //初始化停车场的锁
pthread_cond_init(&cp->space,NULL); //初始化描述停车场是否有空位的条件变量
pthread_cond_init(&cp->car,NULL); //初始化描述停车场是否有车的条件变量
} static void* car_in_handler(void* carpark_in)
{
cp_t *temp;
unsigned int seed;
temp = (cp_t *)carpark_in; //pthread_barrier_wait 函数表示,线程已完成工作,等到其他线程赶来
pthread_barrier_wait(&temp->bar);
while()
{
//将线程挂起一段时间,模拟车辆到来的随机性
usleep(rand_r(&seed) % ONE_SECOND);
pthread_mutex_lock(&temp->lock); //循环等待知道有停车位
while(temp->occupied == temp->capacity)
pthread_cond_wait(&temp->space,&temp->lock); //插入一辆车,用随机数标识
temp->carpark[temp->nextin] = rand_r(&seed) % RANGE; temp->occupied++;
temp->nextin++;
temp->nextin %= temp->capacity;
temp->cars_in++; //可能有的人在等车可取,发送temp->car条件变量
pthread_cond_signal(&temp->car);
pthread_mutex_unlock(&temp->lock);
} return ((void*)NULL);
} static void* car_out_handler(void *carpark_out)
{
cp_t *temp;
unsigned int seed;
temp = (cp_t *)carpark_out;
pthread_barrier_wait(&temp->bar);
for(;;)
{
usleep(rand_r(&seed) % ONE_SECOND); pthread_mutex_lock(&temp->lock); /*
获得锁后访问temp->occupied 变量,此时如果车辆数为0 (occupied == 0)
pthread_cond_wait 进行的操作就是忙等,释放锁(&temp->lock)供其他路线使用
知道temp->car条件改变时再次将锁锁住
*/
while(temp->occupied == )
{
pthread_cond_wait(&temp->car,&temp->lock);
} temp->occupied--;
temp->nextout++;
temp->nextout %= temp->capacity;
temp->cars_out++; pthread_cond_signal(&temp->space);
pthread_mutex_unlock(&temp->lock);
} return ((void *)NULL);
} static void *monitor(void *carpark_in)
{
cp_t *temp;
temp = (cp_t *)carpark_in; for(;;)
{
sleep(PERIOD); pthread_mutex_lock(&temp->lock); printf("Delta:%d\n",temp->cars_in - temp->cars_out - temp->occupied);
printf("Number of cars in carpark:%d\n",temp->occupied); pthread_mutex_unlock(&temp->lock);
} return ((void *)NULL);
} int main(int argc,char **argv)
{
printf("main version 1.0\n");
if(argc != )
{
printf("Usage :%s carparksize\n",argv[]);
exit();
} cp_t outpark; initialise(&outpark,atoi(argv[])); //初始化停车场数据结构 pthread_t car_in,car_out,m; //定义线程变量
pthread_t car_in2,car_out2; /***
创建往停车场停车线程(生成者1)
创建从停车场取车线程(消费者1)
创建往停车场停车线程(生成者2)
创建从停车场取车线程(消费者2)
创建用于监控停车场状况的线程
***/
pthread_create(&car_in,NULL,car_in_handler,(void*)&outpark);
pthread_create(&car_out,NULL,car_out_handler,(void*)&outpark);
pthread_create(&car_in2,NULL,car_in_handler,(void*)&outpark);
pthread_create(&car_out2,NULL,car_out_handler,(void*)&outpark);
pthread_create(&m,NULL,monitor,(void*)&outpark); //pthread_join的第二个参数为NULL,表示不关心线程返回状态,仅仅等待指定线程的终止
pthread_join(car_in,NULL);
pthread_join(car_out,NULL);
pthread_join(car_in2,NULL);
pthread_join(car_out2,NULL);
pthread_join(m,NULL); return ;
}
05-21 19:05