Skip to content

Commit c587faa

Browse files
committed
Allow filenames to be read from stdin when using -L
1 parent 68d0e6d commit c587faa

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

main.go

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -48,17 +48,25 @@ func getFileNames() ([]string, error) {
4848
return names, nil
4949
}
5050

51-
in, err := os.Open(inputFile)
52-
if err != nil {
53-
return nil, err
54-
}
51+
var scanner *bufio.Scanner
52+
if inputFile != "-" {
53+
in, err := os.Open(inputFile)
54+
if err != nil {
55+
return nil, err
56+
}
5557

56-
defer in.Close()
58+
defer in.Close()
59+
scanner = bufio.NewScanner(in)
60+
} else {
61+
scanner = bufio.NewScanner(os.Stdin)
62+
}
5763

58-
scanner := bufio.NewScanner(in)
5964
for scanner.Scan() {
6065
names = append(names, scanner.Text())
6166
}
67+
if err := scanner.Err(); err != nil {
68+
return nil, err
69+
}
6270

6371
return names, nil
6472
}

0 commit comments

Comments
 (0)