mirror of
https://github.com/mainflux/mainflux.git
synced 2025-04-27 13:48:49 +08:00
c966a7802d
Setup top-level glide dependencies file. Migrated all of the manager service code into this repository. Fixed docker build procedure. Extracted executable to the top-level. Signed-off-by: Dejan Mijic <dejan@mainflux.com>
129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
/********************************
|
|
*** Multiplexer for Go ***
|
|
*** Bone is under MIT license ***
|
|
*** Code by CodingFerret ***
|
|
*** github.com/go-zoo ***
|
|
*********************************/
|
|
|
|
package bone
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// Mux have routes and a notFound handler
|
|
// Route: all the registred route
|
|
// notFound: 404 handler, default http.NotFound if not provided
|
|
type Mux struct {
|
|
Routes map[string][]*Route
|
|
prefix string
|
|
notFound http.Handler
|
|
}
|
|
|
|
var (
|
|
static = "static"
|
|
method = []string{"GET", "POST", "PUT", "DELETE", "HEAD", "PATCH", "OPTIONS"}
|
|
)
|
|
|
|
// New create a pointer to a Mux instance
|
|
func New() *Mux {
|
|
return &Mux{Routes: make(map[string][]*Route)}
|
|
}
|
|
|
|
// Prefix set a default prefix for all routes registred on the router
|
|
func (m *Mux) Prefix(p string) *Mux {
|
|
m.prefix = strings.TrimSuffix(p, "/")
|
|
return m
|
|
}
|
|
|
|
// Serve http request
|
|
func (m *Mux) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
|
|
// Check if a route match
|
|
if !m.parse(rw, req) {
|
|
// Check if it's a static ressource
|
|
if !m.staticRoute(rw, req) {
|
|
// Check if the request path doesn't end with /
|
|
if !m.validate(rw, req) {
|
|
m.HandleNotFound(rw, req)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Handle add a new route to the Mux without a HTTP method
|
|
func (m *Mux) Handle(path string, handler http.Handler) {
|
|
for _, mt := range method {
|
|
m.register(mt, path, handler)
|
|
}
|
|
}
|
|
|
|
// HandleFunc is use to pass a func(http.ResponseWriter, *Http.Request) instead of http.Handler
|
|
func (m *Mux) HandleFunc(path string, handler http.HandlerFunc) {
|
|
m.Handle(path, handler)
|
|
}
|
|
|
|
// Get add a new route to the Mux with the Get method
|
|
func (m *Mux) Get(path string, handler http.Handler) *Route {
|
|
return m.register("GET", path, handler)
|
|
}
|
|
|
|
// Post add a new route to the Mux with the Post method
|
|
func (m *Mux) Post(path string, handler http.Handler) *Route {
|
|
return m.register("POST", path, handler)
|
|
}
|
|
|
|
// Put add a new route to the Mux with the Put method
|
|
func (m *Mux) Put(path string, handler http.Handler) *Route {
|
|
return m.register("PUT", path, handler)
|
|
}
|
|
|
|
// Delete add a new route to the Mux with the Delete method
|
|
func (m *Mux) Delete(path string, handler http.Handler) *Route {
|
|
return m.register("DELETE", path, handler)
|
|
}
|
|
|
|
// Head add a new route to the Mux with the Head method
|
|
func (m *Mux) Head(path string, handler http.Handler) *Route {
|
|
return m.register("HEAD", path, handler)
|
|
}
|
|
|
|
// Patch add a new route to the Mux with the Patch method
|
|
func (m *Mux) Patch(path string, handler http.Handler) *Route {
|
|
return m.register("PATCH", path, handler)
|
|
}
|
|
|
|
// Options add a new route to the Mux with the Options method
|
|
func (m *Mux) Options(path string, handler http.Handler) *Route {
|
|
return m.register("OPTIONS", path, handler)
|
|
}
|
|
|
|
// NotFound the mux custom 404 handler
|
|
func (m *Mux) NotFound(handler http.Handler) {
|
|
m.notFound = handler
|
|
}
|
|
|
|
// Register the new route in the router with the provided method and handler
|
|
func (m *Mux) register(method string, path string, handler http.Handler) *Route {
|
|
r := NewRoute(m.prefix+path, handler)
|
|
if valid(path) {
|
|
m.Routes[method] = append(m.Routes[method], r)
|
|
return r
|
|
}
|
|
m.Routes[static] = append(m.Routes[static], r)
|
|
return r
|
|
}
|
|
|
|
// SubRoute register a router as a SubRouter of bone
|
|
func (m *Mux) SubRoute(path string, router Router) *Route {
|
|
r := NewRoute(m.prefix+path, router)
|
|
if valid(path) {
|
|
r.Atts += SUB
|
|
for _, mt := range method {
|
|
m.Routes[mt] = append(m.Routes[mt], r)
|
|
}
|
|
return r
|
|
}
|
|
return nil
|
|
}
|