mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 16:36:27 +01:00
2b7b3293c1
`TCP_USER_TIMEOUT` (since Linux 2.6.37) specifies the maximum amount of time that transmitted data may remain unacknowledged before TCP will forcibly close the connection and return `ETIMEDOUT` to the application. Setting a low TCP user timeout allows RPC connections quickly reroute around unavailable storage nodes during network interruptions.
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
package netutil
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/handshake"
|
|
"github.com/VictoriaMetrics/metrics"
|
|
)
|
|
|
|
func TestConnPoolStartStopSerial(t *testing.T) {
|
|
ms := metrics.NewSet()
|
|
testConnPoolStartStop(t, "foobar", ms)
|
|
ms.UnregisterAllMetrics()
|
|
}
|
|
|
|
func TestConnPoolStartStopConcurrent(t *testing.T) {
|
|
ms := metrics.NewSet()
|
|
concurrency := 5
|
|
ch := make(chan struct{})
|
|
for i := 0; i < concurrency; i++ {
|
|
name := fmt.Sprintf("foobar_%d", i)
|
|
go func() {
|
|
testConnPoolStartStop(t, name, ms)
|
|
ch <- struct{}{}
|
|
}()
|
|
}
|
|
tc := time.NewTimer(time.Second * 5)
|
|
for i := 0; i < concurrency; i++ {
|
|
select {
|
|
case <-tc.C:
|
|
t.Fatalf("timeout")
|
|
case <-ch:
|
|
}
|
|
}
|
|
tc.Stop()
|
|
ms.UnregisterAllMetrics()
|
|
}
|
|
|
|
func testConnPoolStartStop(t *testing.T, name string, ms *metrics.Set) {
|
|
dialTimeout := 5 * time.Second
|
|
compressLevel := 1
|
|
var cps []*ConnPool
|
|
for i := 0; i < 5; i++ {
|
|
addr := fmt.Sprintf("host-%d", i)
|
|
cp := NewConnPool(ms, name, addr, handshake.VMSelectClient, compressLevel, dialTimeout, 0)
|
|
cps = append(cps, cp)
|
|
}
|
|
for _, cp := range cps {
|
|
cp.MustStop()
|
|
// Make sure that Get works properly after MustStop()
|
|
c, err := cp.Get()
|
|
if err == nil {
|
|
t.Fatalf("expecting non-nil error after MustStop()")
|
|
}
|
|
if c != nil {
|
|
t.Fatalf("expecting nil conn after MustStop()")
|
|
}
|
|
}
|
|
}
|