为每行多个令牌使用Scanner的更好方法



我正在尝试解析一个由键,空格,数字和换行符组成的行文件。

我的代码可以工作,但是我感觉不对劲。有更好的方法来使用扫描仪吗?特别是,我不喜欢在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)
    }
}

最新更新