Java NIO文件、流操作

概述

数据输入流、输出流间拷贝

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.ByteBuffer;
import java.nio.channels.Channels;
import java.nio.channels.ReadableByteChannel;
import java.nio.channels.WritableByteChannel;

/**
 * @Description : nio工具类
 * @Version : V1.0.0
 * @Date : 2021/8/13 17:03
 */
public class NIOUtil {
    /**
     * 数据从输入流复制到输出流
     * 
     * @param input 输入流
     * @param output 输出流
     * @return 拷贝字节数
     * @throws IOException
     */
    public static long copy(InputStream input, OutputStream output) throws IOException {
        try (ReadableByteChannel src = Channels.newChannel(input);
                WritableByteChannel desc = Channels.newChannel(output)) {
            ByteBuffer buffer = ByteBuffer.allocate(FileSysConstant.DEFAULT_BUFFER_SIZE);
            long count = 0;
            int n;
            while (FileSysConstant.EOF != (n = src.read(buffer))) {
                buffer.flip();
                desc.write(buffer);
                buffer.compact();
                count += n;
            }
            // 可能有阻塞的没有写入通道
            buffer.flip();
            while (buffer.hasRemaining()) {
                n = desc.write(buffer);
                count += n;
            }
            return count;
        }
    }
}