Init

Author: ericm

go.mod

@@ -0,0 +1,5 @@
+module github.com/icabp-coderdojo-projects/go-rest-api
+
+go 1.13
+
+require github.com/go-chi/chi v4.0.3+incompatible

go.sum

@@ -0,0 +1,2 @@
+github.com/go-chi/chi v4.0.3+incompatible h1:gakN3pDJnzZN5jqFV2TEdF66rTfKeITyR8qu6ekICEY=
+github.com/go-chi/chi v4.0.3+incompatible/go.mod h1:eB3wogJHnLi3x/kFX2A+IbTBlXxmMeXJVKy9tTv1XzQ=

main.go

@@ -0,0 +1,72 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/go-chi/chi"
+	"github.com/go-chi/chi/middleware"
+)
+
+type Post struct {
+	Name string `json:"name"`
+	Body string `json:"body"`
+}
+
+type DB = map[int]Post
+
+var db DB
+
+func (post *Post) ToJSON() []byte {
+	js, _ := json.Marshal(post)
+	return js
+}
+
+func init() {
+	db = make(DB)
+	db[1] = Post{
+		Name: "Init",
+		Body: "body",
+	}
+	db[2000] = Post{
+		Name: "Init 2",
+		Body: "body 2",
+	}
+}
+
+func main() {
+	router := chi.NewRouter()
+	router.Use(
+		middleware.Logger,
+		middleware.RealIP,
+		middleware.Timeout(60*time.Second),
+	)
+
+	router.Get("/post/{id}", GetPost)
+
+	http.ListenAndServe(":3000", router)
+}
+
+func GetPost(w http.ResponseWriter, r *http.Request) {
+	id_str := chi.URLParam(r, "id")
+	var (
+		id  int
+		err error
+	)
+	if id, err = strconv.Atoi(id_str); err != nil {
+		http.Error(w, http.StatusText(500), 500)
+		return
+	}
+	if post, ok := db[id]; ok {
+		w.Header().Add("Content-Type", "application/json")
+		js := post.ToJSON()
+		code, _ := w.Write(js)
+		fmt.Println(code)
+		return
+	} else {
+		http.Error(w, http.StatusText(404), 404)
+	}
+}