forked from EconomistDigitalSolutions/goberry
/
router.go
49 lines (42 loc) · 1.14 KB
/
router.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package main
import (
"fmt"
"log"
"net/http"
"github.com/EconomistDigitalSolutions/ramlapi"
"github.com/gorilla/mux"
)
var router *mux.Router
// NewRouter creates a mux router, sets up
// a static handler and registers the dynamic
// routes and middleware handlers with the mux.
func NewRouter(ramlFile string) *mux.Router {
router = mux.NewRouter().StrictSlash(true)
// Assemble middleware as required.
assembleMiddleware(router)
assembleRoutes(router, ramlFile)
return router
}
// assembleMiddleware sets up the middleware stack.
func assembleMiddleware(r *mux.Router) {
http.Handle("/",
JSONMiddleware(
LoggingMiddleware(
RecoverMiddleware(r))))
}
func assembleRoutes(r *mux.Router, f string) {
// Parse the RAML API specification.
api, err := ramlapi.ProcessRAML(f)
if err != nil {
log.Fatal(err)
}
logChannel("raml-processor", fmt.Sprintf("processing API spec for %s", api.Title))
logChannel("raml-processor", fmt.Sprintf("base URI at %s", api.BaseUri))
ramlapi.Build(api, routerFunc)
}
func routerFunc(data map[string]string) {
router.
Methods(data["verb"]).
Path(data["path"]).
Handler(RouteMap[data["handler"]])
}