Windows の Go で stdin に書きつつ外部プログラムを起動するとハマる

標準入力からデータを受け取るようなプログラムを起動させようとして、stdin に WriteString したらハマりになったので対処法をメモしておきます。

ProtocolBuffer のプラグイン書いてて踏みましたが、Go の周辺ツールとかカスタムしようとすると同じ感じで踏むかもしれません。

Windows で 4KB 以上のデータを書き込んだ場合にのみ発生し、4KB 以下の場合や Windows 以外の OS の場合は発生しません。

例えばこんなコード。

func main() {
    content, _ := ioutil.ReadFile("test.txt") // 適当に 4KB 以上のテキスト

    c := exec.Command("more")
    stdin, _ := c.StdinPipe()

    io.WriteString(stdin, string(content)) // ここで返ってこない

    stdin.Close()
}

恐らく、パイプのバッファが足りなくなって、空くのを待っているけれども、
書き込みが終わってないのでプログラムが起動せず、空ける人がいないという状況と思われる。

以下のようにしたら、とりあえず動いた。

func main() {
    content, _ := ioutil.ReadFile("test.txt") // 適当に 4KB 以上のテキスト

    c := exec.Command("more")
    stdin, _ := c.StdinPipe()

    go func() {
        defer stdin.Close()
        io.WriteString(stdin, string(content))
    }()

    output, _ := c.CombinedOutput()
    fmt.Print(string(output))
}