add album API function handler

Hroman9n · Dec 15, 2023 · 8a298f0 · 8a298f0
1 parent d1ad3ef
commit 8a298f0
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -37,6 +37,23 @@ func headers(w http.ResponseWriter, req *http.Request) {
 	}
 }
 
+func apiAlbum(w http.ResponseWriter, req *http.Request) {
+	rId := req.URL.Path
+	var retData = []album{}
+	if len(rId) > 0 {
+		for _, a := range albums {
+			if a.ID == rId {
+				retData = append(retData, a)
+			}
+		}
+	} else {
+		retData = albums
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	json.NewEncoder(w).Encode(retData)
+}
+
 func main() {
 	http.HandleFunc("/", root)
 	http.HandleFunc("/hello", hello)