Qiita Teams that are logged in
You are not logged in to any team

Log in to Qiita Team
Community
OrganizationAdvent CalendarQiitadon (β)
Service
Qiita JobsQiita ZineQiita Blog
Help us understand the problem. What is going on with this article?

GolangでBufferからzipファイルを作成

More than 3 years have passed since last update.
//中間ファイルを作成せずに圧縮ファイルを生成する手法
package main

import (
    "archive/zip"
    "bytes"
    "io/ioutil"
    "os"
)

func before() {
    f, err := os.Create("sample.txt")
    if err != nil {
        panic(err)
    }
    f.Write([]byte("sample"))
    f.Close()
}

func main() {
    before()

    //圧縮対象ファイル取得
    files := find()

    //圧縮ファイルデータ生成
    //この時点では、メモリ上に存在する
    b := compress(files)

    //例としてこのタイミングで、ファイル化
    //この時点では、あまり意味は無い
    //ローカルに残さず、s3にuploadするなどが可能となる
    //file ioがかからないので、高速化
    if err := save(b); err != nil {
        panic(err)
    }
}

func find() []string {
    return []string{"sample.txt"}
}

func save(b *bytes.Buffer) error {
    zf, err := os.Create("sample.zip")
    if err != nil {
        return err
    }
    zf.Write(b.Bytes())
    zf.Close()
    return nil
}

func compress(files []string) *bytes.Buffer {
    b := new(bytes.Buffer)
    w := zip.NewWriter(b)

    for _, file := range files {
        info, _ := os.Stat(file)

        hdr, _ := zip.FileInfoHeader(info)
        hdr.Name = "files/" + file
        f, err := w.CreateHeader(hdr)
        if err != nil {
            panic(err)
        }

        body, err := ioutil.ReadFile(file)
        if err != nil {
            panic(err)
        }
        f.Write(body)
    }

    w.Close()

    return b
}

本来は、S3にアップロードするなど、ローカル以外に保存する際に
ファイルIOを減らす目的で書きました。

https://play.golang.org/p/x9MS1Z_jmz

shinofara
work: Emacs/VisualStudioCode Golang/PHP/Pyhton/Shell AWS/Docker/VirtualBox/Mac Vagrant/Terraform/Packer/Ansible OSS/LT like: Photo/Diving/Snowboard history: Y! -> schoo -> Y! -> MedPeer
https://log.shinofara.xyz/
moneyforward
「お金を前へ。人生をもっと前へ。」をMissionとして、個人向け、法人向け様々なFintechサービスを開発、提供しています。
https://moneyforward.com/
Why not register and get more from Qiita?
  1. We will deliver articles that match you
    By following users and tags, you can catch up information on technical fields that you are interested in as a whole
  2. you can read useful information later efficiently
    By "stocking" the articles you like, you can search right away