数据流的解压缩

比眉伴天荒 2022-12-23 09:35 116阅读 0赞

# 数据流的解压缩 #

package com.hadoop.compress;
    
    import org.apache.hadoop.conf.Configuration;
    import org.apache.hadoop.fs.Path;
    import org.apache.hadoop.io.IOUtils;
    import org.apache.hadoop.io.compress.*;
    import org.apache.hadoop.util.ReflectionUtils;
    import org.junit.Test;
    
    import java.io.FileInputStream;
    import java.io.FileNotFoundException;
    import java.io.FileOutputStream;
    
    public class CompressDemo { 
    
        /* 压缩 */
        @Test
        public void test() throws Exception { 
            //读数据 - 普通的文件流
            FileInputStream fis =
                    new FileInputStream("D:\\io\\compress\\aaa.txt");
            //写数据 - 压缩流
            CompressionCodec bZip2Codec =
                    ReflectionUtils.newInstance(BZip2Codec.class, new Configuration());
            //getDefaultExtension() : 获取该压缩类型的扩展名
            CompressionOutputStream os = bZip2Codec.createOutputStream(
                    new FileOutputStream("D:\\io\\compress\\aaa.txt"
                            + bZip2Codec.getDefaultExtension()));
            //一边读一边写（流的对拷）
            IOUtils.copyBytes(fis,os,1024,true);
        }
    
        /* 解压缩 */
        @Test
        public void test2() throws Exception { 
            //读数据 - 压缩流
            CompressionCodec bZip2Codec =
                    ReflectionUtils.newInstance(BZip2Codec.class, new Configuration());
            CompressionInputStream is = bZip2Codec.createInputStream(new FileInputStream(
                    "D:\\io\\compress\\aaa" + bZip2Codec.getDefaultExtension()));
            //写数据 - 普通的文件流
            FileOutputStream fos = new FileOutputStream("D:\\io\\compress\\aaa.txt");
            //流的对拷
            IOUtils.copyBytes(is,fos,1024,true);
        }
    
        /* 解压缩 */
        @Test
        public void test3() throws Exception { 
            //创建编解码的工厂类对象 - 不用考虑创建哪种压缩格式的对象
            CompressionCodecFactory factory = new CompressionCodecFactory(new Configuration());
            //判断文件是否可以解压缩 - 如果返回值为null则表示不可解压缩
            //如果可以解压缩那么返回可以解压该文件压缩类型的对象
            CompressionCodec codec = factory.getCodec(new Path("D:\\io\\compress\\aaa.txt.bz2"));
            if (codec == null){ 
                System.out.println("该文件不可解压缩");
                return;
            }
            //读数据 - 压缩流
            CompressionInputStream is = codec.createInputStream(new FileInputStream(
                    "D:\\io\\compress\\aaa" + codec.getDefaultExtension()));
            //写数据 - 普通的文件流
            FileOutputStream fos = new FileOutputStream("D:\\io\\compress\\aaa.txt");
            //流的对拷
            IOUtils.copyBytes(is,fos,1024,true);
        }
    }