Add route priority

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

Author: emilevauge

mux.go

@@ -10,6 +10,7 @@ import (
 	"net/http"
 	"path"
 	"regexp"
+	"sort"
 )
 
 var (
@@ -353,6 +354,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
 // ----------------------------------------------------------------------------
@@ -372,7 +384,7 @@ type RouteMatch struct {
 type contextKey int
 
 const (
-	varsKey contextKey = iota
+	varsKey  contextKey = iota
 	routeKey
 )
 

route.go

@@ -39,6 +39,8 @@ type Route struct {
 	name string
 	// Error resulted from building a route.
 	err error
+	// Priority of this route
+	priority int
 
 	buildVarsFunc BuildVarsFunc
 }
@@ -147,6 +149,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
 // ----------------------------------------------------------------------------