Zipの読み書きメモ

はじめに

JavaにおけるZipファイルの読み書きのメモです。
ソースやテストに使用しているzipファイルは、
https://github.com/xaatw0/quiita/tree/master/src/ziptest
にあります。

Streamの読み込み

zip内のファイルを読み込んで、一つのファイルに連結する。

zip関連.java

public void 連結_zip読み込み(InputStream zipFile, OutputStream output ) throws IOException{

    try(ZipInputStream zipIn = new ZipInputStream(zipFile,Charset.forName("SJIS"))
        ;BufferedOutputStream out = new BufferedOutputStream(output))
    {
        byte[] buffer = new byte[1024];

        ZipEntry entry = null;
        while(null != (entry = zipIn.getNextEntry())){

                int size = 0;
                while(0 < (size = zipIn.read(buffer))){
                    out.write(buffer, 0, size);
                }
            zipIn.closeEntry();
        }
    }
}

/**
 * zip関連.zipには3つのファイルが入っている。エンコードはShiftJIS<br/>
 * 1.txt:あいうえお<br/>
 * 2.txt:かきくけこ(改行)<br/>
 * 3.txt:さしすせそ<br/>
 * 幸いzip内のファイルが名前の順番どおり呼ばれるのでちゃんと動くが、
 * 基本的には動作が保障されてないと思う。
 */
@Test
public void 連結_zip読み込みTest() throws IOException{

    ByteArrayOutputStream output = new ByteArrayOutputStream();
    連結_zip読み込み(getClass().getResourceAsStream("連結_zip読み込みTest.zip"), output);
    assertThat( new String(output.toByteArray(),"SJIS"), is("あいうえおかきくけこ\r\nさしすせそ"));
}

zipの中身をReaderに変換して処理する

zip内のファイルに、日々の合計が「果物名:数」の形式で入っている。zip内のすべてのファイルを読み取り、果物ごとの合計を算出する。

zip関連.java

public Map<String, Integer> zip読み込みして処理(InputStream inputStream) throws IOException{

    Map<String, Integer> result = new HashMap<>();

    try (ZipInputStream zipIn = new ZipInputStream(inputStream,Charset.forName("SJIS"))) {

        // zip内のファイルがなくなるまで読み続ける
        while (null != zipIn.getNextEntry()) {

            String line;
            BufferedReader reader = new BufferedReader(new InputStreamReader(zipIn, "Shift_JIS"));
            while(null != (line = reader.readLine())){

                String[] data = line.split(":");

                String fruit = data[0];
                int value = Integer.parseInt(data[1]);

                // すでに果物があれば加える。なければ、そのまま代入する。
                result.computeIfPresent(fruit, (k,v) -> v + value);
                result.putIfAbsent(fruit, value);
            }
            zipIn.closeEntry();
        }
    }

    return result;
}

/**
 * zip関連.zipには3つのファイルが入っている。エンコードはShiftJIS<br/>
 * 1.txt:りんご:3 みかん:1 バナナ:4<br/>
 * 2.txt:りんご:4 みかん:2<br/>
 * 3.txt:りんご:5 みかん:3 バナナ:1<br/>
 * 「りんご:3」で一行。
 */
@Test
public void zip読み込みして処理Test() throws IOException{

    Map<String, Integer> target =
            zip読み込みして処理(getClass().getResourceAsStream("zip読み込みして処理Test.zip"));

    // 果物は3種類
    assertThat(target.size(), is(3));

    assertThat(target.get("りんご"), is(12));
    assertThat(target.get("みかん"), is(6));
    assertThat(target.get("バナナ"), is(5));
}

zipの書き出し

指定されたInputStreamをzipファイルに買い込む。ファイル名は「順番.txt」にする。

zip関連.java

public void zip書き込み(OutputStream out, InputStream... input) throws IOException{

    try(ZipOutputStream zos = new ZipOutputStream(out)){
        int count = 0;

        for (InputStream stream: input){
            ZipEntry entry = new ZipEntry((count ++) + ".txt" );
            zos.putNextEntry(entry);

            try (InputStream is = new BufferedInputStream(stream)) {
                byte[] buf = new byte[1024];
                for (int len = 0; 0 < (len = is.read(buf))  ;) {
                    zos.write(buf, 0, len);
                }
            }
        }
    }
}

/**
 * zipを書き込んで、zipを展開して内容を確認する。<br/>
 * ByteArrayOuputStreamで実施しているため、close処理が適当、、、
 */
@Test
public void zip書き込みTest() throws IOException{

    String[] data = {"あいうえお","かきくけこ","さしすせそ"};
    ByteArrayInputStream[] input = new  ByteArrayInputStream[data.length];

    for (int i = 0; i < data.length; i++){
        input[i] =  new ByteArrayInputStream(data[i].getBytes());
    }

    // Zipに書き込む(メモリー上に書き込む)
    ByteArrayOutputStream output = new ByteArrayOutputStream();
    zip書き込み(output, input);

    // 以下、zipを展開して、中身を確認する
    try(ZipInputStream zipIn = new ZipInputStream(new ByteArrayInputStream(output.toByteArray())))
    {
        byte[] buffer = new byte[1024];

        for (int i = 0; i < data.length ; i++){
            zipIn.getNextEntry();

            int size = 0;
            while(0 < (size = zipIn.read(buffer))){

                String zipString = new String(Arrays.copyOf(buffer, size));
                assertThat(zipString, is(data[i]));
            }
            zipIn.closeEntry();
        }
    }
}