2024/02/21

使用多线程完成两个文件的拷贝，第一个线程拷贝前一半，第二个线程拷贝后一半，主线程回收两个线程的资源

#include<myhead.h>
void *task1(void *arg);//支线程1
void *task2(void *arg);//支线程2
int length(const char *r);//求文档长度函数
int fcopy(const char *r,const char *w,int start,int len);

//定义传参结构体
struct INFO
{
	const char *r; 	//读取的文件
	const char *w; 	//写入的文件
	int start; 	//开始位置
	int len; 	//长度
};
int main(int argc, const char *argv[])
{
	//判断是否进行了外部传参
	if(argc !=3)
	{
		printf("input file error\n");
		printf("usage:./a.out srcfile destfile\n");
		return -1;
	}
	int len=length(argv[1]);
	//定义结构体变量
	struct INFO t1={argv[1],argv[2],0,len/2};
	struct INFO t2={argv[1],argv[2],len/2,len-len/2};
	//定义线程号变量
	pthread_t tid1=0;
	//创建两个分支线程
	if(pthread_create(&tid1,NULL,task1,&t1) !=0)
	{
		printf("tid1 create error\n");
		return -1;
	}

	pthread_t tid2=0;
	if(pthread_create(&tid2,NULL,task2,&t2) !=0)
	{
		printf("tid2 create error\n");
		return -1;
	}

	//主线程
	printf("主线程\n");
	printf("tid1 = %#lx,tid2=%#lx\n",tid1,tid2);


	//回收线程资源
	pthread_join(tid1,NULL);
	pthread_join(tid2,NULL);
	printf("主线程成功回收分支线程资源\n");

	return 0;
}

//文档长度函数
int length(const char *r)
{
	//只读形式打开源文件
	FILE *rfp=NULL;
	if((rfp=fopen(r,"r"))==NULL)
	{
		perror("fopen rfp error");
		return -1;
	}
	//使用fseek函数和ftell函数定位到文件的数量
	//fseek(文件指针，偏移量，开始的位置)
	//ftell(文件中指针)
	int len=0;
	fseek(rfp,0,SEEK_END); 	//光标偏移到文件的最后位置
	len=ftell(rfp); 	//返回文件的数量
	fclose(rfp);
	return len;	
}

//拷贝函数
int fcopy(const char *r,const char *w,int start,int len)
{
	char buf[128]="";
	//只读形式打开源文件
	FILE *rfp=NULL;
	if((rfp=fopen(r,"r"))==NULL)
	{
		perror("fopen rfp error");
		return -1;
	}
	//追加形式打开写入文件
	FILE *wfp=NULL;
	if((wfp=fopen(w,"a"))==NULL)
	{
		perror("fopen wfp error");
		return -1;
	}
	bzero(buf,sizeof(buf));
	fseek(rfp,start,SEEK_SET); 	//光标定位到开始偏移位置
	int count=fread(buf,sizeof(buf[0]),len,rfp);
	//写入
	fwrite(buf,sizeof(buf[0]),count,wfp);
	//关闭文件
	fclose(rfp);
	fclose(wfp);
	return 0;
}

//定义分支线程1
void *task1(void *arg)
{
	while(1)
	{
		struct INFO *fap=(struct INFO *)arg;
		printf("分支线程1\n");
		//调用文件拷贝函数
		int a=fcopy(fap->r,fap->w,fap->start,fap->len);
		sleep(1);
		//获取当前进程号
		printf("我是分支线程1:tid1=%#lx\n",pthread_self());
		//退出当前线程
		printf("分支线程1资源退出\n");
		pthread_exit(NULL);
	}
}

//定义分支线程2
void *task2(void *arg)
{
	while(1)
	{
		printf("分支线程2\n");
		sleep(1);
		//调用文件拷贝函数
		struct INFO *fap=(struct INFO *)arg;
		int a=fcopy(fap->r,fap->w,fap->start,fap->len);
	
		//获取当前进程号
		printf("我是分支线程2:tid2=%#lx\n",pthread_self());
		//退出当前线程
		printf("分支线程2资源退出\n");
		pthread_exit(NULL);
	}
}