我正在尝试解析一个由键,空格,数字和换行符组成的行文件。
我的代码可以工作,但是我感觉不对劲。有更好的方法来使用扫描仪吗?特别是,我不喜欢在for循环中没有任何保护的Scan()。
func TestScanner(t *testing.T) {
const input = `key1 62128128n
key2 8337182720n
key3 7834959872n
key4 18001920n
key5 593104896n`
scanner := bufio.NewScanner(strings.NewReader(input))
scanner.Split(bufio.ScanWords)
for scanner.Scan() {
key := scanner.Text()
scanner.Scan()
value := scanner.Text();
fmt.Printf("k: %v, v: %vn", key, value)
}
}
您不应该在input
中使用n
,并始终检查错误。
工作示例代码:
package main
import (
"bufio"
"fmt"
"strings"
)
func main() {
const input = `key1 62128128
key2 8337182720
key3 7834959872
key4 18001920
key5 593104896`
scanner := bufio.NewScanner(strings.NewReader(input))
scanner.Split(bufio.ScanWords)
for scanner.Scan() {
key := scanner.Text()
if !scanner.Scan() {
break
}
value := scanner.Text()
fmt.Printf("k: %v, v: %vn", key, value)
}
}
输出:k: key1, v: 62128128
k: key2, v: 8337182720
k: key3, v: 7834959872
k: key4, v: 18001920
k: key5, v: 593104896
你也可以使用Fscan
扫描到所需的类型,像这样:
package main
import "fmt"
import "strings"
func main() {
const input = `key1 62128128
key2 8337182720
key3 7834959872
key4 18001920
key5 593104896`
rdr := strings.NewReader(input)
for {
k, v := "", 0
n, _ := fmt.Fscan(rdr, &k, &v)
if n != 2 {
//fmt.Println(err)
break
}
fmt.Printf("%T: %[1]v, %T: %[2]vn", k, v)
}
}
输出:string: key1, int: 62128128
string: key2, int: 8337182720
string: key3, int: 7834959872
string: key4, int: 18001920
string: key5, int: 593104896
实际上,这样做是完全安全的,因为Scan()验证输入并设置一个错误,而您可以使用Err()获得这个错误。
所以如果你想检查Scan()是否失败,你必须在循环结束时检查,如许多例子所示。
你的代码应该是:
func TestScanner(t *testing.T) {
const input = `key1 62128128
key2 8337182720
key3 7834959872
key4 18001920
key5 593104896`
scanner := bufio.NewScanner(strings.NewReader(input))
scanner.Split(bufio.ScanWords)
for scanner.Scan() {
key := scanner.Text()
scanner.Scan()
value := scanner.Text();
fmt.Printf("k: %v, v: %vn", key, value)
}
if err := scanner.Err(); err != nil {
fmt.Printf("Invalid input: %s", err)
}
}