mirror of https://github.com/jackc/pgx.git
TestStressConnPool now runs for X iterations
...instead of T time. Also run in parallel.v3-numeric-wip
parent
91dea95b68
commit
eb484e1368
|
@ -3,11 +3,12 @@ package pgx_test
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"golang.org/x/net/context"
|
|
||||||
"math/rand"
|
"math/rand"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/net/context"
|
||||||
|
|
||||||
"github.com/jackc/fake"
|
"github.com/jackc/fake"
|
||||||
"github.com/jackc/pgx"
|
"github.com/jackc/pgx"
|
||||||
)
|
)
|
||||||
|
@ -23,6 +24,8 @@ type queryRower interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestStressConnPool(t *testing.T) {
|
func TestStressConnPool(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
maxConnections := 8
|
maxConnections := 8
|
||||||
pool := createConnPool(t, maxConnections)
|
pool := createConnPool(t, maxConnections)
|
||||||
defer pool.Close()
|
defer pool.Close()
|
||||||
|
@ -49,11 +52,12 @@ func TestStressConnPool(t *testing.T) {
|
||||||
{"canceledExecContext", canceledExecContext},
|
{"canceledExecContext", canceledExecContext},
|
||||||
}
|
}
|
||||||
|
|
||||||
var timer *time.Timer
|
var actionCount int
|
||||||
|
|
||||||
if testing.Short() {
|
if testing.Short() {
|
||||||
timer = time.NewTimer(5 * time.Second)
|
actionCount = 1000
|
||||||
} else {
|
} else {
|
||||||
timer = time.NewTimer(60 * time.Second)
|
actionCount = 10000
|
||||||
}
|
}
|
||||||
workerCount := 16
|
workerCount := 16
|
||||||
|
|
||||||
|
@ -77,11 +81,8 @@ func TestStressConnPool(t *testing.T) {
|
||||||
go work()
|
go work()
|
||||||
}
|
}
|
||||||
|
|
||||||
var stop bool
|
for i := 0; i < actionCount; i++ {
|
||||||
for i := 0; !stop; i++ {
|
|
||||||
select {
|
select {
|
||||||
case <-timer.C:
|
|
||||||
stop = true
|
|
||||||
case workChan <- i:
|
case workChan <- i:
|
||||||
case err := <-errChan:
|
case err := <-errChan:
|
||||||
close(workChan)
|
close(workChan)
|
||||||
|
|
Loading…
Reference in New Issue