当输入来自重定向时，为什么 bufio.NewReader 不能正常工作？答案

【问题标题】：why does bufio.NewReader does not work proprely when input come from redirection?当输入来自重定向时，为什么 bufio.NewReader 不能正常工作？
【发布时间】：2020-10-23 15:06:22
【问题描述】：

我目前正在尝试在 CodeChef 网站上解决 this challenge 手动输入输入时，我得到了问题中所述问题的预期答案。

但是当我尝试使用cat in.txt | go run my_program.go 时，我的程序的输出是不同的。

如您所见，我尝试使用fmt.Printf 进行调试，看起来bufio.NewReader 随机错过了来自管道的输入。

我想知道我做错了什么，以及是否有替代方法可以读取包含 Golang 中多个积分器的整行。

感谢您帮助我！

in.txt的内容：

手动输入时正常输出：

使用cat in.txt | go run my_program.go 输入文本时输出调试消息

debug <1 2 3
>
1 1
debug <>
1 1
debug <>
1 1
debug <>
1 1

我的程序：


package main

import (
        "fmt"
        "os"
        "strings"
        "bufio"
        "strconv"
)

func main() {
    t, n := 0, 0
    fmt.Scanf("%d", &t)
    for i := 0; i < t; i++ {
            fmt.Scanf("%d\n", &n)
            v := make([]int, n)
            rd := bufio.NewReader(os.Stdin)
            text, _ := rd.ReadString('\n')
            fmt.Printf("debug <%s>\n", text)
            arr := strings.Split(strings.TrimSuffix(text, "\n"), " ")
            for k := 0; k < len(arr); k++ {
                    v[k], _ =  strconv.Atoi(arr[k])
            }
            fmt.Println(calc_intersect(v))
    }
}

func calc_intersect(v []int) (int, int) {
    smt, lgt := 100, 0
    scenario := make([]int, len(v))
    for sc := 0; sc < len(v); sc++  {
            infect := make([]bool, len(v))
            infect[sc] = true
            scenario[sc] = simulate(v, infect)
    }
    for i := 0; i < len(scenario); i++ {
            if scenario[i] <= smt {
                    smt = scenario[i]
            }
            if scenario[i] > lgt {
                    lgt = scenario[i]
            }
    }
    return smt, lgt
}

func simulate(v []int, infect []bool) int {
            var nb_inf int = 1
            pos := make([]int, len(v))
            for time := 0; time < 25; time++ {
                    for p := 0; p < len(v); p++ {
                            pos[p] = (v[p] * time) + p
                    }
                    for a := 0; a < len(v); a++ {
                            check_intersect(pos, infect, &nb_inf, a)
                    }
            }
            return nb_inf
}

func check_intersect(pos []int, infect []bool, nb_inf *int, a int) {
            for b := 0; b < len(pos); b++ {
                    if a == b {
                            continue
                    }
                    if pos[a] == pos[b] && (infect[a] || infect[b]) && !(infect[a] && infect[b]) {
                            *nb_inf += 1
                            infect[a], infect[b] = true, true
                    }
            }

}

【问题讨论】：

标签： shell go input pipe

【解决方案1】：

您对bufio.NewReader 的使用是错误的。每次通过i 循环创建一个新的缓冲阅读器。如果输入可用，读取器可能会提前读取，并且当您不重新使用未使用缓冲输入的读取器时。

只需将rd := bufio.NewReader(os.Stdin) 移出循环即可解决问题。

当您通过标准输入从控制台读取时，您的代码可以正常工作，因为您的输入速度不够快，读者无法尝试缓冲输入。

【讨论】：