多线程——使用两个线程完成两个文件的拷贝

 1、使用两个线程完成两个文件的拷贝

分支线程1拷贝前一半，分支线程2拷贝后一半，主线程回收两个分支线程的资源

//使用两个线程完成两个文件的拷贝，分支线程1拷贝前一半，分支线程2拷贝后一半
//主线程回收两个分支线程的资源
#include<myhead.h>

//定义求文件长度函数
int get_len(const char *scrfile,const char *desfile)
{
    //以只读的形式打开源文件
    int sfd=open(scrfile,O_RDONLY);
    if(sfd==-1)
    {
        perror("open file1 error");
        return -1;
    }
    //以创造的形式打开目标文件
    int dfd=open(desfile,O_WRONLY | O_CREAT | O_TRUNC,0664);
    if(dfd==-1)
    {
        perror("open file1 error");
        return -1;
    }

    //光标偏移到末尾,返回的值就是文件大小
    int size=lseek(sfd,0,SEEK_END);

    //关闭文件
    close(sfd);
    close(dfd);

    return size;
}

//定义拷贝函数
int copy_file(const char *scrfile,const char *desfile,int start,int len)
{
    //以只读的形式打开源文件
    int sfd=open(scrfile,O_RDONLY);
    if(sfd==-1)
    {
        perror("open file1 error");
        return -1;
    }
    //以只写的形式打开目标文件
    int dfd=open(desfile,O_WRONLY);     //为可以分两个拷贝,不能用清空指令
    if(dfd==-1)
    {
        perror("open file1 error");
        return -1;
    }

    //移动光标位置
    lseek(sfd,start,SEEK_SET);
    lseek(dfd,start,SEEK_SET);

    //定义搬运工
    char buf[128];
    int sun=0;      //记录拷贝的总个数

    while(1)
    {
        //读取源文件数据
        int res=read(sfd,buf,sizeof(buf));
        sun+=res;
        if(res == 0 || sun>len)             //表示文件读取结束
        {
            write(dfd, buf, res-(sun-len));    //父进程将最后一次拷贝结束
            break;
        }
        //写入到目标文件中
        write(dfd, buf, res);
    }
    //关闭文件
    close(sfd);
    close(dfd);

    printf("拷贝成功\n");
    return 0;
}

struct Stu
{
    char const *scrfile;
    char const *desfile;
    int start;
    int len;
};


void*task(void *arg)
{
    char const *scrfile=(*((struct Stu*)arg)).scrfile;
    char const *desfile=(*((struct Stu*)arg)).desfile;
    int start = (*((struct Stu*)arg)).start;
    int len=(*((struct Stu*)arg)).len;
    if(start==0)
    {
        copy_file(scrfile,desfile,start,len);
        //退出线程
        pthread_exit(NULL);
    }else
    {
        copy_file(scrfile,desfile,start,len);
        //退出线程
        pthread_exit(NULL);
    }
    
}



/******************主函数*************************/
int main(int argc, char const *argv[])
{
    if(argc!=3)
    {
        printf("input file error\n");
        return -1;
    }

    //求出源文件长度
    int len=get_len(argv[1],argv[2]);

    struct Stu su1={argv[1],argv[2],0,len/2};
    struct Stu su2={argv[1],argv[2],len/2,len-len/2};

    //定义线程变量
    pthread_t tid1, tid2;
    //创建两个线程
    if(pthread_create(&tid1, NULL, task, &su1) != 0)
    {
        printf("tid1 创建失败\n");
        return -1;
    }

    if(pthread_create(&tid2, NULL,task,&su2) != 0)
    {
        printf("tid2 创建失败\n");
        return -1;
    }

    //阻塞等等线程的结束
    pthread_join(tid1, NULL);
    pthread_join(tid2, NULL);


    return 0;
}

源文件 one.txt 中的内容：

程序运行：

运行后目标文件 two.txt 中的内容：