modify pilosactl import to allow it to read from stdin

pass a "-" argument for stdin
This commit is contained in:
Matt Jaffee 2017-02-14 13:38:58 -06:00
parent 0737a23294
commit 55fa1a24b9

View file

@ -128,15 +128,22 @@ func (cmd *ImportCommand) Run(ctx context.Context) error {
func (cmd *ImportCommand) importPath(ctx context.Context, path string) error {
a := make([]pilosa.Bit, 0, cmd.BufferSize)
// Open file for reading.
f, err := os.Open(path)
if err != nil {
return err
}
defer f.Close()
var r *csv.Reader
if path != "-" {
// Open file for reading.
f, err := os.Open(path)
if err != nil {
return err
}
defer f.Close()
// Read rows as bits.
r = csv.NewReader(f)
} else {
r = csv.NewReader(cmd.Stdin)
}
// Read rows as bits.
r := csv.NewReader(f)
r.FieldsPerRecord = -1
rnum := 0
for {