add bytes returned to FETCH payload (#386)

venkatsridhar95 · web-flow · commit c8010a3bd74f · 2024-04-19T16:27:27.000-07:00
diff --git a/worker/shared/cmdprocessor.go b/worker/shared/cmdprocessor.go
@@ -660,6 +660,7 @@ outloop:
 			for i := range writeCols {
 				readCols[i] = &writeCols[i]
 			}
+			fetchBufferLen := 0
 			for cp.rows.Next() {
 				err = cp.rows.Scan(readCols...)
 				if err != nil {
@@ -681,8 +682,10 @@ outloop:
 						logger.GetLogger().Log(logger.Debug, "query result", outstr)
 					} // causes high volume logs and can cause CI test issues */
 					nss = append(nss, netstring.NewNetstringFrom(common.RcValue, []byte(outstr)))
+					fetchBufferLen += len(outstr)
 				}
 			}
+			calt.AddDataInt("psize", int64(fetchBufferLen))
 			if len(nss) > 0 {
 				resns := netstring.NewNetstringEmbedded(nss)
 				err = WriteAll(cp.SocketOut, resns)

Original file line number	Diff line number	Diff line change
`@@ -660,6 +660,7 @@ outloop:`
`660`	`660`	`for i := range writeCols {`
`661`	`661`	`readCols[i] = &writeCols[i]`
`662`	`662`	`}`
	`663`	`+ fetchBufferLen := 0`
`663`	`664`	`for cp.rows.Next() {`
`664`	`665`	`err = cp.rows.Scan(readCols...)`
`665`	`666`	`if err != nil {`
`@@ -681,8 +682,10 @@ outloop:`
`681`	`682`	`logger.GetLogger().Log(logger.Debug, "query result", outstr)`
`682`	`683`	`} // causes high volume logs and can cause CI test issues */`
`683`	`684`	`nss = append(nss, netstring.NewNetstringFrom(common.RcValue, []byte(outstr)))`
	`685`	`+ fetchBufferLen += len(outstr)`
`684`	`686`	`}`
`685`	`687`	`}`
	`688`	`+ calt.AddDataInt("psize", int64(fetchBufferLen))`
`686`	`689`	`if len(nss) > 0 {`
`687`	`690`	`resns := netstring.NewNetstringEmbedded(nss)`
`688`	`691`	`err = WriteAll(cp.SocketOut, resns)`