From 5a97668ad69582df4ae5b1671a997f0517365907 Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@victoriametrics.com>
Date: Tue, 23 Jan 2024 18:39:28 +0200
Subject: [PATCH] lib/handshake: substitute time.Now() with
 fastttime.UnixTimestamp(), since profiling shows time.Now() is slow

---
 lib/handshake/buffered_conn.go | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/lib/handshake/buffered_conn.go b/lib/handshake/buffered_conn.go
index b496ee5e7..abfd82c7a 100644
--- a/lib/handshake/buffered_conn.go
+++ b/lib/handshake/buffered_conn.go
@@ -9,6 +9,7 @@ import (
 	"time"
 
 	"github.com/VictoriaMetrics/VictoriaMetrics/lib/encoding/zstd"
+	"github.com/VictoriaMetrics/VictoriaMetrics/lib/fasttime"
 )
 
 type bufferedWriter interface {
@@ -74,13 +75,13 @@ func (bc *BufferedConn) SetWriteDeadline(t time.Time) error {
 
 // Read reads up to len(p) from bc to p.
 func (bc *BufferedConn) Read(p []byte) (int, error) {
-	startTime := time.Now()
+	startTime := fasttime.UnixTimestamp()
 	if deadlineExceeded(bc.readDeadline, startTime) {
 		return 0, os.ErrDeadlineExceeded
 	}
 	n, err := bc.br.Read(p)
 	if err != nil && err != io.EOF {
-		err = fmt.Errorf("cannot read data in %.3f seconds: %w", time.Since(startTime).Seconds(), err)
+		err = fmt.Errorf("cannot read data in %d seconds: %w", fasttime.UnixTimestamp()-startTime, err)
 	}
 	return n, err
 }
@@ -89,22 +90,22 @@ func (bc *BufferedConn) Read(p []byte) (int, error) {
 //
 // Do not forget to call Flush if needed.
 func (bc *BufferedConn) Write(p []byte) (int, error) {
-	startTime := time.Now()
+	startTime := fasttime.UnixTimestamp()
 	if deadlineExceeded(bc.writeDeadline, startTime) {
 		return 0, os.ErrDeadlineExceeded
 	}
 	n, err := bc.bw.Write(p)
 	if err != nil {
-		err = fmt.Errorf("cannot write data in %.3f seconds: %w", time.Since(startTime).Seconds(), err)
+		err = fmt.Errorf("cannot write data in %d seconds: %w", fasttime.UnixTimestamp()-startTime, err)
 	}
 	return n, err
 }
 
-func deadlineExceeded(deadline, currentTime time.Time) bool {
+func deadlineExceeded(deadline time.Time, currentTimestamp uint64) bool {
 	if deadline.IsZero() {
 		return false
 	}
-	return currentTime.After(deadline)
+	return currentTimestamp > uint64(deadline.Unix())
 }
 
 // Close closes bc.