Golangでzipファイルを解凍する

こんにちは。初投稿です。どうぞお手柔らかにm(__)m。

Go言語に触れ始めて数か月、それほどガッツリ書いてるわけではないですが、少しわかってきた気がします（危険な時期）

さて、タイトルの通りZipファイルを解凍するのですが、よくあるZip解凍ツールのように「ファイル名を展開先のディレクトリ名に使う」のってよくありませんか？ないですかそうですか。

例えば「test20210318.zip」であれば「test20210318」というディレクトリを作って、その下層に展開する、というものです。
実際に業務でニーズがありまして、同じファイル構成でzip圧縮されたファイル群を、繰り返し解凍するような仕組みです。

作ったサンプルがこちら。

sample/unzip/main.go

package main

import (
	"archive/zip"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"regexp"
)

// unZip zipファイルを展開する
func unZip(src, dest string) error {
	r, err := zip.OpenReader(src)
	if err != nil {
		return err
	}
	defer r.Close()

	ext := filepath.Ext(src)
	rep := regexp.MustCompile(ext + "$")
	dir := filepath.Base(rep.ReplaceAllString(src, ""))

	destDir := filepath.Join(dest, dir)
	// ファイル名のディレクトリを作成する
	if err := os.MkdirAll(destDir, os.ModeDir); err != nil {
		return err
	}

	for _, f := range r.File {
		if f.Mode().IsDir() {
			// ディレクトリは無視して構わない
			continue
		}
		if err := saveUnZipFile(destDir, *f); err != nil {
			return err
		}
	}

	return nil
}

// saveUnZipFile 展開したZipファイルをそのままローカルに保存する
func saveUnZipFile(destDir string, f zip.File) error {
	// 展開先のパスを設定する
	destPath := filepath.Join(destDir, f.Name)
	// 子孫ディレクトリがあれば作成する
	if err := os.MkdirAll(filepath.Dir(destPath), f.Mode()); err != nil {
		return err
	}
	// Zipファイルを開く
	rc, err := f.Open()
	if err != nil {
		return err
	}
	defer rc.Close()
	// 展開先ファイルを作成する
	destFile, err := os.Create(destPath)
	if err != nil {
		return err
	}
	defer destFile.Close()
	// 展開先ファイルに書き込む
	if _, err := io.Copy(destFile, rc); err != nil {
		return err
	}

	return nil
}

func main() {
	// パスを取得
	rootPath, _ := os.Getwd()
	rootDir := filepath.Dir(rootPath)
	fmt.Println(rootDir)

	zipPath := filepath.Join(rootDir, "downloads", "test20210318.zip")
	destDir := filepath.Join(rootDir, "reserves")

	if err := unZip(zipPath, destDir); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	os.Exit(0)
}

main()のzipPathとdestDirはお好みで。

$ go version
go version go1.14.6 linux/amd64
$ go run sample/unzip/main.go

Windowsユーザーなので、VSCodeで書いて、ターミナルをWSLにして実行しています。
Goの環境設定などは先人の記事を参考にしてください。

実はもっと無意味に複雑なことをしていましたが、あちこち記事を漁るうちに気が付いたことがあります。

##zipファイル内のディレクトリを毎度作成する必要はない

zip.Fileを走査中、ディレクトリがあっても無視して構いません。

	for _, f := range r.File {
		if f.Mode().IsDir() {
			// ディレクトリは無視して構わない
			continue
		}

なぜかというと、zip.File.Nameにパスが含まれているからです。
後続処理のファイル作成時にディレクトリも作成すれば済みます。

	// 子孫ディレクトリがあれば作成する
	if err := os.MkdirAll(filepath.Dir(destPath), f.Mode()); err != nil {
		return err
	}

os.MkdirAllは既にディレクトリがあってもエラーを返しませんので、そのまま後続処理に流れます。
ついでに、zip.File.Mode()を使えば、zip内のパーミッションを引き継ぐことができます。

##io.Copyを使う

以前はバッファにいったん全読み込みして、ファイルに書き出すような処理を書いていました。

以前のコード

		buf := make([]byte, f.UncompressedSize64)
		_, err = io.ReadFull(rc, buf)

これだとメモリを無駄に消費してしまうため、readerからwriterに直接よしなにしてくれるio.Copyが良いそうです。
また、io.CopyNというのもあります。これは書き込み時にバイト数を指定できるものです。
出自のわからないzipファイルを解凍するような場合は、zip爆弾を回避するためにも利用は必須かと思います。