Modifying service

Author: aperezg

cmd/gopherapi/main.go

@@ -6,6 +6,8 @@ import (
 	"log"
 	"net/http"
 
+	"github.com/friendsofgo/gopherapi/pkg/modifying"
+
 	"github.com/friendsofgo/gopherapi/pkg/adding"
 	"github.com/friendsofgo/gopherapi/pkg/fetching"
 
@@ -27,10 +29,11 @@ func main() {
 	}
 
 	repo := inmem.NewRepository(gophers)
-	fetching := fetching.NewService(repo)
-	adding := adding.NewService(repo)
+	fS := fetching.NewService(repo)
+	aS := adding.NewService(repo)
+	mS := modifying.NewService(repo)
 
-	s := server.New(fetching, adding)
+	s := server.New(fS, aS, mS)
 
 	fmt.Println("The gopher server is on tap now: http://localhost:8080")
 	log.Fatal(http.ListenAndServe(":8080", s.Router()))

pkg/modifying/service.go

@@ -0,0 +1,25 @@
+package modifying
+
+import (
+	gopher "github.com/friendsofgo/gopherapi/pkg"
+)
+
+// Service provides modifying operations.
+type Service interface {
+	ModifyGopher(ID, name, image string, age int) error
+}
+
+type service struct {
+	repository gopher.Repository
+}
+
+// NewService creates a modifying service with the necessary dependencies
+func NewService(repository gopher.Repository) Service {
+	return &service{repository}
+}
+
+// ModifyGopher modify a gopher data
+func (s *service) ModifyGopher(ID, name, image string, age int) error {
+	g := gopher.New(ID, name, image, age)
+	return s.repository.UpdateGopher(ID, *g)
+}

pkg/server/api.go

@@ -6,14 +6,16 @@ import (
 
 	"github.com/friendsofgo/gopherapi/pkg/adding"
 	"github.com/friendsofgo/gopherapi/pkg/fetching"
+	"github.com/friendsofgo/gopherapi/pkg/modifying"
 
 	"github.com/gorilla/mux"
 )
 
 type api struct {
-	router   http.Handler
-	fetching fetching.Service
-	adding   adding.Service
+	router    http.Handler
+	fetching  fetching.Service
+	adding    adding.Service
+	modifying modifying.Service
 }
 
 // Server representation of gopher server
@@ -22,16 +24,18 @@ type Server interface {
 	FetchGophers(w http.ResponseWriter, r *http.Request)
 	FetchGopher(w http.ResponseWriter, r *http.Request)
 	AddGopher(w http.ResponseWriter, r *http.Request)
+	ModifyGopher(w http.ResponseWriter, r *http.Request)
 }
 
 // New initialize the server
-func New(fS fetching.Service, aS adding.Service) Server {
-	a := &api{fetching: fS, adding: aS}
+func New(fS fetching.Service, aS adding.Service, mS modifying.Service) Server {
+	a := &api{fetching: fS, adding: aS, modifying: mS}
 
 	r := mux.NewRouter()
 	r.HandleFunc("/gophers", a.FetchGophers).Methods(http.MethodGet)
 	r.HandleFunc("/gophers/{ID:[a-zA-Z0-9_]+}", a.FetchGopher).Methods(http.MethodGet)
 	r.HandleFunc("/gophers", a.AddGopher).Methods(http.MethodPost)
+	r.HandleFunc("/gophers/{ID:[a-zA-Z0-9_]+}", a.ModifyGopher).Methods(http.MethodPut)
 
 	a.router = r
 	return a
@@ -77,6 +81,7 @@ func (a *api) AddGopher(w http.ResponseWriter, r *http.Request) {
 	var g addGopherRequest
 	err := decoder.Decode(&g)
 
+	w.Header().Set("Content-Type", "application/json")
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
 		json.NewEncoder(w).Encode("Error unmarshalling request body")
@@ -91,3 +96,32 @@ func (a *api) AddGopher(w http.ResponseWriter, r *http.Request) {
 
 	w.WriteHeader(http.StatusCreated)
 }
+
+type modifyGopherRequest struct {
+	Name  string `json:"name"`
+	Image string `json:"image"`
+	Age   int    `json:"age"`
+}
+
+// ModifyGopher modify gopher data
+func (a *api) ModifyGopher(w http.ResponseWriter, r *http.Request) {
+	decoder := json.NewDecoder(r.Body)
+
+	var g addGopherRequest
+	err := decoder.Decode(&g)
+
+	w.Header().Set("Content-Type", "application/json")
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode("Error unmarshalling request body")
+		return
+	}
+	vars := mux.Vars(r)
+	if err := a.modifying.ModifyGopher(vars["ID"], g.Name, g.Image, g.Age); err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		json.NewEncoder(w).Encode("Can't modify a gopher")
+		return
+	}
+
+	w.WriteHeader(http.StatusNoContent)
+}

pkg/server/api_test.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/friendsofgo/gopherapi/pkg/adding"
 	"github.com/friendsofgo/gopherapi/pkg/fetching"
+	"github.com/friendsofgo/gopherapi/pkg/modifying"
 
 	sample "github.com/friendsofgo/gopherapi/cmd/sample-data"
 	gopher "github.com/friendsofgo/gopherapi/pkg"
@@ -140,5 +141,7 @@ func buildServer() Server {
 	repo := inmem.NewRepository(sample.Gophers)
 	fetching := fetching.NewService(repo)
 	adding := adding.NewService(repo)
-	return New(fetching, adding)
+	modifying := modifying.NewService(repo)
+
+	return New(fetching, adding, modifying)
 }