最近优化了一个程序,使其从单线程处理变为多线程处理
线程为常驻线程
简单写了一个例子
编译:gcc testthread.c -otest -lpthread
注意编译的时候一定要加-lpthread
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <stdbool.h>
#include <unistd.h>//sleep函数需要这个头文件
#define MAX_THREAD_NUM 5//最大线程数 可根据需要增加相关配置
pthread_mutex_t LOCK = PTHREAD_MUTEX_INITIALIZER;//线程间互斥锁
int gv_iCurrectThreadNum = 0;//当前线程数
//gcc testthread.c -otest -lpthread 编译命令,注意一定要加-lpthread
void* thread_function(void *ptr)//ptr是线程启动的时候传的参数
{
pthread_mutex_lock(&LOCK);//加锁
gv_iCurrectThreadNum++;//各线程要修改的公共部分需要加锁
pthread_mutex_unlock(&LOCK);//解锁
//int *iThreadNum = (int*)ptr;这样写也可以
int iThreadNum = *((int*)ptr);//注意传参之前是什么类型,就要转成什么类型的指针
while(true)
{
printf("thread:%d\n", iThreadNum);//正常逻辑处理的地方,这里随便加了打印
sleep(1);
}
pthread_mutex_lock(&LOCK);
gv_iCurrectThreadNum--;
pthread_mutex_unlock(&LOCK);
}
int main()
{
int iTempThreadNum = 0;
int iRet = 0;
pthread_t *pThread;
//申请内存,正常情况是不需要的,但这里我们要创建的常驻线程
pThread = (pthread_t*)malloc(sizeof(pthread_t) * MAX_THREAD_NUM);
for( ; iTempThreadNum < MAX_THREAD_NUM; iTempThreadNum++)
{
//pthread_create第2个参数是线程分离属性的,NULL表示非分享
//最后一个参数是传给线程处理函数的
//不传的时候可以为NULL,也可以传结构体
iRet = pthread_create(&pThread[iTempThreadNum], NULL, &thread_function, (void*)&iTempThreadNum);
sleep(1);
if(iRet != 0)
printf("create thread failed!\n currect thread num is %d\n", gv_iCurrectThreadNum);
}
printf("create thread succeed!\n currect thread num is %d\n", gv_iCurrectThreadNum);
while(true)
{
iTempThreadNum = gv_iCurrectThreadNum;
if(iTempThreadNum < MAX_THREAD_NUM)
printf("\n有线程掉了\n");
if(iTempThreadNum <= 0)//所有线程都掉了
break;
sleep(20);
}
free(pThread);//释放我们自己申请的内存,其实不是很必要,因为进程马上就要结束了,但为了养成良好的习惯,建议malloc和free要成对的写
return 0;
}