chore(golang): switch to go-chi router

Closes #12

Part of #9

Author: php-coder

examples/go/app.go

@@ -3,12 +3,14 @@ package main
 import "fmt"
 import "net/http"
 import "os"
+import "github.com/go-chi/chi"
 
 func main() {
-	registerRoutes()
+	r := chi.NewRouter()
+	registerRoutes(r)
 
 	fmt.Println("Listen on 3000")
-	if err := http.ListenAndServe(":3000", nil); err != nil {
+	if err := http.ListenAndServe(":3000", r); err != nil {
 		fmt.Fprintf(os.Stderr, "ListenAndServe failed: %v", err)
 		os.Exit(1)
 	}

examples/go/routes.go

@@ -1,22 +1,35 @@
 package main
 
 import "net/http"
+import "github.com/go-chi/chi"
 
-func registerRoutes() {
+func registerRoutes(r chi.Router) {
 
-	http.HandleFunc("/v1/categories/count", func(w http.ResponseWriter, _ *http.Request) {
+	r.Get("/v1/categories/count", func(w http.ResponseWriter, _ *http.Request) {
 		w.Write([]byte("TODO"))
 	})
 
-	http.HandleFunc("/v1/collections/:collectionId/categories/count", func(w http.ResponseWriter, _ *http.Request) {
+	r.Get("/v1/collections/{collectionId}/categories/count", func(w http.ResponseWriter, _ *http.Request) {
 		w.Write([]byte("TODO"))
 	})
 
-	http.HandleFunc("/v1/categories", func(w http.ResponseWriter, _ *http.Request) {
+	r.Get("/v1/categories", func(w http.ResponseWriter, _ *http.Request) {
 		w.Write([]byte("TODO"))
 	})
 
-	http.HandleFunc("/v1/categories/:categoryId", func(w http.ResponseWriter, _ *http.Request) {
+	r.Post("/v1/categories", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+
+	r.Get("/v1/categories/{categoryId}", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+
+	r.Put("/v1/categories/{categoryId}", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+
+	r.Delete("/v1/categories/{categoryId}", func(w http.ResponseWriter, _ *http.Request) {
 		w.Write([]byte("TODO"))
 	})
 

src/cli.js

@@ -48,13 +48,22 @@ const flattenQuery = (query) => query.replace(/\n[ ]*/g, ' ');
 // "WHERE id = :p.categoryId OR id = :b.id" => "WHERE id = :categoryId OR id = :id"
 const removePlaceholders = (query) => query.replace(/:[pb]\./g, ':');
 
+// "/categories/:id" => "/categories/{id}"
+// (used only with Golang's go-chi)
+const convertPathPlaceholders = (path) => {
+    return path.replace(/:[^\/]+/g, (placeholder) => '{' +  placeholder.slice(1) + '}');
+};
+
 const createEndpoints = async (destDir, lang, config) => {
     const fileName = `routes.${lang}`
     console.log('Generate', fileName);
     const resultFile = path.join(destDir, fileName);
 
     for (let endpoint of config) {
         let path = endpoint.path;
+        if (lang === 'go') {
+            path = convertPathPlaceholders(path)
+        }
         if (endpoint.hasOwnProperty('get')) {
             console.log('GET', path, '=>', removePlaceholders(flattenQuery(endpoint.get)));
         } else if (endpoint.hasOwnProperty('get_list')) {
@@ -99,7 +108,10 @@ const createEndpoints = async (destDir, lang, config) => {
             // "SELECT *\n   FROM foo" => "'SELECT * FROM foo'"
             "formatQuery": (query) => {
                 return "'" + removePlaceholders(flattenQuery(query)) + "'";
-            }
+            },
+
+            // (used only with Golang's go-chi)
+            "convertPathPlaceholders": convertPathPlaceholders,
         }
     );
 

src/templates/app.go

@@ -3,12 +3,14 @@ package main
 import "fmt"
 import "net/http"
 import "os"
+import "github.com/go-chi/chi"
 
 func main() {
-	registerRoutes()
+	r := chi.NewRouter()
+	registerRoutes(r)
 
 	fmt.Println("Listen on 3000")
-	if err := http.ListenAndServe(":3000", nil); err != nil {
+	if err := http.ListenAndServe(":3000", r); err != nil {
 		fmt.Fprintf(os.Stderr, "ListenAndServe failed: %v", err)
 		os.Exit(1)
 	}

src/templates/routes.go.ejs

@@ -1,11 +1,40 @@
 package main
 
 import "net/http"
+import "github.com/go-chi/chi"
 
-func registerRoutes() {
-<% endpoints.forEach(function(endpoint) { %>
-	http.HandleFunc("<%- endpoint.path %>", func(w http.ResponseWriter, _ *http.Request) {
+func registerRoutes(r chi.Router) {
+<%
+endpoints.forEach(function(endpoint) {
+	const path = convertPathPlaceholders(endpoint.path);
+	if (endpoint.hasOwnProperty('get') || endpoint.hasOwnProperty('get_list')) {
+%>
+	r.Get("<%- path %>", func(w http.ResponseWriter, _ *http.Request) {
 		w.Write([]byte("TODO"))
 	})
-<% }) %>
+<%
+	}
+	if (endpoint.hasOwnProperty('post')) {
+%>
+	r.Post("<%- path %>", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+<%
+	}
+	if (endpoint.hasOwnProperty('put')) {
+%>
+	r.Put("<%- path %>", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+<%
+	}
+	if (endpoint.hasOwnProperty('delete')) {
+%>
+	r.Delete("<%- path %>", func(w http.ResponseWriter, _ *http.Request) {
+		w.Write([]byte("TODO"))
+	})
+<%
+	}
+})
+%>
 }