Java的IO操作（四） - SequenceInputStream类，实例：一个文件分割、合并工具

SequenceInputStream可以看做是多个InputStream对象的有序集合。当一个InputStream对象的数据读取完后，它会自动取出下一个InputStream对象进行读取，直到所有的InputStream对象都读取完为止。

利用这个特点，我们来编写一个文件分割、合并工具。

使用说明：

SequenceDemo [Option] [filePath] [number]

Option：

-c : 表示要合并文件

-s : 表示要分割文件

filePath：

指定一个文件路径

number：

指定单个文件大小或文件数量

如：

java cls.SequeceDemo -s F:\QQ.exe 102400

表示将QQ.exe分割成多个文件大小为102400字节的小文件。

java cls.SequeceDemo -m F:\QQ.exe 13

表示将已经被分割好的13个QQ.exe文件合并成一个整文件。

注意，构造SequenceOutputStream对象时需要一个Enumeration对象的引用做为参数。Enumeration是一个接口，我们必须实现这个接口中的hasMoreElements()方法和nextElement()方法。

完整代码：

package cls;

import java.io.*;
import java.util.*;

public class SequenceStreamDemo
{
    public static void main(String[] args) throws Exception
    {
        switch(args[0].charAt(1))
        {
            case 's': // 分割文件
            {
                SequenceStreamDemo.seperate(args[1],Long.parseLong(args[2]));
                break;
            }
            case 'm': // 合并文件
            {
                SequenceStreamDemo.merge(args[1],Integer.parseInt(args[2]));
                break;
            }
            default: // 参数输入错误
            {
                System.out.println("非法参数");
                break;
            }
        }
    }
    
    /* 合并文件
    * fileName : 文件名
    * amount : 要合并的小文件的数量
    **/
    public static void merge(String fileName,int amount) throws Exception
    {
        // 创建合并后的文件
        File file = new File(fileName);
        FileOutputStream fis = new FileOutputStream(file);
        BufferedOutputStream bos = new BufferedOutputStream(fis);
        
        // 将多个小文件对象保存在List中
        List<InputStream> list = new ArrayList<InputStream>();
        for(int i = 0 ; i < amount ; ++i)
        {
            File f = new File(fileName + "_" + i);
            list.add(new FileInputStream(f));
        }
        
        final Iterator<InputStream> it = list.iterator();
        
        // 构建Enumeration对象
        Enumeration<InputStream> em = new Enumeration<InputStream>()
        {
            public boolean hasMoreElements()
            {
                return it.hasNext();
            }
            public InputStream nextElement()
            {
                return it.next();
            }
        };
        
        // 使用SequenceInputStream合并文件
        BufferedInputStream bis = new BufferedInputStream(new SequenceInputStream(em),8192);
        int data;
        
        while((data = bis.read()) != -1)
        {
            bos.write(data);
        }
        
        bis.close();
        bos.flush();
        bos.close();
        
        System.out.println("合并完成");
    }
    /* 分割文件
    * fileName : 文件名
    * size : 单个文件的大小
    **/
    public static void seperate(String fileName,long size) throws Exception
    {
        File file = new File(fileName);
        long number;
        
        // 计算分割后的文件数量
        if(0 == file.length() % size)
        {
            number = file.length() / size;
        }
        else
        {
            number = file.length() / size + 1;
        }
        
        FileInputStream fis = new FileInputStream(file);
        BufferedInputStream bis = new BufferedInputStream(fis);
        
        // 开始分割
        byte[] buf = new byte[1];
        for(int i = 0 ; i < number ; ++i)
        {
            FileOutputStream fos = new FileOutputStream(new File(fileName + "_" + i));
            BufferedOutputStream bos = new BufferedOutputStream(fos);
            
            long curSize = 0L;
            while(bis.read(buf) != -1)
            {
                bos.write(buf);
                
                curSize++;
                if(curSize == size) // 已经分割成了一个小文件
                {
                    bos.flush();
                    bos.close();
                    break;
                }
            }
            if(curSize < size) // 打包剩下的数据
            {
                bos.flush();
                bos.close();
                break;
            }
            
        }
        
        System.out.println("分割完成");
    }
}