lab/go-api/main.go
Oliver Davies 075bbaac5e Return a JSON object if the response is successful
Signed-off-by: Oliver Davies <oliver@oliverdavies.uk>
2025-09-20 23:33:48 +01:00

46 lines
889 B
Go

package main
import (
"encoding/json"
"fmt"
"net/http"
"strconv"
)
func handler(w http.ResponseWriter, r *http.Request) {
fmt.Println("Running handler...")
code := getResponseCode(r)
w.WriteHeader(code)
w.Header().Set("Content-Type", "application/json")
if (code != 200) {
json.NewEncoder(w).Encode(map[string]any{})
} else {
json.NewEncoder(w).Encode(map[string]string{
"name": "Oliver Davies",
})
}
}
func getResponseCode(r *http.Request) int {
// If the `force-fail` header is set, get and return its value.
if failCode := r.Header.Get("force-fail"); failCode != "" {
fmt.Println("`force-fail` header set...")
if code, err := strconv.Atoi(failCode); err == nil {
fmt.Printf("Setting the response code to %d...", code)
return code
}
}
return http.StatusOK
}
func main() {
http.HandleFunc("/", handler)
http.ListenAndServe(":8080", nil)
}