feat: add support for header based route

pixiu/pkg/common/http/manager_test.go

@@ -112,7 +112,7 @@ func TestCreateHttpConnectionManager(t *testing.T) {
 			RouteTrie: trie.NewTrieWithDefault("POST/api/v1/**", model.RouteAction{
 				Cluster:                     "test_dubbo",
 				ClusterNotFoundResponseCode: 505,
-			}),
+			}, nil),
 			Dynamic: false,
 		},
 		HTTPFilters: []*model.HTTPFilter{

pixiu/pkg/common/router/router.go

@@ -18,15 +18,21 @@
 package router
 
 import (
+	stdHttp "net/http"
 	"strings"
 	"sync"
 )
 
+import (
+	"github.com/pkg/errors"
+)
+
 import (
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/common/constant"
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/common/router/trie"
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/common/util/stringutil"
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/context/http"
+	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/logger"
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/model"
 	"github.com/apache/dubbo-go-pixiu/pixiu/pkg/server"
 )
@@ -46,6 +52,7 @@ func CreateRouterCoordinator(routeConfig *model.RouteConfiguration) *RouterCoord
 		server.GetRouterManager().AddRouterListener(rc)
 	}
 	rc.initTrie()
+	rc.initRegex()
 	return rc
 }
 
@@ -54,14 +61,38 @@ func (rm *RouterCoordinator) Route(hc *http.HttpContext) (*model.RouteAction, er
 	rm.rw.RLock()
 	defer rm.rw.RUnlock()
 
-	return rm.activeConfig.Route(hc.Request)
+	return rm.route(hc.Request)
 }
 
 func (rm *RouterCoordinator) RouteByPathAndName(path, method string) (*model.RouteAction, error) {
 	rm.rw.RLock()
 	defer rm.rw.RUnlock()
 
-	return rm.activeConfig.RouteByPathAndMethod(path, method)
+	return rm.activeConfig.RouteByPathAndMethod(path, method, nil)
+}
+
+func (rm *RouterCoordinator) route(req *stdHttp.Request) (*model.RouteAction, error) {
+	// match those route that only contains headers first
+	var matched []*model.Router
+	for _, route := range rm.activeConfig.Routes {
+		if len(route.Match.Prefix) > 0 {
+			continue
+		}
+		if route.Match.MatchHeader(req) {
+			matched = append(matched, route)
+		}
+	}
+
+	// always return the first match of header if got any
+	if len(matched) > 0 {
+		if len(matched[0].Route.Cluster) == 0 {
+			return nil, errors.New("action is nil. please check your configuration.")
+		}
+		return &matched[0].Route, nil
+	}
+
+	// match those route that only contains prefix or both prefix and header
+	return rm.activeConfig.Route(req)
 }
 
 func getTrieKey(method string, path string, isPrefix bool) string {
@@ -83,6 +114,21 @@ func (rm *RouterCoordinator) initTrie() {
 	}
 }
 
+func (rm *RouterCoordinator) initRegex() {
+	for _, router := range rm.activeConfig.Routes {
+		headers := router.Match.Headers
+		for i := range headers {
+			if headers[i].Regex && len(headers[i].Values) > 0 {
+				// regexp always use first value of header
+				err := headers[i].SetValueRegex(headers[i].Values[0])
+				if err != nil {
+					logger.Errorf("invalid regexp in headers[%d]: %v", i, err)
+				}
+			}
+		}
+	}
+}
+
 // OnAddRouter add router
 func (rm *RouterCoordinator) OnAddRouter(r *model.Router) {
 	//TODO: lock move to trie node
@@ -99,7 +145,7 @@ func (rm *RouterCoordinator) OnAddRouter(r *model.Router) {
 		} else {
 			key = getTrieKey(method, r.Match.Path, isPrefix)
 		}
-		_, _ = rm.activeConfig.RouteTrie.Put(key, r.Route)
+		_, _ = rm.activeConfig.RouteTrie.Put(key, r.Route, r.Match.HeaderMap())
 	}
 }