linux C++多線程操作文件&輸出加鎖

下述demo將指定目錄下文件存入vector,然後一分爲二交給兩個線程(threadTask1,threadTask2)去分別處理,對輸出函數printDirent加鎖防止紊亂。

#include <iostream>
#include <string>
#include <sys/types.h>
#include <dirent.h>
#include <unistd.h>
#include <stdio.h>
#include <fstream>
#include <iomanip>
#include <ctime>
#include <stdio.h>
#include <stdlib.h>
#include <pthread.h>
#include <unistd.h>
#include <list>
#include <numeric>
#include <algorithm>
#include <vector>

static pthread_mutex_t testlock;
using namespace std; 
void *threadTask1(void *ptr);
void *threadTask2(void *ptr);
typedef vector<dirent> VECTORDIRENT;
VECTORDIRENT listDirent;
VECTORDIRENT::iterator iter; 
void printStr(string buff);
void printDirent(struct dirent ptrTest);

void *threadTask1(void *ptr)
{
	int vectorI=0;
	int vectorLen = listDirent.size()/2;
	int count=0;
        for(;vectorI<vectorLen;vectorI++)
        {
    	printDirent(listDirent[vectorI]);
    	count++;
	}  
}

void *threadTask2(void *ptr)
{
	int vectorI=listDirent.size()/2;
	int vectorLen = listDirent.size();
	int count=0;
        for(;vectorI<vectorLen;vectorI++)
        {
    	printDirent(listDirent[vectorI]);
    	count++;
	}
}


void printDirent(struct dirent ptrTest){
	pthread_mutex_lock(&testlock);
	cout<<ptrTest.d_name<<endl;
	pthread_mutex_unlock(&testlock);
	
}

int main(void)
{
    DIR    *dirTest;
    struct    dirent    *ptrTest;
    string testFolderPathName = "/home/undoner/";
    dirTest = opendir(testFolderPathName.data()); ///open the dir
    int fileCount = 0;
    while((ptrTest = readdir(dirTest)) != NULL) ///read the list of this dir
    {   
		if(ptrTest->d_type==8){
			listDirent.push_back(*ptrTest);
		}
    }
    cout<<listDirent.size()<< endl; 
    cout<<"listDirent.begin()--- listDirent.end():"<<endl; 
    for (iter = listDirent.begin(); iter != listDirent.end(); ++iter) {
    	cout << iter->d_name << " "; 
    }
    pthread_t thread1,thread2;
	int ret1,ret2;
	ret1=pthread_create(&thread1,NULL,threadTask1,NULL);
	ret2=pthread_create(&thread2,NULL,threadTask2,NULL);
	//ret3=pthread_create(&thread3,NULL,thread,NULL);
	if(ret1!=0){
	printf ("Create pthread error!\n");
	exit (1);
	}
	if(ret2!=0){
	printf ("Create pthread error!\n");
	exit (1);
	}
	pthread_join(thread1,NULL);
	pthread_join(thread2,NULL);	
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章