62 lines
1.1 KiB
Go
62 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
. "github.com/tendermint/tendermint/common"
|
|
)
|
|
|
|
const Version = "0.0.1"
|
|
const readBufferSize = 1024
|
|
|
|
// Parse command-line options
|
|
func parseFlags() (outpath string, version bool) {
|
|
flag.StringVar(&outpath, "outpath", "stdinwriter.out", "Output file name")
|
|
flag.BoolVar(&version, "version", false, "Version")
|
|
flag.Parse()
|
|
return
|
|
}
|
|
|
|
func main() {
|
|
|
|
// Read options
|
|
outpath, version := parseFlags()
|
|
if version {
|
|
fmt.Println(Fmt("stdinwriter version %v", Version))
|
|
return
|
|
}
|
|
|
|
outfile, err := OpenAutoFile(outpath)
|
|
if err != nil {
|
|
Exit(Fmt("stdinwriter couldn't create outfile %v", outfile))
|
|
}
|
|
|
|
go writeToOutfile(outfile)
|
|
|
|
// Trap signal
|
|
TrapSignal(func() {
|
|
outfile.Close()
|
|
fmt.Println("stdinwriter shutting down")
|
|
})
|
|
}
|
|
|
|
func writeToOutfile(outfile *AutoFile) {
|
|
// Forever, read from stdin and write to AutoFile.
|
|
buf := make([]byte, readBufferSize)
|
|
for {
|
|
n, err := os.Stdin.Read(buf)
|
|
outfile.Write(buf[:n])
|
|
if err != nil {
|
|
outfile.Close()
|
|
if err == io.EOF {
|
|
os.Exit(0)
|
|
} else {
|
|
Exit("stdinwriter errored")
|
|
}
|
|
}
|
|
}
|
|
}
|