C面向對象編程示例——線程調度

/*
* 線程封裝演示-C版本mythr.h
* write by panzhh
* 2011.12.24
*/
#ifndef MYTHR_H
#define MYTHR_H
#include <pthread.h>
typedef unsigned int bool;
#define true 1
#define false 0
typedef void (*RUN)(void *arg);
typedef struct __mythr
{
RUN run; /*線程工作函數*/
void *arg; /*傳給run的參數*/
bool bthrun; /*循環標志*/
bool bthrSUSEnd;/*掛起標志*/

/*private item*/
pthread_t tid;
pthread_mutex_t mutex;
pthread_cond_t cont;
}MYTHR, *PMYTHR;
int thr_start(PMYTHR pthr); /*開啟線程*/
int thr_wait(PMYTHR pthr); /*等待線程*/
int thr_cancel(PMYTHR pthr);/*取消線程*/
void thr_exit(PMYTHR pthr); /*退出線程*/
void thr_suspend(PMYTHR pthr); /*掛起線程*/
void thr_resume(PMYTHR pthr); /*恢復線程*/
#endif
/*
* 線程封裝演示-C版本mythr.c
* write by panzhh
* 2011.12.24
*/
#include <stdio.h>
#include <stdlib.h>
#include "mythr.h"
static void *routine(void *arg)
{
PMYTHR pthr=(PMYTHR)arg;
pthread_mutex_init(&pthr->mutex, NULL);
pthread_cond_init (&pthr->cont, NULL);
pthread_setcancelstate(PTHREAD_CANCEL_ENABLE, NULL);
pthread_setcanceltype(PTHREAD_CANCEL_ASYNCHRONOUS, NULL);
do{
pthread_mutex_lock(&pthr->mutex);
if(pthr->bthrsusend){
if(0 != pthread_cond_wait (&pthr->cont, &pthr->mutex)){
perror("work:pthread_cond_wait");
return (void*)-1;
}
}
pthread_mutex_unlock(&pthr->mutex);
pthr->run(pthr->arg);
}while(pthr->bthrun);
pthread_cond_destroy(&pthr->cont);
pthread_mutex_destroy(&pthr->mutex);
return (void *)0;
}

int thr_start(PMYTHR pthr)
{
if(NULL != pthr->run){
return pthread_create(&pthr->tid, NULL, routine, (void *)pthr);
}

return -1;
}

int thr_wait(PMYTHR pthr)
{
if(NULL != pthr->run){
return pthread_join(pthr->tid, NULL);
}
return -1;
}

int thr_cancel(PMYTHR pthr)
{
if(NULL != pthr->run){
pthread_cancel(pthr->tid);
pthread_cond_destroy(&pthr->cont);
pthread_mutex_destroy(&pthr->mutex);
return 0;
}
return -1;
}

void thr_exit(PMYTHR pthr)
{
if(NULL != pthr->run){
pthread_mutex_lock(&pthr->mutex);
if(pthr->bthrsusend){
pthr->bthrsusend = false;
pthread_cond_signal(&pthr->cont);
}
pthr->bthrun = false;
pthread_mutex_unlock(&pthr->mutex);
}
}

void thr_suspend(PMYTHR pthr)
{
pthread_mutex_lock(&pthr->mutex);
if(!pthr->bthrsusend){
pthr->bthrsusend = true;
}
pthread_mutex_unlock(&pthr->mutex);
}

void thr_resume(PMYTHR pthr)
{
pthread_mutex_lock(&pthr->mutex);
if(pthr->bthrsusend){
pthr->bthrsusend = false;
pthread_cond_signal(&pthr->cont);
}
pthread_mutex_unlock (&pthr->mutex);
}
/*
* 線程封裝演示-C版本-測試代碼
* write by panzhh
* 2011.12.24
*/
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "mythr.h"

unsigned int k = 0;
void thrfunc(void *arg)
{
printf("Thr---%s---%u\r\n", (char*)arg, k++);
sleep(1);
}
MYTHR thr = {
.run=thrfunc, /*線程工作函數*/
.arg="test", /*傳給run的參數*/
.bthrun=true, /*循環標志*/
.bthrsusend=true, /*掛起標志*/
};

int main(int argc, char **argv)
{
thr_start(&thr);

//調度線程
while(1){
printf("Please Input:\r\n");
char ch = getchar();

switch(ch){
/*suspend*/
case 's':
case 'S':
thr_suspend(&thr);
break;
/*resume*/
case 'r':
case 'R':
thr_resume(&thr);
break;
/*cancel*/
case 'c':
case 'C':
thr_cancel(&thr);
goto EXIT_STEP;
/*exit*/
case 'e':
case 'E':
thr_exit(&thr);
goto EXIT_STEP;
/*exit*/
case 'q':
case 'Q':
exit(1);
};
}
EXIT_STEP:
thr_wait(&thr);
printf("---done---\r\n");
return 0;
}