關於使用 Java 分片讀\寫文件

分片讀取文件方法：

    /**
     * 分片讀取文件塊
     *
     * @param path      文件路徑
     * @param position  角標
     * @param blockSize 文件塊大小
     * @return 文件塊內容
     */
    public static byte[] read(String path, long position, int blockSize) throws Exception {
        // ----- 校驗文件，當文件不存在時，拋出文件不存在異常
        checkFilePath(path, false);
        // ----- 讀取文件
        ByteBuffer block = ByteBuffer.allocate(blockSize);
        try (AsynchronousFileChannel channel = AsynchronousFileChannel.open(Paths.get(path), StandardOpenOption.READ)) {
            Future<Integer> read = channel.read(block, position);
            while (!read.isDone()) {
                // ----- 睡1毫秒， 不搶占資源
                Thread.sleep(1L);
            }
        }
        return block.array();
    }

 

分片寫文件方法：

    /**
     * 分片寫文件
     *
     * @param path     文件目標位置
     * @param block    文件塊內容
     * @param position 角標
     * @throws Exception
     */
    public static void write(String path, byte[] block, long position) throws Exception {
        // ----- 校驗文件，當文件不存在時，創建新文件
        checkFilePath(path, true);
        // ----- 寫文件
        ByteBuffer buffer = ByteBuffer.wrap(block);
        try (AsynchronousFileChannel channel = AsynchronousFileChannel.open(Paths.get(path), StandardOpenOption.WRITE)) {
            Future<Integer> write = channel.write(buffer, position);
            while (!write.isDone()) {
                // ----- 睡1毫秒， 不搶占資源
                Thread.sleep(1L);
            }
        }
    }

 

檢查文件是否存在方法：

    /**
     * 校驗文件
     *
     * @param path 文件路徑
     * @param flag 當文件不存在時是否創建文件 [true: 創建文件；false: 拋出文件不存在異常]
     * @return
     * @throws Exception
     */
    public static File checkFilePath(String path, boolean flag) throws Exception {
        if (StringUtils.isBlank(path)) {
            throw new RuntimeException("The file path cannot be empty.");
        }
        File file = new File(path);
        if (!file.exists()) {
            if (flag) {
                // ----- 當文件不存在時，創建新文件
                if (!file.createNewFile()) {
                    throw new RuntimeException("Failed to create file.");
                }
            } else {
                // ----- 拋出文件不存在異常
                throw new RuntimeException("File does not exist.");
            }
        }
        return file;
    }

 

測試：

    /*** 分片讀寫文件的每片默認大小： 10M */
    private static final Integer DEFAULT_BLOCK_SIZE = 1024 * 1024 * 10;

    public static void main(String[] args) throws Exception {
        String path = "F:\\compression\\Spring源碼深度解析.pdf";
        String toPath = "F:\\compression\\" + System.currentTimeMillis() + ".pdf";
        File file = FileUtil.checkFilePath(path, false);
        long position = 0, length = file.length();
        while (length > DEFAULT_BLOCK_SIZE) {
            // ----- 如果文件大小大於默認分塊大小，循環讀寫文件，每次循環讀取一塊，直到剩余文件大小小於默認分塊大小
            byte[] block = FileUtil.read(path, position, DEFAULT_BLOCK_SIZE);
            FileUtil.write(toPath, block, position);
            position += DEFAULT_BLOCK_SIZE;
            length -= DEFAULT_BLOCK_SIZE;
        }
        if (length > 0) {
            // ----- 如果文件大小小於默認分塊大小且大於零，正常讀寫文件
            byte[] block = FileUtil.read(path, position, (int) length);
            FileUtil.write(toPath, block, position);
        }
    }