check status code when polling finished

Signed-off-by: walkure <walkure@3pf.jp>
prometheus-community · Jan 23, 2023 · 60cd408 · 60cd408
1 parent 9955b76
commit 60cd408
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/client/main.go b/cmd/client/main.go
@@ -200,6 +200,11 @@ func (c *Coordinator) doPoll(client *http.Client) error {
 	}
 	defer resp.Body.Close()
 
+	if resp.StatusCode != http.StatusOK {
+		level.Warn(c.logger).Log("msg", "Unexpected response", "statusCode", resp.StatusCode)
+		return fmt.Errorf("unexpected response status=%d", resp.StatusCode)
+	}
+
 	request, err := http.ReadRequest(bufio.NewReader(resp.Body))
 	if err != nil {
 		level.Error(c.logger).Log("msg", "Error reading request:", "err", err)