properly calculate readCalls

This commit is contained in:
Aliaksandr Valialkin 2020-07-10 11:11:17 +03:00
parent 9c350bc20d
commit 6fe3c48a6e
3 changed files with 3 additions and 3 deletions

View File

@ -25,7 +25,6 @@ var (
//
// callback shouldn't hold rows after returning.
func ParseStream(req *http.Request, callback func(rows []Row) error) error {
readCalls.Inc()
q := req.URL.Query()
format := q.Get("format")
cds, err := ParseColumnDescriptors(format)
@ -53,6 +52,7 @@ func ParseStream(req *http.Request, callback func(rows []Row) error) error {
}
func (ctx *streamContext) Read(r io.Reader, cds []ColumnDescriptor) bool {
readCalls.Inc()
if ctx.err != nil {
return false
}

View File

@ -24,7 +24,6 @@ var (
//
// callback shouldn't hold rows after returning.
func ParseStream(r io.Reader, isGzipped bool, precision, db string, callback func(db string, rows []Row) error) error {
readCalls.Inc()
if isGzipped {
zr, err := common.GetGzipReader(r)
if err != nil {
@ -62,6 +61,7 @@ func ParseStream(r io.Reader, isGzipped bool, precision, db string, callback fun
}
func (ctx *streamContext) Read(r io.Reader, tsMultiplier int64) bool {
readCalls.Inc()
if ctx.err != nil {
return false
}

View File

@ -21,7 +21,6 @@ var maxLineLen = flag.Int("import.maxLineLen", 100*1024*1024, "The maximum lengt
//
// callback shouldn't hold rows after returning.
func ParseStream(req *http.Request, callback func(rows []Row) error) error {
readCalls.Inc()
r := req.Body
if req.Header.Get("Content-Encoding") == "gzip" {
zr, err := common.GetGzipReader(r)
@ -43,6 +42,7 @@ func ParseStream(req *http.Request, callback func(rows []Row) error) error {
}
func (ctx *streamContext) Read(r io.Reader) bool {
readCalls.Inc()
if ctx.err != nil {
return false
}