goweb - gin - GET, POST, DELETE, LIST

Author: KenanBek

goweb/gowebgin/main.go

@@ -1,14 +1,45 @@
 package main
 
-import "github.com/gin-gonic/gin"
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+)
 
 func main() {
+	mdb := NewMemDb()
+
 	r := gin.Default()
 	r.GET("/ping", func(c *gin.Context) {
-		c.JSON(200, gin.H{
+		c.JSON(http.StatusOK, gin.H{
 			"message": "pong",
 		})
 	})
 
+	r.GET("/users", func(c *gin.Context) {
+		c.JSON(http.StatusOK, mdb.ListUsers())
+	})
+
+	r.POST("/users", func(c *gin.Context) {
+		var user User
+		err := c.BindJSON(&user)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, err.Error())
+		} else {
+			c.JSON(http.StatusCreated, mdb.AddUser(&user))
+		}
+	})
+
+	r.GET("/users/:username", func(c *gin.Context) {
+		username := c.Param("username")
+		c.JSON(http.StatusOK, mdb.GetUser(username))
+	})
+
+	r.DELETE("/users/:username", func(c *gin.Context) {
+		username := c.Param("username")
+		mdb.DeleteUser(username)
+		c.JSON(http.StatusOK, gin.H{})
+	})
+
 	r.Run() // listen and serve on 0.0.0.0:8080 (for windows "localhost:8080")
 }

goweb/gowebgin/model.go

@@ -0,0 +1,70 @@
+package main
+
+import "log"
+
+// User is a in-memory user model.
+type User struct {
+	Username  string `json:"username"`
+	FirstName string `json:"first_name"`
+	LastName  string `json:"last_name"`
+}
+
+// NewMemDb creates and returns a new instance of MemDb.
+func NewMemDb() *MemDb {
+	return &MemDb{
+		users:    []*User{},
+		kvsUsers: map[string]int{},
+	}
+}
+
+// MemDb is a memory database.
+type MemDb struct {
+	users    []*User
+	kvsUsers map[string]int
+}
+
+// AddUser stores user in in-memory database.
+func (db *MemDb) AddUser(u *User) *User {
+	log.Println("AddUser", u)
+
+	ind, ok := db.kvsUsers[u.Username]
+	if !ok {
+		db.kvsUsers[u.Username] = len(db.users)
+		db.users = append(db.users, u)
+	} else {
+		db.users[ind] = u
+	}
+
+	return db.users[ind]
+}
+
+// GetUser returns user in in-memory database.
+func (db *MemDb) GetUser(username string) *User {
+	log.Println("GetUser", username)
+
+	ind, ok := db.kvsUsers[username]
+	if ok {
+		return db.users[ind]
+	}
+
+	return nil
+}
+
+// DeleteUser deletes user in in-memory database.
+func (db *MemDb) DeleteUser(username string) {
+	log.Println("DeleteUser", username)
+
+	if ind, ok := db.kvsUsers[username]; ok {
+		db.users[ind] = db.users[len(db.users)-1]
+		db.users = db.users[:len(db.users)-1]
+
+		delete(db.kvsUsers, username)
+	}
+}
+
+// ListUsers returns all users in in-memory database.
+func (db *MemDb) ListUsers() []*User {
+	log.Println("ListUsers")
+
+	return db.users
+}