add build router chain

cluster/directory/base_directory.go

@@ -49,6 +49,10 @@ func (dir *BaseDirectory) RouterChain() router.Chain {
 	return dir.routerChain
 }
 
+func (dir *BaseDirectory) SetRouterChain(routerChain router.Chain) {
+	dir.routerChain = routerChain
+}
+
 func GetRouterURLSet() *gxset.HashSet {
 	return routerURLSet
 }

cluster/router/chain/factory.go

-/*
- * Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements.  See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License.  You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-package chain
-
-import (
-	"github.com/apache/dubbo-go/cluster/router"
-	"github.com/apache/dubbo-go/common"
-	"github.com/apache/dubbo-go/common/extension"
-)
-
-func init() {
-	extension.SetRouterChainsFactory("chain", NewRouterChainFactory)
-}
-
-type RouterChainFactory struct{}
-
-func (c RouterChainFactory) Router(url *common.URL) (router.Chain, error) {
-	return NewRouterChain(url)
-}
-
-func NewRouterChainFactory() router.ChainFactory {
-	return RouterChainFactory{}
-}

cluster/router/router.go

@@ -29,11 +29,6 @@ type RouterFactory interface {
 	Router(*common.URL) (Router, error)
 }
 
-// ChainFactory Extension - Router Chain
-type ChainFactory interface {
-	Router(*common.URL) (Chain, error)
-}
-
 type Router interface {
 	Route([]protocol.Invoker, *common.URL, protocol.Invocation) []protocol.Invoker
 	Priority() int64

common/extension/router_factory.go

@@ -22,8 +22,7 @@ import (
 )
 
 var (
-	routers      = make(map[string]func() router.RouterFactory)
-	routerChains = make(map[string]func() router.ChainFactory)
+	routers = make(map[string]func() router.RouterFactory)
 )
 
 func SetRouterFactory(name string, fun func() router.RouterFactory) {
@@ -37,17 +36,6 @@ func GetRouterFactory(name string) router.RouterFactory {
 	return routers[name]()
 }
 
-func SetRouterChainsFactory(name string, fun func() router.ChainFactory) {
-	routerChains[name] = fun
-}
-
-func GetRouterChainsFactory(name string) router.ChainFactory {
-	if routers[name] == nil {
-		panic("router_chain_factory for " + name + " is not existing, make sure you have import the package.")
-	}
-	return routerChains[name]()
-}
-
 func GetRouters() map[string]func() router.RouterFactory {
 	return routers
 }

registry/directory/directory.go

@@ -31,6 +31,7 @@ import (
 import (
 	"github.com/apache/dubbo-go/cluster/directory"
 	"github.com/apache/dubbo-go/cluster/router"
+	"github.com/apache/dubbo-go/cluster/router/chain"
 	"github.com/apache/dubbo-go/common"
 	"github.com/apache/dubbo-go/common/constant"
 	"github.com/apache/dubbo-go/common/extension"
@@ -180,7 +181,7 @@ func toRouters(urls []*common.URL) []router.Router {
 			continue
 		}
 		url.Protocol = routerKey
-		factory := extension.GetRouterFactory(url.GetParam(constant.ROUTER_KEY, "condition"))
+		factory := extension.GetRouterFactory(url.GetParam(constant.ROUTER_KEY, routerKey))
 		router, e := factory.Router(url)
 		if e != nil {
 			logger.Error("factory.Router(url){%s} , error : %s", url, e)
@@ -219,18 +220,34 @@ func (dir *registryDirectory) toGroupInvokers() []protocol.Invoker {
 		//len is 1 it means no group setting ,so do not need cluster again
 		for _, invokers := range groupInvokersMap {
 			groupInvokersList = invokers
+			dir.buildRouterChain(invokers)
 		}
 	} else {
 		for _, invokers := range groupInvokersMap {
 			staticDir := directory.NewStaticDirectory(invokers)
 			cluster := extension.GetCluster(dir.GetUrl().SubURL.GetParam(constant.CLUSTER_KEY, constant.DEFAULT_CLUSTER))
 			groupInvokersList = append(groupInvokersList, cluster.Join(staticDir))
+
+			dir.buildRouterChain(invokers)
 		}
 	}
 
 	return groupInvokersList
 }
 
+func (dir *registryDirectory) buildRouterChain(invokers []protocol.Invoker) error {
+	if len(invokers) == 0 {
+		return perrors.Errorf("invokers == null")
+	}
+	url := invokers[0].GetUrl()
+	routerChain, e := chain.NewRouterChain(&url)
+	if e != nil {
+		return e
+	}
+	dir.SetRouterChain(routerChain)
+	return nil
+}
+
 func (dir *registryDirectory) uncacheInvoker(url *common.URL) {
 	logger.Debugf("service will be deleted in cache invokers: invokers key is  %s!", url.Key())
 	dir.cacheInvokersMap.Delete(url.Key())