Use bufio.Scanner

This commit is contained in:
Tomás Senart
2013-08-16 19:46:40 +02:00
parent 7d13447b64
commit 2a476cbb49

View File

@@ -22,16 +22,11 @@ func NewTargetsFromFile(filename string) (Targets, error) {
}
func NewTargets(source io.Reader) (Targets, error) {
reader := bufio.NewReader(source)
targets := make([]*http.Request, 0)
scanner := bufio.NewScanner(source)
for {
line, err := reader.ReadString('\n')
if err != nil && err == io.EOF {
break
} else if err != nil {
return targets, err
}
for scanner.Scan() {
line := scanner.Text()
if line = strings.TrimSpace(line); line == "" { // Empty line
continue
}
@@ -46,6 +41,9 @@ func NewTargets(source io.Reader) (Targets, error) {
}
targets = append(targets, req)
}
if err := scanner.Err(); err != nil {
return targets, err
}
return targets, nil
}