Add --stream option

.github/workflows/go.yml

@@ -2,24 +2,23 @@ name: Go
 
 on:
   push:
-    branches: [ '*' ]
+    branches: ["*"]
   pull_request:
-    branches: [ '*' ]
+    branches: ["*"]
 
 jobs:
-
   build:
     runs-on: ubuntu-latest
     steps:
-    - uses: actions/checkout@v2
+      - uses: actions/checkout@v2
 
-    - name: Set up Go
-      uses: actions/setup-go@v2
-      with:
-        go-version: 1.17
+      - name: Set up Go
+        uses: actions/setup-go@v2
+        with:
+          go-version: 1.19
 
-    - name: build
-      run: make build
+      - name: build
+        run: make build
 
-    - name: test
-      run: make test
+      - name: test
+        run: make test

README.md

@@ -267,7 +267,7 @@ the number of iterations. `Loop` must be used with `range` e.g.
 ```
 
 `Loop` can take a second argument, so that you can specify a range and
-`fakedata` will generate a random number of interations in that range. For
+`fakedata` will generate a random number of iterations in that range. For
 example:
 
 ```html

cmd/import/main.go

@@ -3,7 +3,6 @@ package main
 import (
 	"encoding/json"
 	"fmt"
-	"io/ioutil"
 	"log"
 	"net/http"
 	"os"
@@ -119,7 +118,7 @@ func main() {
 		content := fmt.Sprintf(fileTemplate, task.Var, task.Key, task.Var, data)
 
 		// Write to Go file
-		if err := ioutil.WriteFile(file, []byte(content), 0644); err != nil {
+		if err := os.WriteFile(file, []byte(content), 0644); err != nil {
 			log.Fatal(err)
 		}
 	}

integration/cli_test.go

@@ -3,7 +3,6 @@ package main
 import (
 	"flag"
 	"fmt"
-	"io/ioutil"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -56,7 +55,7 @@ func (tf *testFile) path() string {
 
 func (tf *testFile) write(content string) {
 	tf.t.Helper()
-	err := ioutil.WriteFile(tf.path(), []byte(content), 0644)
+	err := os.WriteFile(tf.path(), []byte(content), 0644)
 	if err != nil {
 		tf.t.Fatalf("could not write %s: %v", tf.name, err)
 	}
@@ -74,7 +73,7 @@ func (tf *testFile) asFile() *os.File {
 func (tf *testFile) load() string {
 	tf.t.Helper()
 
-	content, err := ioutil.ReadFile(tf.path())
+	content, err := os.ReadFile(tf.path())
 	if err != nil {
 		tf.t.Fatalf("could not read file %s: %v", tf.name, err)
 	}

integration/golden/broken-template.golden

@@ -1 +1 @@
-template: template:1: unexpected "}" in operand
+template: template:1: bad character U+007D '}'

integration/golden/file-does-not-exist.golden

@@ -9,6 +9,7 @@ Usage: fakedata [option ...] field...
   -c, --generators-with-constraints   lists available generators with constraints
   -l, --limit int                     limits rows up to n (default 10)
   -s, --separator string              specifies separator for the column format (default " ")
+  -S, --stream                        streams rows till the end of time
   -t, --table string                  table name of the sql format (default "TABLE")
   -T, --template string               Use template as input
   -v, --version                       shows version information

integration/golden/help.golden

@@ -7,6 +7,7 @@ Usage: fakedata [option ...] field...
   -c, --generators-with-constraints   lists available generators with constraints
   -l, --limit int                     limits rows up to n (default 10)
   -s, --separator string              specifies separator for the column format (default " ")
+  -S, --stream                        streams rows till the end of time
   -t, --table string                  table name of the sql format (default "TABLE")
   -T, --template string               Use template as input
   -v, --version                       shows version information

integration/golden/path-empty.golden

@@ -9,6 +9,7 @@ Usage: fakedata [option ...] field...
   -c, --generators-with-constraints   lists available generators with constraints
   -l, --limit int                     limits rows up to n (default 10)
   -s, --separator string              specifies separator for the column format (default " ")
+  -S, --stream                        streams rows till the end of time
   -t, --table string                  table name of the sql format (default "TABLE")
   -T, --template string               Use template as input
   -v, --version                       shows version information

integration/golden/unknown-format.golden

@@ -9,6 +9,7 @@ Usage: fakedata [option ...] field...
   -c, --generators-with-constraints   lists available generators with constraints
   -l, --limit int                     limits rows up to n (default 10)
   -s, --separator string              specifies separator for the column format (default " ")
+  -S, --stream                        streams rows till the end of time
   -t, --table string                  table name of the sql format (default "TABLE")
   -T, --template string               Use template as input
   -v, --version                       shows version information

integration/golden/unknown-generators.golden

@@ -9,6 +9,7 @@ Usage: fakedata [option ...] field...
   -c, --generators-with-constraints   lists available generators with constraints
   -l, --limit int                     limits rows up to n (default 10)
   -s, --separator string              specifies separator for the column format (default " ")
+  -S, --stream                        streams rows till the end of time
   -t, --table string                  table name of the sql format (default "TABLE")
   -T, --template string               Use template as input
   -v, --version                       shows version information

main.go

@@ -3,7 +3,7 @@ package main
 import (
 	"bytes"
 	"fmt"
-	"io/ioutil"
+	"io"
 	"math/rand"
 	"os"
 	"time"
@@ -42,7 +42,7 @@ func isPipe() bool {
 
 func findTemplate(path string) string {
 	if path != "" {
-		tp, err := ioutil.ReadFile(path)
+		tp, err := os.ReadFile(path)
 		if err != nil {
 			fmt.Printf("unable to read input: %s", err)
 			os.Exit(1)
@@ -52,7 +52,7 @@ func findTemplate(path string) string {
 	}
 
 	if isPipe() {
-		tp, err := ioutil.ReadAll(os.Stdin)
+		tp, err := io.ReadAll(os.Stdin)
 		if err != nil {
 			fmt.Printf("unable to read input: %s", err)
 			os.Exit(1)
@@ -70,6 +70,7 @@ func main() {
 		generatorFlag   = flag.StringP("generator", "g", "", "show help for a specific generator")
 		constraintsFlag = flag.BoolP("generators-with-constraints", "c", false, "lists available generators with constraints")
 		limitFlag       = flag.IntP("limit", "l", 10, "limits rows up to n")
+		streamFlag      = flag.BoolP("stream", "S", false, "streams rows till the end of time")
 		formatFlag      = flag.StringP("format", "f", "column", "generates rows in f format. Available formats: column|sql")
 		tableFlag       = flag.StringP("table", "t", "TABLE", "table name of the sql format")
 		separatorFlag   = flag.StringP("separator", "s", " ", "specifies separator for the column format")
@@ -164,6 +165,11 @@ func main() {
 		os.Exit(1)
 	}
 
+	if *streamFlag {
+		for {
+			fmt.Println(columns.GenerateRow(formatter))
+		}
+	}
 	for i := 0; i < *limitFlag; i++ {
 		fmt.Println(columns.GenerateRow(formatter))
 	}

pkg/fakedata/generator.go

@@ -2,7 +2,6 @@ package fakedata
 
 import (
 	"fmt"
-	"io/ioutil"
 	"math/rand"
 	"os"
 	"sort"
@@ -184,7 +183,7 @@ func file(path string) (func() string, error) {
 
 	filePath := strings.Trim(path, "'\"")
 
-	content, err := ioutil.ReadFile(filePath)
+	content, err := os.ReadFile(filePath)
 	if err != nil {
 		return nil, fmt.Errorf("could not read file %s: %v", filePath, err)
 	}