// GetKey returns an http.Handler that can get a key registered by Gorilla mux
// as "key" in the path. It gets the value of the key from db
func GetKey(db storage.DB) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
key := r.URL.Query().Get("key")
if key == "" {
http.Error(w, "missing key name in query string", http.StatusBadRequest)
return
}
val, err := db.Get(key)
if err == storage.ErrNotFound {
http.Error(w, "not found", http.StatusNotFound)
return
} else if err != nil {
http.Error(w, fmt.Sprintf("error getting value from database: %s", err), http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusOK)
w.Write(val)
})
}
// PutKey returns an http.Handler that can set a value for the key registered by Gorilla
// mux as "key" in the path. It expects the value to be in the body of the PUT request
func PutKey(db storage.DB) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
key := r.URL.Query().Get("key")
if key == "" {
http.Error(w, "missing key name in path", http.StatusBadRequest)
return
}
defer r.Body.Close()
val, err := ioutil.ReadAll(r.Body)
if err != nil {
http.Error(w, "error reading PUT body", http.StatusBadRequest)
return
}
if err := db.Set(key, val); err != nil {
http.Error(w, "error setting value in DB", http.StatusInternalServerError)
return
}
w.WriteHeader(http.StatusOK)
})
}
