You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

61 lines
1.1 KiB

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")
}
}
}
}