handle fatal errors a bit more gracefully

rogerwelin · May 17, 2020 · 3661474 · 3661474
1 parent 4d20a04
commit 3661474
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/pkg/client/load.go b/pkg/client/load.go
@@ -35,26 +35,26 @@ func (c *Cassowary) runLoadTest(outPutChan chan<- durationMetrics, workerChan ch
 		if c.FileMode {
 			request, err = http.NewRequest("GET", c.BaseURL+URLitem, nil)
 			if err != nil {
-				panic(err)
+				log.Fatalf("%v", err)
 			}
 		} else {
 			switch c.HTTPMethod {
 			case "POST":
 				request, err = http.NewRequest("POST", c.BaseURL, bytes.NewBuffer(c.Data))
 				request.Header.Set("Content-Type", "application/json")
 				if err != nil {
-					panic(err)
+					log.Fatalf("%v", err)
 				}
 			case "PUT":
 				request, err = http.NewRequest("PUT", c.BaseURL, bytes.NewBuffer(c.Data))
 				request.Header.Set("Content-Type", "application/json")
 				if err != nil {
-					panic(err)
+					log.Fatalf("%v", err)
 				}
 			default:
 				request, err = http.NewRequest("GET", c.BaseURL, nil)
 				if err != nil {
-					panic(err)
+					log.Fatalf("%v", err)
 				}
 			}
 		}
@@ -90,7 +90,7 @@ func (c *Cassowary) runLoadTest(outPutChan chan<- durationMetrics, workerChan ch
 		request = request.WithContext(httptrace.WithClientTrace(context.Background(), trace))
 		resp, err := c.Client.Do(request)
 		if err != nil {
-			panic(err)
+			log.Fatalf("%v", err)
 		}
 		if resp != nil {
 			_, err = io.Copy(ioutil.Discard, resp.Body)