Improve 502 error handling

casdoor · Sep 9, 2023 · 26865e8 · 26865e8
1 parent af4ffa4
commit 26865e8
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/casdoorsdk/base.go b/casdoorsdk/base.go
@@ -174,12 +174,16 @@ func (c *Client) DoPostBytesRaw(url string, contentType string, body io.Reader)
 		}
 	}(resp.Body)
 
-	respByte, err := ioutil.ReadAll(resp.Body)
+	respBytes, err := ioutil.ReadAll(resp.Body)
 	if err != nil {
 		return nil, err
 	}
 
-	return respByte, nil
+	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusForbidden {
+		return nil, fmt.Errorf("%s", string(respBytes))
+	}
+
+	return respBytes, nil
 }
 
 // doGetBytesRawWithoutCheck is a general function to get response from param url through HTTP Get method without checking response status
@@ -207,6 +211,10 @@ func (c *Client) doGetBytesRawWithoutCheck(url string) ([]byte, error) {
 		return nil, err
 	}
 
+	if resp.StatusCode != http.StatusOK && resp.StatusCode != http.StatusForbidden {
+		return nil, fmt.Errorf("%s", string(respBytes))
+	}
+
 	return respBytes, nil
 }