REST API | MongoDB | Update user controller

Author: aditya43

118-MongoDB-REST/05-mongodb/01-update-user-controller/README.md

@@ -0,0 +1,61 @@
+# Install Mongo
+
+# Go get driver for mongo
+
+```
+go get gopkg.in/mgo.v2
+go get gopkg.in/mgo.v2/bson
+```
+
+# In this step:
+
+Don't run this code
+
+Just making updates - a several step process.
+
+We will need a mongo session to use in the CRUD methods.
+
+We need our UserController to have access to a mongo session.
+
+Let's add this to controllers/user.go
+
+```
+UserController struct { 
+ session *mgo.Session
+}
+```
+
+And now add this to controllers/user.go
+
+```
+func NewUserController(s *mgo.Session) *UserController { 
+ return &UserController{s}
+}
+```
+
+And now add this to main.go
+
+```
+func getSession() *mgo.Session {
+// Connect to our local mongo
+s, err := mgo.Dial("mongodb://localhost")
+
+// Check if connection error, is mongo running?
+if err != nil {
+panic(err)
+}
+return s
+}
+```
+
+and this
+
+```
+uc := controllers.NewUserController(getSession()) 
+```
+
+1. Enter this at the terminal
+
+```
+curl http://localhost:8080/user/1
+```

118-MongoDB-REST/05-mongodb/01-update-user-controller/controllers/user.go

@@ -0,0 +1,61 @@
+package controllers
+
+import (
+"encoding/json"
+"fmt"
+"net/http"
+
+"github.com/aditya43/118-MongoDB-REST/05-mongodb/01-update-user-controller/models"
+"github.com/julienschmidt/httprouter"
+"gopkg.in/mgo.v2"
+)
+
+// added session to our userController
+type UserController struct {
+session *mgo.Session
+}
+
+// added session to our userController
+func NewUserController(s *mgo.Session) *UserController {
+return &UserController{s}
+}
+
+func (uc UserController) GetUser(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+u := models.User{
+Name: "James Bond",
+Gender: "male",
+Age: 32,
+Id: p.ByName("id"),
+}
+
+uj, err := json.Marshal(u)
+if err != nil {
+fmt.Println(err)
+}
+w.Header().Set("Content-Type", "application/json")
+w.WriteHeader(http.StatusOK) // 200
+fmt.Fprintf(w, "%s\n", uj)
+}
+
+func (uc UserController) CreateUser(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
+u := models.User{}
+
+_ = json.NewDecoder(r.Body).Decode(&u)
+
+u.Id = "007"
+
+uj, err := json.Marshal(u)
+if err != nil {
+fmt.Println(err)
+}
+
+w.Header().Set("Content-Type", "application/json")
+w.WriteHeader(http.StatusCreated) // 201
+fmt.Fprintf(w, "%s\n", uj)
+}
+
+func (uc UserController) DeleteUser(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+// TODO: only write code to delete user
+w.WriteHeader(http.StatusOK) // 200
+fmt.Fprint(w, "Write code to delete user\n")
+}

118-MongoDB-REST/05-mongodb/01-update-user-controller/main.go

@@ -0,0 +1,31 @@
+package main
+
+import (
+"log"
+"net/http"
+
+"github.com/aditya43/118-MongoDB-REST/05-mongodb/01-update-user-controller/controllers"
+"github.com/julienschmidt/httprouter"
+"gopkg.in/mgo.v2"
+)
+
+func main() {
+r := httprouter.New()
+// Get a UserController instance
+uc := controllers.NewUserController(getSession())
+r.GET("/user/:id", uc.GetUser)
+r.POST("/user", uc.CreateUser)
+r.DELETE("/user/:id", uc.DeleteUser)
+log.Fatal(http.ListenAndServe("localhost:8080", r))
+}
+
+func getSession() *mgo.Session {
+// Connect to our local mongo
+s, err := mgo.Dial("mongodb://localhost")
+
+// Check if connection error, is mongo running?
+if err != nil {
+panic(err)
+}
+return s
+}

118-MongoDB-REST/05-mongodb/01-update-user-controller/models/user.go

@@ -0,0 +1,8 @@
+package models
+
+type User struct {
+Name string `json:"name"`
+Gender string `json:"gender"`
+Age int `json:"age"`
+Id string `json:"id"`
+}