From ac32d560614d2fbff910335d40cea3ac8cfae00f Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Thu, 21 May 2020 20:36:57 -0400 Subject: refactoring: extract stdin and boltdb readers Signed-off-by: Dave Henderson --- data/datasource_stdin.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 data/datasource_stdin.go (limited to 'data/datasource_stdin.go') diff --git a/data/datasource_stdin.go b/data/datasource_stdin.go new file mode 100644 index 00000000..f23905d2 --- /dev/null +++ b/data/datasource_stdin.go @@ -0,0 +1,19 @@ +package data + +import ( + "io/ioutil" + "os" + + "github.com/pkg/errors" +) + +func readStdin(source *Source, args ...string) ([]byte, error) { + if stdin == nil { + stdin = os.Stdin + } + b, err := ioutil.ReadAll(stdin) + if err != nil { + return nil, errors.Wrapf(err, "Can't read %s", stdin) + } + return b, nil +} -- cgit v1.2.3