lib/protoparser: move common code for detecting timeouts to ReadLinesBlockExt

Updates https://github.com/VictoriaMetrics/VictoriaMetrics/issues/696
This commit is contained in:
Aliaksandr Valialkin 2020-08-14 20:39:43 +03:00
parent 4beab7ad39
commit 7554be172d
4 changed files with 20 additions and 37 deletions

View file

@ -60,10 +60,14 @@ again:
return dstBuf, tailBuf, nil return dstBuf, tailBuf, nil
} }
var ne net.Error var ne net.Error
if errors.As(err, &ne) && ne.Timeout() && fasttime.UnixTimestamp() == startTime { if errors.As(err, &ne) && ne.Timeout() {
// Prevent from busy loop when timeout erorrs are returned immediately. if fasttime.UnixTimestamp() == startTime {
// See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/696 . // Prevent from busy loop when timeout erorrs are returned immediately.
return dstBuf, tailBuf, fmt.Errorf("detected busy loop with repeated timeout error") // See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/696 .
return dstBuf, tailBuf, fmt.Errorf("detected busy loop with repeated timeout error")
}
// Return empty results for an ordinary timeout.
return dstBuf, tailBuf, nil
} }
return dstBuf, tailBuf, err return dstBuf, tailBuf, err
} }

View file

@ -1,7 +1,6 @@
package graphite package graphite
import ( import (
"errors"
"flag" "flag"
"fmt" "fmt"
"io" "io"
@ -54,17 +53,11 @@ func (ctx *streamContext) Read(r io.Reader) bool {
} }
ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf) ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf)
if ctx.err != nil { if ctx.err != nil {
var ne net.Error if ctx.err != io.EOF {
if errors.As(ctx.err, &ne) && ne.Timeout() { readErrors.Inc()
// Flush the read data on timeout and try reading again. ctx.err = fmt.Errorf("cannot read graphite plaintext protocol data: %w", ctx.err)
ctx.err = nil
} else {
if ctx.err != io.EOF {
readErrors.Inc()
ctx.err = fmt.Errorf("cannot read graphite plaintext protocol data: %w", ctx.err)
}
return false
} }
return false
} }
ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf)) ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf))
rowsRead.Add(len(ctx.Rows.Rows)) rowsRead.Add(len(ctx.Rows.Rows))

View file

@ -1,7 +1,6 @@
package opentsdb package opentsdb
import ( import (
"errors"
"flag" "flag"
"fmt" "fmt"
"io" "io"
@ -53,17 +52,11 @@ func (ctx *streamContext) Read(r io.Reader) bool {
} }
ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf) ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf)
if ctx.err != nil { if ctx.err != nil {
var ne net.Error if ctx.err != io.EOF {
if errors.As(ctx.err, &ne) && ne.Timeout() { readErrors.Inc()
// Flush the read data on timeout and try reading again. ctx.err = fmt.Errorf("cannot read OpenTSDB put protocol data: %w", ctx.err)
ctx.err = nil
} else {
if ctx.err != io.EOF {
readErrors.Inc()
ctx.err = fmt.Errorf("cannot read OpenTSDB put protocol data: %w", ctx.err)
}
return false
} }
return false
} }
ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf)) ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf))
rowsRead.Add(len(ctx.Rows.Rows)) rowsRead.Add(len(ctx.Rows.Rows))

View file

@ -1,7 +1,6 @@
package prometheus package prometheus
import ( import (
"errors"
"fmt" "fmt"
"io" "io"
"net" "net"
@ -54,17 +53,11 @@ func (ctx *streamContext) Read(r io.Reader) bool {
} }
ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf) ctx.reqBuf, ctx.tailBuf, ctx.err = common.ReadLinesBlock(r, ctx.reqBuf, ctx.tailBuf)
if ctx.err != nil { if ctx.err != nil {
var ne net.Error if ctx.err != io.EOF {
if errors.As(ctx.err, &ne) && ne.Timeout() { readErrors.Inc()
// Flush the read data on timeout and try reading again. ctx.err = fmt.Errorf("cannot read Prometheus exposition data: %w", ctx.err)
ctx.err = nil
} else {
if ctx.err != io.EOF {
readErrors.Inc()
ctx.err = fmt.Errorf("cannot read Prometheus exposition data: %w", ctx.err)
}
return false
} }
return false
} }
ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf)) ctx.Rows.Unmarshal(bytesutil.ToUnsafeString(ctx.reqBuf))
rowsRead.Add(len(ctx.Rows.Rows)) rowsRead.Add(len(ctx.Rows.Rows))