Go Scanner:如何知道我们是在处理新行还是被截断的字符串?
我基本上需要处理从流中读取的有限缓冲区中的每个字符串行。使用bufio.Scanner,我可以一行一行地使用扫描器,但不得不使用似乎过于复杂的解决方案来检测“截断”。有更好的方法吗?谢谢。我对任何自由党或任何东西都不感兴趣Go Scanner:如何知道我们是在处理新行还是被截断的字符串?,go,buffer,Go,Buffer,我基本上需要处理从流中读取的有限缓冲区中的每个字符串行。使用bufio.Scanner,我可以一行一行地使用扫描器,但不得不使用似乎过于复杂的解决方案来检测“截断”。有更好的方法吗?谢谢。我对任何自由党或任何东西都不感兴趣 func (p *Parser) Read(data []byte, tmpline *string, n int, bufSize int) { var line string strdata := string(data) scanner := b
func (p *Parser) Read(data []byte, tmpline *string, n int, bufSize int) {
var line string
strdata := string(data)
scanner := bufio.NewScanner(strings.NewReader(strdata))
line = ""
for scanner.Scan() {
if line != "" {
if p.lineProcessor != nil {
p.lineProcessor(line)
}
}
line = scanner.Text()
if *tmpline != "" {
line = *tmpline + line //prepend line here
*tmpline = ""
}
}
if n == bufSize && data[bufSize-1] != '\n' { //detecting a fragment here, overcomplicated?
*tmpline = line
} else {
//fmt.Println("last line >[" + line + "]")
if p.lineProcessor != nil {
p.lineProcessor(line)
}
}
}
func (p *Parser) Handle(r io.Reader, bufSize int) ([]byte, error) {
var out []byte
buf := make([]byte, bufSize)
var n int
var err error
tmpline := ""
for {
n, err = r.Read(buf[:])
if n > 0 {
data := buf[:n]
p.Read(data, &tmpline, n, bufSize)
}
if err != nil {
if err == io.EOF {
err = nil
}
break
}
}
return out, err
}
main
...
cmd := exec.Command("ping", "8.8.8.8")
//var stdout, stderr []byte
var errStdout, errStderr error
//this is the type of stream I want to process **stdoutIn** and **stderrIn**
stdoutIn, _ := cmd.StdoutPipe()
stderrIn, _ := cmd.StderrPipe()
// 10 is the bufSize
parser.Init(stdoutIn, stderrIn, wg, 10, lineProcessor)
...
编辑:我尝试实施建议的解决方案。现在,我已经打破了工作(尽管看起来过于复杂)。输出似乎非常时髦。以下是完整代码和完整输出:
package main
import (
"bufio"
"fmt"
"io"
"log"
"os/exec"
"sync"
)
type LineProcessor func(string)
type Parser struct {
r io.Reader
lineProcessor LineProcessor
scanner bufio.Scanner
}
func (p *Parser) Init(stdoutIn io.Reader, stderrIn io.Reader,
wg sync.WaitGroup,
lineProcessor LineProcessor) {
wg.Add(2)
p.lineProcessor = lineProcessor
p.scanner = *bufio.NewScanner(stdoutIn)
go p.Handler(stdoutIn, wg)
go p.Handler(stderrIn, wg)
}
func (p *Parser) Handler(r io.Reader, wg sync.WaitGroup) { // ([]byte, error)
var line string
for p.scanner.Scan() {
line = p.scanner.Text()
if p.lineProcessor != nil {
p.lineProcessor(line)
}
}
wg.Done()
}
func lineProcessor(line string) {
fmt.Println(line)
}
func main() {
var err error
cmd := exec.Command("ping", "8.8.8.8")
var errStdout, errStderr error
stdoutIn, _ := cmd.StdoutPipe()
stderrIn, _ := cmd.StderrPipe()
var parser Parser
var wg sync.WaitGroup
parser.Init(stdoutIn, stderrIn, wg, lineProcessor)
err = cmd.Start()
if err != nil {
log.Fatalf("cmd.Start() failed with '%s'\n", err)
}
fmt.Printf("Waiting\n")
wg.Wait()
err = cmd.Wait()
if err != nil {
log.Fatalf("cmd.Run() failed with %s\n", err)
}
if errStdout != nil || errStderr != nil {
log.Fatal("failed to capture stdout or stderr\n")
}
}
$./buggysolution
Waiting
PING 8.8.8.8 (8.8.8.8): 56 data bytes
64 bytes from 8.8.8.8: icmp_seq=0 ttl=52 time=4.786 ms
64 bytes from 8.8.8.8: icmp_seq=2 ttl=52 time=3.661 ms
64 bytes from 8.8.8.8: icmp_seq=4 ttl=52 time=4.117 ms
64 bytes from 8.8.8.8: icmp_seq=6 ttl=52 time=4.172 ms
64 bytes from 8.8.8.8: icmp_seq=8 ttl=52 time=3.584 ms
64 bytes from 8.8.8.8: icmp_seq=10 ttl=52 time=4.301 ms
mp_seq=11 ttl=52 time=4.534 ms
64 bytes from 8.8.8.8: icmp_seq=12 ttl=52 time=4.349 ms
64 bytes from 8.8.8.8: icmp_seq=13 ttl=52 time=4.923 ms
64 bytes from 8.8.8.8: icmp_seq=14 ttl=52 time=4.349 ms
64 bytes from 8.8.8.8: icmp_seq=15 ttl=52 time=4.106 ms
64 bytes from 8.8.8.8: icmp_seq=16 ttl=52 time=4.270 ms
64 bytes from 8.8.8.8: icmp_seq=17 ttl=52 time=4.231 ms
64 bytes from 8.8.8.8: icmp_seq=18 ttl=52 time=4.915 ms
64 bytes from 8.8.8.8: icmp_seq=19 ttl=52 time=4.487 ms
64 bytes from 8.8.8.8: icmp_seq=20 ttl=52 time=4.182 ms
64 bytes from 8.8.8.8: icmp_seq=21 ttl=52 time=4.369 ms
64 bytes from 8.8.8.8: icmp_seq=22 ttl=52 time=4.287 ms
64 bytes from 8.8.8.8: icmp_seq=23 ttl=52 time=3.922 ms
64 bytes from 8.8.8.8: icmp_seq=24 ttl=52 time=4.905 ms
64 bytes from 8.8.8.8: icmp_seq=25 ttl=52 time=4.226 ms
64 bytes from 8.8.8.8: icmp_seq=27 ttl=52 time=4.052 ms
64 bytes from 8.8.8.8: icmp_seq=29 ttl=52 time=3.453 ms
64 bytes from 8.8.8.8: icmp_seq=31 ttl=52 time=5.103 ms
64 bytes from 8.8.8.8: icmp_seq=33 ttl=52 time=4.066 ms
64 bytes from 8.8.8.8: icmp_seq=35 ttl=52 time=4.128 ms
64 bytes from 8.8.8.8: icmp_seq=37 ttl=52 time=4.982 ms
64 bytes from 8.8.8.8: icmp_seq=1 ttl=52 time=4.206 ms
.64 bytes from 8.8.8.8: icmp_seq=39 ttl=52 time=4.215 ms
.8: icmp_seq=3 ttl=52 time=4.218 ms
tl=52 time=3.650 ms
8: icmp_seq=2 ttl=52 time=3.661 ms
ttl=52 time=4.791 ms
: icmp_seq=5 ttl=52 time=3.581 ms
ttl=52 time=4.211 ms
icmp_seq=4 ttl=52 time=4.117 ms
4 ttl=52 time=4.245 ms
icmp_seq=7 ttl=52 time=4.955 ms
45 ttl=52 time=4.518 ms
cmp_seq=6 ttl=52 time=4.172 ms
=46 ttl=52 time=4.764 ms
seq 9
^C
感谢您的修复。这不是使用
bufio.Scanner
的方式
bufio.Scanner
带一个读卡器,可以直接从读卡器中返回行
如果令牌大于bufio.Scanner.maxTokenSize
,则bufio.Scanner.maxTokenSize
将失败,默认值为MaxScanTokenSize
,但它不会返回截断的结果
您还可以在处检查测试套件(并非使用的MaxTokenSize方法是使用特殊方案定义的,仅在测试期间存在,请参阅)
您还可以在扫描之前定义自己的基础缓冲区以更改该值
要推出您自己的版本,它可能看起来像这样,但是沿着这条路走下去需要编写额外的测试和基准测试
在多次评论之后,这里是OP想要的:
package main
import (
"bufio"
"io"
"log"
"os/exec"
)
func main() {
cmd := exec.Command("ping", "8.8.8.8")
var out io.Reader
{
stdout, err := cmd.StdoutPipe()
if err != nil {
log.Fatal(err)
}
stderr, err := cmd.StderrPipe()
if err != nil {
log.Fatal(err)
}
out = io.MultiReader(stdout, stderr)
}
if err := cmd.Start(); err != nil {
log.Fatal(err)
}
// defer cmd.Process.Kill()
s := bufio.NewScanner(out)
for s.Scan() {
log.Println(s.Text())
}
// if out closes, cmd closed.
log.Println("all done")
}
行何时被“截断”?您可以添加一些示例输入行吗?您所说的“截断”到底是什么意思?例如,如果您正在读取的缓冲区是4行中的712字节,但您的缓冲区只有512和p。Read将只获得512字节,最后一行中只有3行和34字节,因此,您需要存储剩余的数据,并在下一个缓冲区的第一行(包含200字节)前加上前缀。您首先对流进行分块,然后尝试从中读取行。这太过分了,很可能是错误的。取而代之的是,直接使用扫描仪从读卡器中读取。@bserdar您能举个例子吗?这段代码的输入(数据字节数组)来自处理“永远”运行的exec.Command执行的输出,它不是网络数据或平面文件。感谢@mh cbon的努力,如果我不想使用Scanner,还有什么替代方案?我编辑并添加了更多信息,现在使用bufio查看是否清晰。扫描仪是否正确。但是重新实例化一个新的扫描仪来扫描一行是不正确的用法。假设ping输出到stdout,您可以制作一个
bufio.NewScanner(cmd.stdout)
,并直接处理扫描仪弹出的行。请查看我所做的编辑,看看这是否是您的意思,看看您是否能找到错误。谢谢!!