Goメモ-148 (embed メモ (埋め込み対象を絞り込み, embed.FSで取得))

• 概要
• サンプル
• 参考資料

概要

Go 1.16 から追加となった embedパッケージについてのメモ５個目。

前回は以下。

devlights.hatenablog.com

devlights.hatenablog.com

devlights.hatenablog.com

devlights.hatenablog.com

embedパッケージのドキュメントは以下。

golang.org

今回は、複数のファイルを埋め込む際に対象を絞り込んで、その内容を embed.FSで取得するサンプルです。

embed.FSは、こちらも Go1.16 で追加された io/fsの 各インターフェースを embed 側で実装しているものです。

embed.FSは、以下のインターフェースを実装しています。

• fs.ReadDirFS
• fs.ReadFileFS

なので、ファイルとディレクトリの両方を扱えるってことですね。

embedパッケージで、複数のファイルを一つの変数で埋め込んで利用する場合は、embed.FSを利用する必要があります。

埋め込みたい変数に、必要なだけ //go:embed xxxxを書けば埋め込まれます。

その際に、シェルでやるように特定のパターンに従ったファイルのみという指定もできます。

例えば以下のようにします。

//go:embed data/go*.txtvar content embed.FS

上記では、dataというディレクトリの下の go*.txtにマッチするファイルのみ埋め込んでほしいと指示しています。

サンプル

以下のサンプルでは、同一ディレクトリに配置されている helloworld.txtと dataディレクトリを埋め込んで利用しています。

helloworld.txt というファイルは以下の内容です。

hello
world

dataディレクトリの中は以下な感じ。

$ tree data
data
├── dotnet.txt
├── golang.txt
├── go.txt
└── python.txt

0 directories, 4 files

それぞれのファイルの中身は言語の名前が一行で入っているだけです。

んで、利用する側は以下のような感じ。

package embeds

import (
    "embed""fmt""io/fs""github.com/devlights/gomy/output"
)

//go:embed data/go*.txtvar filteredContent embed.FS // 埋め込むファイルをフィルタリングしている// EmbedFsFilter は、embed パッケージの機能を確認するサンプルです (埋め込むファイルをフィルタリング)func EmbedFsFilter() error {
    var (
        entries []fs.DirEntry
        buf     []byte
        err     error
    )

    entries, err = filteredContent.ReadDir("data")
    if err != nil {
        return err
    }

    for _, entry := range entries {
        var (
            name = entry.Name()
            path = fmt.Sprintf("data/%s", name)
        )

        buf, err = content.ReadFile(path)
        if err != nil {
            return err
        }

        output.Stdoutl(fmt.Sprintf("[%s]", name), string(buf))
    }

    returnnil
}

実行すると以下のようになります。

$ make run

go get -d ./...
go run -race github.com/devlights/try-golang/cmd/trygolang -onetime-example""

ENTER EXAMPLE NAME: embed_fs_filter
[Name]"embed_fs_filter"[go.txt]             golang
[golang.txt]         golang


[Elapsed]95.487µs

指定したパターンにマッチしたファイルのみが埋め込まれています。実行時には当然 helloworld.txt も data ディレクトリも必要ありません。

注意点として、バイナリにはそのまま埋め込まれていますので、バイナリエディタとかで除くと当然丸見えです。センシティブなデータは埋め込まないようにしましょう。（しないと思いますが）

参考資料

golang.org

過去の記事については、以下のページからご参照下さい。

• いろいろ備忘録日記まとめ

devlights.github.io

サンプルコードは、以下の場所で公開しています。

• いろいろ備忘録日記サンプルソース置き場

github.com

github.com

github.com