mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
dfc7cc7a62
Extend the timeout from 1s to defaultRequestTimeout 5s. The 1s may bring unwanted burden to the target member. If the member is busy at recovering, it has limited bandwidth for client requests. A short timeout at client side will retry quickly while keeping the on-going connections. Thus, etcd will queue lots of requests and connections and takes long time to clear them. This finally causes the timeout of member health check. This problem is a general one that how etcd handles amounts of requests at the same time in a good way. We don't plan to address it at current stage.
117 lines
2.4 KiB
Go
117 lines
2.4 KiB
Go
// Copyright 2015 CoreOS, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"net"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/coreos/etcd/Godeps/_workspace/src/golang.org/x/net/context"
|
|
"github.com/coreos/etcd/client"
|
|
)
|
|
|
|
type Stresser interface {
|
|
// Stress starts to stress the etcd cluster
|
|
Stress() error
|
|
// Cancel cancels the stress test on the etcd cluster
|
|
Cancel()
|
|
// Report reports the success and failure of the stress test
|
|
Report() (success int, failure int)
|
|
}
|
|
|
|
type stresser struct {
|
|
Endpoint string
|
|
|
|
KeySize int
|
|
KeySuffixRange int
|
|
|
|
N int
|
|
// TODO: not implemented
|
|
Interval time.Duration
|
|
|
|
mu sync.Mutex
|
|
failure int
|
|
success int
|
|
|
|
cancel func()
|
|
}
|
|
|
|
func (s *stresser) Stress() error {
|
|
cfg := client.Config{
|
|
Endpoints: []string{s.Endpoint},
|
|
Transport: &http.Transport{
|
|
Dial: (&net.Dialer{
|
|
Timeout: time.Second,
|
|
KeepAlive: 30 * time.Second,
|
|
}).Dial,
|
|
MaxIdleConnsPerHost: s.N,
|
|
},
|
|
}
|
|
c, err := client.New(cfg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
kv := client.NewKeysAPI(c)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
s.cancel = cancel
|
|
|
|
for i := 0; i < s.N; i++ {
|
|
go func() {
|
|
for {
|
|
setctx, setcancel := context.WithTimeout(ctx, client.DefaultRequestTimeout)
|
|
key := fmt.Sprintf("foo%d", rand.Intn(s.KeySuffixRange))
|
|
_, err := kv.Set(setctx, key, randStr(s.KeySize), nil)
|
|
setcancel()
|
|
if err == context.Canceled {
|
|
return
|
|
}
|
|
s.mu.Lock()
|
|
if err != nil {
|
|
s.failure++
|
|
} else {
|
|
s.success++
|
|
}
|
|
s.mu.Unlock()
|
|
}
|
|
}()
|
|
}
|
|
|
|
<-ctx.Done()
|
|
return nil
|
|
}
|
|
|
|
func (s *stresser) Cancel() {
|
|
s.cancel()
|
|
}
|
|
|
|
func (s *stresser) Report() (success int, failure int) {
|
|
s.mu.Lock()
|
|
defer s.mu.Unlock()
|
|
return s.success, s.failure
|
|
}
|
|
|
|
func randStr(size int) string {
|
|
data := make([]byte, size)
|
|
for i := 0; i < size; i++ {
|
|
data[i] = byte(int('a') + rand.Intn(26))
|
|
}
|
|
return string(data)
|
|
}
|