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

概要
サンプル
参考資料

概要

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

前回は以下。

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*.txt
var 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*.txt
var 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))
    }

    return nil
}

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

$ 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