com.star.io.file.filevisitor.CopyDirVisitor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of utils Show documentation

some utility class for java develop

The newest version!

package com.star.io.file.filevisitor;

import com.star.io.file.PathUtil;

import java.io.IOException;
import java.nio.file.FileVisitResult;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.SimpleFileVisitor;
import java.nio.file.StandardCopyOption;
import java.nio.file.attribute.BasicFileAttributes;

/**
 * 拷贝文件
 *
 * @author starhq
 */
public class CopyDirVisitor extends SimpleFileVisitor {

    /**
     * 目标
     */
    private final Path src;
    /**
     * 目的地
     */
    private final Path dest;
    /**
     * 拷贝选项
     */
    private StandardCopyOption copyOption;

    /**
     * 构造方法
     *
     * @param src        目标
     * @param dest       目的地
     * @param copyOption 拷贝选项
     */
    public CopyDirVisitor(final Path src, final Path dest, final StandardCopyOption copyOption) {
        super();
        this.src = src;
        this.dest = dest;
        this.copyOption = copyOption;
    }

    /**
     * 构造方法
     *
     * @param src  目标
     * @param dest 目的地
     */
    public CopyDirVisitor(final Path src, final Path dest) {
        this(src, dest, StandardCopyOption.REPLACE_EXISTING);
    }

    /**
     * 目标文件夹不存在则创建
     *
     * @param dir   目录
     * @param attrs 文件属性
     * @return 是否继续访问
     */
    @Override
    public FileVisitResult preVisitDirectory(final Path dir, final BasicFileAttributes attrs) {

        final Path targetPath = dest.resolve(src.relativize(dir));
        PathUtil.mkDirs(targetPath);
        return FileVisitResult.CONTINUE;
    }

    /**
     * 拷贝文件,默认是存在就替换
     *
     * @param file  文件
     * @param attrs 文件属性
     * @return 是否继续访问
     */
    @Override
    public FileVisitResult visitFile(final Path file, final BasicFileAttributes attrs) throws IOException {

        Files.copy(file, dest.resolve(src.relativize(file)), copyOption);
        return FileVisitResult.CONTINUE;
    }

}