mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-11-21 14:44:00 +00:00
09c388a8e4
While fasthttp.Client uses less CPU and RAM when scraping targets with small responses (up to 10K metrics), it doesn't work well when scraping targets with big responses such as kube-state-metrics. In this case it could use big amounts of additional memory comparing to net/http.Client, since fasthttp.Client reads the full response in memory and then tries re-using the large buffer for further scrapes. Additionally, fasthttp.Client-based scraping had various issues with proxying, redirects and scrape timeouts like the following ones: - https://github.com/VictoriaMetrics/VictoriaMetrics/issues/1945 - https://github.com/VictoriaMetrics/VictoriaMetrics/issues/5425 - https://github.com/VictoriaMetrics/VictoriaMetrics/issues/2794 - https://github.com/VictoriaMetrics/VictoriaMetrics/issues/1017 This should help reducing memory usage for the case when target returns big response and this response is scraped by fasthttp.Client at first before switching to stream parsing mode for subsequent scrapes. Now the switch to stream parsing mode is performed on the first scrape after reading the response body in memory and noticing that its size exceeds the value passed to -promscrape.minResponseSizeForStreamParse command-line flag. Updates https://github.com/VictoriaMetrics/VictoriaMetrics/issues/5567 Overrides https://github.com/VictoriaMetrics/VictoriaMetrics/pull/4931
135 lines
2.7 KiB
Go
135 lines
2.7 KiB
Go
package promscrape
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/netutil"
|
|
"github.com/VictoriaMetrics/metrics"
|
|
)
|
|
|
|
func statStdDial(ctx context.Context, _, addr string) (net.Conn, error) {
|
|
d := getStdDialer()
|
|
network := netutil.GetTCPNetwork()
|
|
conn, err := d.DialContext(ctx, network, addr)
|
|
dialsTotal.Inc()
|
|
if err != nil {
|
|
dialErrors.Inc()
|
|
if !netutil.TCP6Enabled() && !isTCPv4Addr(addr) {
|
|
err = fmt.Errorf("%w; try -enableTCP6 command-line flag if you scrape ipv6 addresses", err)
|
|
}
|
|
return nil, err
|
|
}
|
|
conns.Inc()
|
|
sc := &statConn{
|
|
Conn: conn,
|
|
}
|
|
return sc, nil
|
|
}
|
|
|
|
func getStdDialer() *net.Dialer {
|
|
stdDialerOnce.Do(func() {
|
|
stdDialer = &net.Dialer{
|
|
Timeout: 30 * time.Second,
|
|
KeepAlive: 30 * time.Second,
|
|
DualStack: netutil.TCP6Enabled(),
|
|
}
|
|
})
|
|
return stdDialer
|
|
}
|
|
|
|
var (
|
|
stdDialer *net.Dialer
|
|
stdDialerOnce sync.Once
|
|
)
|
|
|
|
var (
|
|
dialsTotal = metrics.NewCounter(`vm_promscrape_dials_total`)
|
|
dialErrors = metrics.NewCounter(`vm_promscrape_dial_errors_total`)
|
|
conns = metrics.NewCounter(`vm_promscrape_conns`)
|
|
)
|
|
|
|
type statConn struct {
|
|
closed uint64
|
|
net.Conn
|
|
}
|
|
|
|
func (sc *statConn) Read(p []byte) (int, error) {
|
|
n, err := sc.Conn.Read(p)
|
|
connReadsTotal.Inc()
|
|
if err != nil {
|
|
connReadErrors.Inc()
|
|
}
|
|
connBytesRead.Add(n)
|
|
return n, err
|
|
}
|
|
|
|
func (sc *statConn) Write(p []byte) (int, error) {
|
|
n, err := sc.Conn.Write(p)
|
|
connWritesTotal.Inc()
|
|
if err != nil {
|
|
connWriteErrors.Inc()
|
|
}
|
|
connBytesWritten.Add(n)
|
|
return n, err
|
|
}
|
|
|
|
func (sc *statConn) Close() error {
|
|
err := sc.Conn.Close()
|
|
if atomic.AddUint64(&sc.closed, 1) == 1 {
|
|
conns.Dec()
|
|
}
|
|
return err
|
|
}
|
|
|
|
var (
|
|
connReadsTotal = metrics.NewCounter(`vm_promscrape_conn_reads_total`)
|
|
connWritesTotal = metrics.NewCounter(`vm_promscrape_conn_writes_total`)
|
|
connReadErrors = metrics.NewCounter(`vm_promscrape_conn_read_errors_total`)
|
|
connWriteErrors = metrics.NewCounter(`vm_promscrape_conn_write_errors_total`)
|
|
connBytesRead = metrics.NewCounter(`vm_promscrape_conn_bytes_read_total`)
|
|
connBytesWritten = metrics.NewCounter(`vm_promscrape_conn_bytes_written_total`)
|
|
)
|
|
|
|
func isTCPv4Addr(addr string) bool {
|
|
s := addr
|
|
for i := 0; i < 3; i++ {
|
|
n := strings.IndexByte(s, '.')
|
|
if n < 0 {
|
|
return false
|
|
}
|
|
if !isUint8NumString(s[:n]) {
|
|
return false
|
|
}
|
|
s = s[n+1:]
|
|
}
|
|
n := strings.IndexByte(s, ':')
|
|
if n < 0 {
|
|
return false
|
|
}
|
|
if !isUint8NumString(s[:n]) {
|
|
return false
|
|
}
|
|
s = s[n+1:]
|
|
|
|
// Verify TCP port
|
|
n, err := strconv.Atoi(s)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return n >= 0 && n < (1<<16)
|
|
}
|
|
|
|
func isUint8NumString(s string) bool {
|
|
n, err := strconv.Atoi(s)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return n >= 0 && n < (1<<8)
|
|
}
|