使用java类SequenceInputStream 对一个文件进行分割和合并的操作

来源:互联网 发布:歌手阿朵 知乎 编辑:程序博客网 时间:2024/05/09 02:30

package cn.itcast.day_18.sequence;

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;
import java.io.InputStream;
import java.io.SequenceInputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;

//对一个文件进行分割和合并的操作
public class Exercise {

 public static void main(String[] args) throws IOException {
  
//   split(new File("zl3.jpg"));    //分割文件
   merge(new File("zl3.jpg"));    //合并文件
 }
 
 public static void merge(File dir) throws IOException {
  // 创建一个ArrayList<InputStream>
  ArrayList<InputStream> list = new ArrayList<InputStream>();
  
  // 获取dir下所有子文件
  File[] subFiles = dir.listFiles();
  
  // 循环遍历子文件, 每个文件创建一个输入流, 装入ArrayList
  for (File file : subFiles)
   list.add(new FileInputStream(file));
   
  // 循环结束之后, 从ArrayList中获取Enumeration
  Enumeration<InputStream> e = Collections.enumeration(list);
   
  // 创建序列流, 传入Enumeration
  SequenceInputStream sis = new SequenceInputStream(e);
  
  // 从序列流中读取数据, 将数据写出到一个临时文件中
  File file = new File(dir.getAbsoluteFile().getParent(), ".temp");
  FileOutputStream fos = new FileOutputStream(file);
  int len;
  byte[] buffer = new byte[1024];
  while((len = sis.read(buffer)) != -1)
   fos.write(buffer, 0, len);
  fos.close();
  sis.close();
  
  // 删除原文件夹, 需要先删里面的文件
  for (File f : subFiles)
   f.delete();
  dir.delete();
  
  // 将临时文件改为原文件夹的名字
  file.renameTo(dir);
  System.out.println("文件合并成功");
 }
 
 public static void split(File file) throws IOException{
  File dir = new File(".temp");
  dir.mkdir();
  
  // 计算每一份的大小
  long partLen = (file.length() + 4) / 5;  // (10 + 4) / 5 = 2,  (14 + 4) / 5 = 3
  int fileNum = 1;
  long len = 0;
  
  BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
  BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(new File(dir, fileNum + ".jpg")));
  
  int b;
  while((b = bis.read()) != -1) {
   if(len++ == partLen){
    bos.close();
    bos = new BufferedOutputStream(new FileOutputStream(new File(dir, ++fileNum + ".jpg")));
    len = 0;
   }
   bos.write(b);
  }
  bis.close();
  bos.close();
  
  //System.out.println(file.delete());
  file.delete();
  dir.renameTo(file);
  System.out.println("文件分割成功");
 }

}

原创粉丝点击