version: add error handling

Signed-off-by: hwdef <[email protected]>
dragonflyoss · Oct 16, 2019 · 4d50854 · 4d50854
1 parent 0d2bbbe
commit 4d50854
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/version/version.go b/version/version.go
@@ -123,9 +123,10 @@ func Handler(w http.ResponseWriter, r *http.Request) {
 	data, err := json.Marshal(DFVersion)
 	if err != nil {
 		http.Error(w, fmt.Sprintf("error encoding JSON: %s", err), http.StatusInternalServerError)
+	} else if _, err := w.Write(data); err != nil {
+		http.Error(w, fmt.Sprintf("error writing the data to the connection: %s", err), http.StatusInternalServerError)
 	} else {
 		w.WriteHeader(http.StatusOK)
-		w.Write(data)
 	}
 }