Use stdin for example data reader instead of file path

README.md

@@ -349,7 +349,15 @@ There are example Produder and Consumer code in `/cmd` directory. These should h
 
 The examples run locally against [Kinesis Lite](https://github.com/mhart/kinesalite).
 
-    $ kinesalite
+    $ kinesalite &
+
+Produce data to the stream:
+
+		$ cat cmd/producer/users.txt  | go run cmd/producer/main.go --stream myStream
+
+Consume data from the stream:
+
+		$ go run cmd/consumer/main.go --stream myStream
 
 ## Contributing
 

cmd/producer/README.md

@@ -4,4 +4,4 @@ A prepopulated file with JSON users is available on S3 for seeing the stream.
 
 ## Running the code
 
-    $ go run main.go --stream streamName
+    $ cat users.txt | go run main.go --stream streamName

cmd/producer/main.go

@@ -21,13 +21,6 @@ func main() {
 	)
 	flag.Parse()
 
-	// open dummy user data
-	f, err := os.Open("users.txt")
-	if err != nil {
-		log.Fatal("Cannot open users.txt file")
-	}
-	defer f.Close()
-
 	var records []*kinesis.PutRecordsRequestEntry
 
 	var client = kinesis.New(session.Must(session.NewSession(
@@ -43,7 +36,8 @@ func main() {
 	}
 
 	// loop over file data
-	b := bufio.NewScanner(f)
+	b := bufio.NewScanner(os.Stdin)
+
 	for b.Scan() {
 		records = append(records, &kinesis.PutRecordsRequestEntry{
 			Data:         b.Bytes(),
@@ -79,8 +73,15 @@ func createStream(client *kinesis.Kinesis, streamName *string) error {
 			ShardCount: aws.Int64(2),
 		},
 	)
+	if err != nil {
+		return err
+	}
 
-	return err
+	return client.WaitUntilStreamExists(
+		&kinesis.DescribeStreamInput{
+			StreamName: streamName,
+		},
+	)
 }
 
 func putRecords(client *kinesis.Kinesis, streamName *string, records []*kinesis.PutRecordsRequestEntry) {