Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

etcd-runner: Added client connection timeout flag

Browse Source
This commit is contained in:
sharat 2016-10-26 01:33:31 +05:30
parent ddc94aaf9e
commit 6e7e346c93
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools/functional-tester/etcd-runner/main.go
View File

@ -33,6 +33,8 @@ import (
"github.com/coreos/etcd/clientv3/concurrency"
)
var clientTimeout int64
func init() {
rand.Seed(time.Now().UTC().UnixNano())
}
@ -43,6 +45,7 @@ func main() {
endpointStr := flag.String("endpoints", "localhost:2379", "endpoints of etcd cluster")
mode := flag.String("mode", "watcher", "test mode (election, lock-racer, lease-renewer, watcher)")
round := flag.Int("rounds", 100, "number of rounds to run")
flag.Int64Var(&clientTimeout, "client-timeout", 60, "max timeout seconds for a client to get connection")
flag.Parse()
eps := strings.Split(*endpointStr, ",")
@ -405,7 +408,7 @@ func generateRandomKey(strlen uint) string {
func newClient(eps []string) *clientv3.Client {
c, err := clientv3.New(clientv3.Config{
Endpoints: eps,
DialTimeout: 5 * time.Second,
DialTimeout: time.Duration(clientTimeout) * time.Second,
})
if err != nil {
log.Fatal(err)