set time out

perf-client.go

@@ -68,11 +68,24 @@ func (c *clientPerfReader) Read(p []byte) (n int, err error) {
 
 var _ io.Reader = &clientPerfReader{}
 
+const (
+	// MaxClientPerfTimeout for max time out for client perf
+	MaxClientPerfTimeout = time.Second * 30
+	// MinClientPerfTimeout for min time out for client perf
+	MinClientPerfTimeout = time.Second * 5
+)
+
 // ClientPerf - perform net from client to MinIO servers
-func (adm *AdminClient) ClientPerf(ctx context.Context) (result ClientPerfResult, err error) {
-	defer func() {
-		result.Endpoint = adm.endpointURL.String()
-	}()
+func (adm *AdminClient) ClientPerf(ctx context.Context, dur time.Duration) (result ClientPerfResult, err error) {
+	if dur > MaxClientPerfTimeout {
+		dur = MaxClientPerfTimeout
+	}
+	if dur < MinClientPerfTimeout {
+		dur = MinClientPerfTimeout
+	}
+	ctx, cancel := context.WithTimeout(ctx, dur)
+	defer cancel()
+	result.Endpoint = adm.endpointURL.String()
 	queryVals := make(url.Values)
 	reader := &clientPerfReader{}
 	reader.Start()