Add route priority

Signed-off-by: Emile Vauge <[email protected]>

Author: emilevauge

mux.go

@@ -11,6 +11,7 @@ import (
 	"net/http"
 	"path"
 	"regexp"
+	"sort"
 )
 
 var (
@@ -403,6 +404,17 @@ func (r *Router) walk(walkFn WalkFunc, ancestors []*Route) error {
 	return nil
 }
 
+type routes []*Route
+
+func (r routes) Len() int           { return len(r) }
+func (r routes) Swap(i, j int)      { r[i], r[j] = r[j], r[i] }
+func (r routes) Less(i, j int) bool { return r[i].GetPriority() > r[j].GetPriority() }
+
+// SortRoutes sort routes by route priority
+func (r *Router) SortRoutes() {
+	sort.Sort(routes(r.routes))
+}
+
 // ----------------------------------------------------------------------------
 // Context
 // ----------------------------------------------------------------------------

route.go

@@ -23,6 +23,8 @@ type Route struct {
 	name string
 	// Error resulted from building a route.
 	err error
+	// Priority of this route
+	priority int
 
 	// "global" reference to all named routes
 	namedRoutes map[string]*Route
@@ -153,6 +155,19 @@ func (r *Route) GetName() string {
 	return r.name
 }
 
+// Priority -----------------------------------------------------------------------
+
+// Priority sets the priority for the route
+func (r *Route) Priority(priority int) *Route {
+	r.priority = priority
+	return r
+}
+
+// GetPriority returns the priority for the route.
+func (r *Route) GetPriority() int {
+	return r.priority
+}
+
 // ----------------------------------------------------------------------------
 // Matchers
 // ----------------------------------------------------------------------------