From 84d5063722539f11849bc8739668a16fb3027b6a Mon Sep 17 00:00:00 2001 From: "Stephen M. McQuay" Date: Fri, 7 Sep 2012 11:32:18 -0600 Subject: [PATCH] added a test of using either stdin or a file from argv --- stdin-or-argv/go.go | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 stdin-or-argv/go.go diff --git a/stdin-or-argv/go.go b/stdin-or-argv/go.go new file mode 100644 index 0000000..5718de5 --- /dev/null +++ b/stdin-or-argv/go.go @@ -0,0 +1,29 @@ +package main + +import ( + "fmt" + // "io" + "io/ioutil" + "log" + // "bufio" + "os" + "strings" +) + +var usage = "usage: splitter [file]" + +func main() { + var of *os.File + var err error + if len(os.Args) == 2 { + filename := os.Args[1] + of, err = os.Open(filename) + if err != nil { + log.Fatal(err) + } + } else { + of = os.Stdin + } + b, err := ioutil.ReadAll(of) + fmt.Printf("%v\n", strings.Fields(string(b))) +}