added status code

hello-slide · Aug 8, 2021 · 6fc1fc1 · 6fc1fc1
1 parent 3cba232
commit 6fc1fc1
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/status.go b/status.go
@@ -24,6 +24,18 @@ func ErrorResponse(w http.ResponseWriter, statusCode int, err error) {
 		ErrorResponse(w, 1, err)
 	}
 
+	switch statusCode {
+	case 1:
+		ErrorStatus(w)
+		break
+	case 2:
+		w.WriteHeader(http.StatusUnauthorized)
+		break
+	default:
+		ErrorStatus(w)
+		break
+	}
+
 	w.Header().Set("Content-Type", "application/json")
 	w.Write(responseText)
 }