Răsfoiți Sursa

set json response on failover handers

Abhishek Kondur 1 an în urmă
părinte
comite
2633c10ba8
2 a modificat fișierele cu 13 adăugiri și 2 ștergeri
  1. 11 0
      logic/errors.go
  2. 2 2
      pro/controllers/failover.go

+ 11 - 0
logic/errors.go

@@ -44,6 +44,17 @@ func ReturnSuccessResponse(response http.ResponseWriter, request *http.Request,
 	json.NewEncoder(response).Encode(httpResponse)
 }
 
+// ReturnSuccessResponseWithJson - processes message and adds header
+func ReturnSuccessResponseWithJson(response http.ResponseWriter, request *http.Request, res interface{}, message string) {
+	var httpResponse models.SuccessResponse
+	httpResponse.Code = http.StatusOK
+	httpResponse.Response = res
+	httpResponse.Message = message
+	response.Header().Set("Content-Type", "application/json")
+	response.WriteHeader(http.StatusOK)
+	json.NewEncoder(response).Encode(httpResponse)
+}
+
 // ReturnErrorResponse - processes error and adds header
 func ReturnErrorResponse(response http.ResponseWriter, request *http.Request, errorMessage models.ErrorResponse) {
 	httpResponse := &models.ErrorResponse{Code: errorMessage.Code, Message: errorMessage.Message}

+ 2 - 2
pro/controllers/failover.go

@@ -57,7 +57,7 @@ func createfailOver(w http.ResponseWriter, r *http.Request) {
 	}
 	go mq.PublishPeerUpdate()
 	w.Header().Set("Content-Type", "application/json")
-	logic.ReturnSuccessResponse(w, r, "created failover successfully")
+	logic.ReturnSuccessResponseWithJson(w, r, node, "created failover successfully")
 }
 
 // swagger:route DELETE /api/v1/node/failover node deletefailOver
@@ -94,7 +94,7 @@ func deletefailOver(w http.ResponseWriter, r *http.Request) {
 		mq.PublishPeerUpdate()
 	}()
 	w.Header().Set("Content-Type", "application/json")
-	logic.ReturnSuccessResponse(w, r, "relayed successfully")
+	logic.ReturnSuccessResponseWithJson(w, r, node, "created failover successfully")
 }
 
 // swagger:route POST /api/node/{nodeid}/failOverME node failOver_me