|
|
|
|
|
|
|
|
"github.com/gorilla/mux" |
|
|
"github.com/gorilla/mux" |
|
|
) |
|
|
) |
|
|
|
|
|
|
|
|
type Router struct { |
|
|
|
|
|
|
|
|
type Server struct { |
|
|
middleware *middleware |
|
|
middleware *middleware |
|
|
mux *mux.Router |
|
|
mux *mux.Router |
|
|
ready bool |
|
|
ready bool |
|
|
subrouters []*Router |
|
|
|
|
|
|
|
|
subservers []*Server |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
func New() *Router { |
|
|
|
|
|
return &Router{ |
|
|
|
|
|
|
|
|
func New() *Server { |
|
|
|
|
|
return &Server{ |
|
|
middleware: new(middleware), |
|
|
middleware: new(middleware), |
|
|
mux: mux.NewRouter().StrictSlash(true), |
|
|
mux: mux.NewRouter().StrictSlash(true), |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// build constructs all router/subrouters along with their middleware modules chain. |
|
|
|
|
|
func (self *Router) build() http.Handler { |
|
|
|
|
|
|
|
|
// build constructs all server/subservers along with their middleware modules chain. |
|
|
|
|
|
func (self *Server) build() http.Handler { |
|
|
if !self.ready { |
|
|
if !self.ready { |
|
|
// * add router into middlware stack to serve as final http.Handler. |
|
|
|
|
|
|
|
|
// * add server mux into middlware stack to serve as final http.Handler. |
|
|
self.Use(func(http.Handler) http.Handler { |
|
|
self.Use(func(http.Handler) http.Handler { |
|
|
return self.mux |
|
|
return self.mux |
|
|
}) |
|
|
}) |
|
|
// * add subrouters into middlware stack to serve as final http.Handler. |
|
|
|
|
|
for index := 0; index < len(self.subrouters); index++ { |
|
|
|
|
|
router := self.subrouters[index] |
|
|
|
|
|
router.Use(func(http.Handler) http.Handler { |
|
|
|
|
|
return router.mux |
|
|
|
|
|
|
|
|
// * add subservers into middlware stack to serve as final http.Handler. |
|
|
|
|
|
for index := 0; index < len(self.subservers); index++ { |
|
|
|
|
|
server := self.subservers[index] |
|
|
|
|
|
server.Use(func(http.Handler) http.Handler { |
|
|
|
|
|
return server.mux |
|
|
}) |
|
|
}) |
|
|
} |
|
|
} |
|
|
self.ready = true |
|
|
self.ready = true |
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// register adds the http.Handler/http.HandleFunc into Gorilla mux. |
|
|
// register adds the http.Handler/http.HandleFunc into Gorilla mux. |
|
|
func (self *Router) register(method string, pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) register(method string, pattern string, handler interface{}) { |
|
|
// finds the full function name (with package) as its mappings. |
|
|
// finds the full function name (with package) as its mappings. |
|
|
var name = runtime.FuncForPC(reflect.ValueOf(handler).Pointer()).Name() |
|
|
var name = runtime.FuncForPC(reflect.ValueOf(handler).Pointer()).Name() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Get is a shortcut for mux.HandleFunc(pattern, handler).Methods("GET"), |
|
|
// Get is a shortcut for mux.HandleFunc(pattern, handler).Methods("GET"), |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
func (self *Router) Get(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Get(pattern string, handler interface{}) { |
|
|
self.register("GET", pattern, handler) |
|
|
self.register("GET", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Head is a shortcut for mux.HandleFunc(pattern, handler).Methods("HEAD") |
|
|
// Head is a shortcut for mux.HandleFunc(pattern, handler).Methods("HEAD") |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
func (self *Router) Head(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Head(pattern string, handler interface{}) { |
|
|
self.register("HEAD", pattern, handler) |
|
|
self.register("HEAD", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Options is a shortcut for mux.HandleFunc(pattern, handler).Methods("OPTIONS") |
|
|
// Options is a shortcut for mux.HandleFunc(pattern, handler).Methods("OPTIONS") |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// NOTE method OPTIONS is **NOT** cachable, beware of what you are going to do. |
|
|
// NOTE method OPTIONS is **NOT** cachable, beware of what you are going to do. |
|
|
func (self *Router) Options(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Options(pattern string, handler interface{}) { |
|
|
self.register("OPTIONS", pattern, handler) |
|
|
self.register("OPTIONS", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Post is a shortcut for mux.HandleFunc(pattern, handler).Methods("POST") |
|
|
// Post is a shortcut for mux.HandleFunc(pattern, handler).Methods("POST") |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
func (self *Router) Post(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Post(pattern string, handler interface{}) { |
|
|
self.register("POST", pattern, handler) |
|
|
self.register("POST", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Put is a shortcut for mux.HandleFunc(pattern, handler).Methods("PUT") |
|
|
// Put is a shortcut for mux.HandleFunc(pattern, handler).Methods("PUT") |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
func (self *Router) Put(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Put(pattern string, handler interface{}) { |
|
|
self.register("PUT", pattern, handler) |
|
|
self.register("PUT", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Delete is a shortcut for mux.HandleFunc(pattern, handler).Methods("DELETE") |
|
|
// Delete is a shortcut for mux.HandleFunc(pattern, handler).Methods("DELETE") |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
// it also fetch the full function name of the handler (with package) to name the route. |
|
|
func (self *Router) Delete(pattern string, handler interface{}) { |
|
|
|
|
|
|
|
|
func (self *Server) Delete(pattern string, handler interface{}) { |
|
|
self.register("Delete", pattern, handler) |
|
|
self.register("Delete", pattern, handler) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Group creates a new application group under the given path prefix. |
|
|
// Group creates a new application group under the given path prefix. |
|
|
func (self *Router) Group(prefix string) *Router { |
|
|
|
|
|
|
|
|
func (self *Server) Group(prefix string) *Server { |
|
|
var middleware = new(middleware) |
|
|
var middleware = new(middleware) |
|
|
self.mux.PathPrefix(prefix).Handler(middleware) |
|
|
self.mux.PathPrefix(prefix).Handler(middleware) |
|
|
var mux = self.mux.PathPrefix(prefix).Subrouter() |
|
|
var mux = self.mux.PathPrefix(prefix).Subrouter() |
|
|
|
|
|
|
|
|
router := &Router{middleware: middleware, mux: mux} |
|
|
|
|
|
self.subrouters = append(self.subrouters, router) |
|
|
|
|
|
return router |
|
|
|
|
|
|
|
|
server := &Server{middleware: middleware, mux: mux} |
|
|
|
|
|
self.subservers = append(self.subservers, server) |
|
|
|
|
|
return server |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Name returns route name for the given request, if any. |
|
|
// Name returns route name for the given request, if any. |
|
|
func (self *Router) Name(r *http.Request) (name string) { |
|
|
|
|
|
|
|
|
func (self *Server) Name(r *http.Request) (name string) { |
|
|
var match mux.RouteMatch |
|
|
var match mux.RouteMatch |
|
|
if self.mux.Match(r, &match) { |
|
|
if self.mux.Match(r, &match) { |
|
|
name = match.Route.GetName() |
|
|
name = match.Route.GetName() |
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Use add the middleware module into the stack chain. |
|
|
// Use add the middleware module into the stack chain. |
|
|
func (self *Router) Use(module func(http.Handler) http.Handler) { |
|
|
|
|
|
|
|
|
func (self *Server) Use(module func(http.Handler) http.Handler) { |
|
|
self.middleware.stack = append(self.middleware.stack, module) |
|
|
self.middleware.stack = append(self.middleware.stack, module) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// ServeHTTP dispatches the request to the handler whose |
|
|
// ServeHTTP dispatches the request to the handler whose |
|
|
// pattern most closely matches the request URL. |
|
|
// pattern most closely matches the request URL. |
|
|
func (self *Router) ServeHTTP(w http.ResponseWriter, r *http.Request) { |
|
|
|
|
|
|
|
|
func (self *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) { |
|
|
self.build().ServeHTTP(w, r) |
|
|
self.build().ServeHTTP(w, r) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Run starts the application server to serve incoming requests at the given address. |
|
|
// Run starts the application server to serve incoming requests at the given address. |
|
|
func (self *Router) Run() { |
|
|
|
|
|
|
|
|
func (self *Server) Run() { |
|
|
configure() |
|
|
configure() |
|
|
runtime.GOMAXPROCS(maxprocs) |
|
|
runtime.GOMAXPROCS(maxprocs) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// Vars returns the route variables for the current request, if any. |
|
|
// Vars returns the route variables for the current request, if any. |
|
|
func (self *Router) Vars(r *http.Request) map[string]string { |
|
|
|
|
|
|
|
|
func (self *Server) Vars(r *http.Request) map[string]string { |
|
|
return mux.Vars(r) |
|
|
return mux.Vars(r) |
|
|
} |
|
|
} |