Linux C 多线程

最近优化了一个程序，使其从单线程处理变为多线程处理

线程为常驻线程

简单写了一个例子

编译：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;
}

猜你喜欢