commit

country.go

@@ -25,17 +25,20 @@ type dbCountry struct {
 func (db *dbCountry) Init() {
 	db.allCountries = make(map[string]Country)
 }
+// function to return country with given id
+func (db *dbCountry) ReturnCountry(codeID  string) Country {
+	a := db.allCountries[codeID]
+	return a
+}
+
+
 
 // function to add to db
 func (db *dbCountry) Add(c Country) {
 	db.allCountries[c.Code] = c
 }
 
-// function to return country with given id
-func (db *dbCountry) ReturnCountry(codeID  string) Country {
-	a := db.allCountries[codeID]
-	return a
-}
+
 
 
 // function for returning all countries

country_api.go

@@ -95,12 +95,11 @@ func replyWithCountry(w http.ResponseWriter, dbc countryInterface, id string, li
 
 // handler for the country endpoint
 func handlerCountry(w http.ResponseWriter, r *http.Request) {
-	http.Header.Set(w.Header(), "content-type", "application/json")
+	http.Header.Add(w.Header(), "content-type", "application/json")
 	parts := strings.Split(r.URL.Path, "/")
 
 	addCountries(w)
-	//var limit string = parts[5]
-	if len(parts) > 7  {
+	if len(parts) > 6 || parts[5] != ""  {
 		status := 400
 		http.Error(w, http.StatusText(status), status)
 	} else {

handlers.go

@@ -10,7 +10,7 @@ import (
 func handlerSpecies(w http.ResponseWriter, r *http.Request) {
 	http.Header.Set(w.Header(), "content-type", "application/json")
 	parts := strings.Split(r.URL.Path, "/")
-	if len(parts) > 8 {
+	if len(parts) >  6 || parts[5] != "" {
 		status := 400
 		http.Error(w, http.StatusText(status), status)
 	} else {