Java で ZipOutputStream を byte配列へ変換したい

Question

zip ファイルの中のファイルの内容を少し変更して、変更した zip ファイルを BLOB 型のカラムに保存しようと考えています。

zip ファイルの読み込みには、ZipInputStream、ZipEntry を使っています。(これは問題ありません。)
変更用には ByteArrayOutputStream を使用しているのですが、処理の最後に ZipOutputStream を byte[] にする方法が分かりません。
※byte[] を BLOB のカラムに登録したいので。

ByteArrayOutputStream を使って、stream.read なんて感じで変換しようと試みましたが、ByteArrayOutputStream には read メソッドはありませんでした。
そのため、ByteArrayOutputStream を toByteArray することもできませんでした。

よろしくお願いいたします。

Comments

[xebme]

参考： https://stackoverflow.com/questions/357851/in-java-how-to-zip-file-from-byte-array

[tsukari]

早速コメントいただきありがとうございます。参考に教えていただいたサイトを見本に実装してみます。
結果は改めてご連絡いたします。

[tsukari]

public static byte[] zipBytes(String filename, byte[] input) throws IOException {
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ZipOutputStream zos = new ZipOutputStream(baos);
ZipEntry entry = new ZipEntry(filename);
entry.setSize(input.length);
zos.putNextEntry(entry);
zos.write(input);
zos.closeEntry();
zos.close();
return baos.toByteArray();
}

が参考になりました。

[tsukari]

xebmeさんにベストアンサーを付けるのはどうすれば良いのでしょうか？

Answer 1

参考にさせていただいた内容をもとに、以下のように実装しました。
※パラメータの inputStream は MultipartFile から作成したもので、zip ファイルです。

バイト配列の取得位置が try-with-resources ではまだ ZipOutputStream を close していない場所だったので、ret = baos.toByteArray(); の前に、明示的に zipOutStream.close(); を入れました。

これで、7-zipの場合の解凍時のペイロードのメッセージも無くなり、また、Windowsエクスプローラでのダブルクリックでも解凍できるようになりました。

public byte[] sampleZip(InputStream inputStream) {

        byte[] ret = new byte[1048476];

        ZipEntry zipEntry;
        ZipEntry zipOutputEntry;

        try (ZipInputStream zipInStream = new ZipInputStream(new BufferedInputStream(inputStream),
                Charset.forName("SJIS"));
                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                ZipOutputStream zipOutStream = new ZipOutputStream(baos);) {
            while ((zipEntry = zipInStream.getNextEntry()) != null) {
                if (zipEntry.isDirectory()) {
                    continue;
                }
                String entryName = FilenameUtils.getName(zipEntry.getName()); // FilenameUtils：独自のクラス
                String fileName = FilenameUtils.getBaseName(entryName);
                String fileExt = FilenameUtils.getExtension(entryName);
                SCOPE: {
                    if (StringUtils.isEmpty(fileName)
                            || fileName.length() < 5
                            || StringUtils.isEmpty(fileExt)) {
                        break SCOPE;
                    }
                    byte[] fileBin = IOUtils.toByteArray(zipInStream);
                    if (fileBin.length <= 0) {
                        break SCOPE;
                    }
                    if (!(fileExt.equals("xls") || fileExt.equals("xlsx"))) {
                        zipOutputEntry = new ZipEntry(entryName);
                        zipOutputEntry.setSize(fileBin.length);
                        zipOutStream.putNextEntry(zipOutputEntry);
                        zipOutStream.write(fileBin);
                        break SCOPE;
                    }

                    XSSFWorkbook workbook = new XSSFWorkbook();

                    workbook = addSheet(new ByteArrayInputStream(fileBin)); // addSheet：独自の関数
                    fileBin = getBytes(workbook);
                    if (workbook != null) {
                        workbook.close();
                        zipOutputEntry = new ZipEntry(entryName);
                        zipOutputEntry.setSize(fileBin.length);
                        zipOutStream.putNextEntry(zipOutputEntry);
                        zipOutStream.write(fileBin);
                    }
                }
                zipInStream.closeEntry();
                zipOutStream.closeEntry();
            }
            zipOutStream.close();  //追加 2020.03.05
            ret = baos.toByteArray();
        } catch (Exception e) {
            ret = null;
        }

        return ret;
    }

Comments

[tsukari]

解決した方法の baos.toByteArray() にて zip ファイルのバイト配列はできましたが、そのバイト配列から作成したファイルは 7-zip 等でしか解凍できませんでした。(※Windowsエクスプローラーでのダブルクリックでは解凍不可)
やはり一度ファイルに出力し、Files.readAllBytes でバイト配列にしなければならないかもしれません。

[tsukari]

実ファイルへの出力は回避できました。