Fix unhandled null responses

sdslabs · Jan 1, 2025 · 5081abe · 5081abe
1 parent d8b47d8
commit 5081abe
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/helper/extracter.go b/helper/extracter.go
@@ -2,6 +2,7 @@ package helper
 
 import (
 	"encoding/json"
+	"errors"
 	"io"
 	"net/http"
 	"strconv"
@@ -25,6 +26,11 @@ func ExtractErrorCode(Error error) int {
 }
 
 func ExtractSuccessMessage(r *http.Response) string {
+
+	if r == nil {
+		log.ErrorLogger("Error message extractor failed: ", errors.New("response is nil"))
+		return "Kratos Connection Refused"
+	}
 
 	body, err := io.ReadAll(r.Body)
 
@@ -65,6 +71,11 @@ func ExtractSuccessMessage(r *http.Response) string {
 }
 
 func ExtractInfoMessage(r *http.Response) string {
+
+	if r == nil {
+		log.ErrorLogger("Error message extractor failed: ", errors.New("response is nil"))
+		return "Kratos Connection Refused"
+	}
 
 	body, err := io.ReadAll(r.Body)
 
@@ -106,6 +117,11 @@ func ExtractInfoMessage(r *http.Response) string {
 
 func ExtractErrorMessage(r *http.Response) string {
 
+	if r == nil {
+		log.ErrorLogger("Error message extractor failed: ", errors.New("response is nil"))
+		return "Kratos Connection Refused"
+	}
+
 	body, err := io.ReadAll(r.Body)
 
 	if err != nil {