carlota

controllers/library.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"net/http"
 	"strconv"
+	"strings"
 
 	"cloud.google.com/go/datastore"
 	"github.com/go-martini/martini"
@@ -24,30 +25,19 @@ func (lc LibraryController) GetByKey(params martini.Params, w http.ResponseWrite
 
 	id, err := strconv.Atoi(params["id"])
 
-	if err != nil {
-		fmt.Println(err)
-		w.WriteHeader(http.StatusBadRequest)
-		return
-	}
+	HandleError(err, w)
 
 	var book models.Book
 	key := datastore.IDKey("Book", int64(id), nil)
 
 	err = client.Get(ctx, key, &book)
 
-	if err != nil {
-		fmt.Println(err)
-		w.WriteHeader(http.StatusInternalServerError)
-		return
-	}
+	HandleError(err, w)
 
 	jsonStr, err := json.MarshalIndent(book, "", "  ")
 
-	if err != nil {
-		fmt.Println(err)
-		w.WriteHeader(http.StatusInternalServerError)
-		return
-	}
+	HandleError(err, w)
+
 
 	w.WriteHeader(http.StatusOK)
 	w.Write(jsonStr)
@@ -61,7 +51,7 @@ func (lc LibraryController) ListAll(r *http.Request, w http.ResponseWriter) {
 
 	var output []models.Book
 
-	it := client.Run(ctx, datastore.NewQuery("Book"))
+	it := client.Run(ctx, datastore.NewQuery("Book").Order("Id"))
 	for {
 		var b models.Book
 		_, err := it.Next(&b)
@@ -74,12 +64,160 @@ func (lc LibraryController) ListAll(r *http.Request, w http.ResponseWriter) {
 
 	jsonStr, err := json.MarshalIndent(output, "", "  ")
 
+	HandleError(err, w)
+
+	w.WriteHeader(http.StatusOK)
+	w.Write(jsonStr)
+}
+
+func HandleError (err error, w http.ResponseWriter) {
 	if err != nil {
 		fmt.Println(err)
-		w.WriteHeader(http.StatusInternalServerError)
+		w.WriteHeader(http.StatusBadRequest)
 		return
 	}
+}
+
+func (lc LibraryController) SearchByName(params martini.Params, w http.ResponseWriter) {
+	ctx := context.Background()
+	client, _ := datastore.NewClient(ctx, "acme-books")
+
+	defer client.Close()
+
+	name := strings.ReplaceAll(params["name"], "%20"," ")
+
+
+	fmt.Println(name)
+
+
+	//HandleError(err, w)
+
+	var output []models.Book
+
+	q := datastore.NewQuery("Book").Filter("Title=", name)
+
+	it := client.Run(ctx,q)
+
+	for {
+		var b models.Book
+		_, err := it.Next(&b)
+		fmt.Println(b)
+		if err == iterator.Done {
+			fmt.Println(err)
+			break
+		}
+		output = append(output, b)
+	}
+
+	jsonStr, err := json.MarshalIndent(output, "", "  ")
+
+	HandleError(err, w)
 
 	w.WriteHeader(http.StatusOK)
 	w.Write(jsonStr)
 }
+
+func (lc LibraryController) Borrow (params martini.Params, w http.ResponseWriter) {
+	ctx := context.Background()
+	client, _ := datastore.NewClient(ctx, "acme-books")
+
+	defer client.Close()
+
+	id, err := strconv.Atoi(params["id"])
+
+	HandleError(err, w)
+
+	var book models.Book
+	key := datastore.IDKey("Book", int64(id), nil)
+
+	err = client.Get(ctx, key, &book)
+
+	switch {
+	case err != nil:
+		w.WriteHeader(http.StatusBadRequest)
+		w.Write([]byte("invalid id"))
+	case book.Borrowed:
+		w.WriteHeader(http.StatusBadRequest)
+		w.Write([]byte("already borrowed"))
+	default:
+		book.Borrowed = true
+		_, err = client.Put(ctx, key, &book)
+		HandleError(err, w)
+
+		jsonStr, err := json.MarshalIndent(book, "", "  ")
+
+		HandleError(err, w)
+
+		w.WriteHeader(http.StatusOK)
+		w.Write(jsonStr)
+	}
+
+
+}
+
+func (lc LibraryController) Return (params martini.Params, w http.ResponseWriter) {
+	ctx := context.Background()
+	client, _ := datastore.NewClient(ctx, "acme-books")
+
+	defer client.Close()
+
+	id, err := strconv.Atoi(params["id"])
+
+	HandleError(err, w)
+
+	var book models.Book
+	key := datastore.IDKey("Book", int64(id), nil)
+
+	err = client.Get(ctx, key, &book)
+
+	switch {
+	case err != nil:
+		w.WriteHeader(http.StatusBadRequest)
+		w.Write([]byte("invalid id"))
+		return
+	case !book.Borrowed:
+		w.WriteHeader(http.StatusBadRequest)
+		w.Write([]byte("already returned"))
+	default:
+		book.Borrowed = false
+		_, err = client.Put(ctx, key, &book)
+		HandleError(err, w)
+
+		jsonStr, err := json.MarshalIndent(book, "", "  ")
+
+		HandleError(err, w)
+
+		w.WriteHeader(http.StatusOK)
+		w.Write(jsonStr)
+	}
+}
+
+func (lc LibraryController) CreateBook (r *http.Request, w http.ResponseWriter) {
+	ctx := context.Background()
+	client, _ := datastore.NewClient(ctx, "acme-books")
+
+	defer client.Close()
+
+	var book models.Book
+
+
+	err := json.NewDecoder(r.Body).Decode(&book)
+	HandleError(err, w)
+
+	key := datastore.IncompleteKey("Book", nil)
+	fmt.Println(key)
+
+	newKey, err := client.Put(ctx, key, &book)
+	book.Id = newKey.ID
+	fmt.Println(book)
+	HandleError(err, w)
+
+
+	jsonStr, err := json.MarshalIndent(book, "", "  ")
+
+	HandleError(err, w)
+
+	w.WriteHeader(http.StatusOK)
+	w.Write(jsonStr)
+
+}

main.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log"
 	"os"
+	"strings"
 
 	"cloud.google.com/go/datastore"
 
@@ -44,13 +45,15 @@ func bootstrapBooks() {
 	books := []models.Book{
 		{Id: 1, Author: "George Orwell", Title: "1984", Borrowed: false},
 		{Id: 2, Author: "George Orwell", Title: "Animal Farm", Borrowed: false},
-		{Id: 3, Author: "Robert Jordan", Title: "Eye of the world", Borrowed: false},
-		{Id: 4, Author: "Various", Title: "Collins Dictionary", Borrowed: false},
+		{Id: 3, Author: "Variorewgr4g4gus", Title: "Collins Dictionary", Borrowed: false},
+		{Id: 4, Author: "Robert Jordan", Title: "Eye of the world", Borrowed: false},
 	}
 
 	var keys []*datastore.Key
 
-	for _, book := range books {
+	for index, book := range books {
+		books[index].Title = strings.Title(book.Title)
+		books[index].Author = strings.Title(book.Author)
 		keys = append(keys, datastore.IDKey("Book", book.Id, nil))
 	}
 

server/router.go

@@ -13,6 +13,10 @@ func NewRouter() *martini.ClassicMartini {
 
 	router.Get("/books", libraryController.ListAll)
 	router.Get("/books/:id", libraryController.GetByKey)
+	router.Get("/searchByName/:name", libraryController.SearchByName)
+	router.Put("/:id/borrow", libraryController.Borrow)
+	router.Put("/:id/return", libraryController.Return)
+	router.Post("/book", libraryController.CreateBook)
 
 	return router
 }