Implement setResponseCookies.

Co-authored-by: Hector Hurtado <hector.hurtado@bbva.com>
2019-10-21 10:01:34 +02:00
parent ffc737d1b7
commit 0f8d0266b9
2 changed files with 106 additions and 1 deletions
@@ -147,7 +147,12 @@ func setResponseStatus(w http.ResponseWriter, r *http.Request, h *model.Handler)

 func setResponseHeaders(w http.ResponseWriter, r *http.Request, h *model.Handler) {
 	name := mux.Vars(r)["name"]
-	vb, _ := ioutil.ReadAll(r.Body)
+	vb, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
 	hds := h.Writer.Header()
 	if _, ok := hds[name]; ok {
 		hds[name] = append(hds[name], string(vb))
@@ -155,3 +160,15 @@ func setResponseHeaders(w http.ResponseWriter, r *http.Request, h *model.Handler
 		hds[name] = []string{string(vb)}
 	}
 }
+
+func setResponseCookies(w http.ResponseWriter, r *http.Request, h *model.Handler) {
+	name := mux.Vars(r)["name"]
+	vb, err := ioutil.ReadAll(r.Body)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		return
+	}
+
+	c := &http.Cookie{Name: name, Value: string(vb)}
+	http.SetCookie(h.Writer, c)
+}