| /*
|
| * Copyright 1999-2011 Alibaba Group.
|
| *
|
| * Licensed 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 com.alibaba.dubbo.rpc.cluster.directory; |
| |
| import java.util.ArrayList;
|
| import java.util.List;
|
|
|
| import com.alibaba.dubbo.common.Constants;
|
| import com.alibaba.dubbo.common.URL;
|
| import com.alibaba.dubbo.common.extension.ExtensionLoader;
|
| import com.alibaba.dubbo.common.logger.Logger;
|
| import com.alibaba.dubbo.common.logger.LoggerFactory;
|
| import com.alibaba.dubbo.rpc.Invocation;
|
| import com.alibaba.dubbo.rpc.Invoker;
|
| import com.alibaba.dubbo.rpc.RpcException;
|
| import com.alibaba.dubbo.rpc.cluster.Directory;
|
| import com.alibaba.dubbo.rpc.cluster.Router;
|
| import com.alibaba.dubbo.rpc.cluster.RouterFactory;
|
| import com.alibaba.dubbo.rpc.cluster.router.MockInvokersSelector;
|
| |
| /** |
| * 增加router的Directory
|
| * |
| * @author chao.liuc |
| */ |
| public abstract class AbstractDirectory<T> implements Directory<T> { |
|
|
| // 日志输出
|
| private static final Logger logger = LoggerFactory.getLogger(AbstractDirectory.class);
|
| |
| private final URL url ;
|
| |
| private volatile boolean destroyed = false; |
| |
| private volatile List<Router> routers; |
| |
| public AbstractDirectory(URL url) { |
| this(url, null); |
| } |
| |
| public AbstractDirectory(URL url, List<Router> routers) { |
| if (url == null) |
| throw new IllegalArgumentException("url == null"); |
| this.url = url; |
| setRouters(routers); |
| } |
| |
| public List<Invoker<T>> list(Invocation invocation) throws RpcException { |
| if (destroyed){ |
| throw new RpcException("Directory already destroyed .url: "+ getUrl()); |
| } |
| List<Invoker<T>> invokers = doList(invocation);
|
| List<Router> localRouters = this.routers; // local reference
|
| if (localRouters != null && localRouters.size() > 0) {
|
| for (Router router: localRouters){
|
| try {
|
| if (router.getUrl() == null || router.getUrl().getParameter(Constants.RUNTIME_KEY, true)) { |
| invokers = router.route(invokers, getUrl(), invocation);
|
| }
|
| } catch (Throwable t) {
|
| logger.error("Failed to execute router: " + getUrl() + ", cause: " + t.getMessage(), t);
|
| } |
| }
|
| }
|
| return invokers; |
| } |
| |
| public URL getUrl() { |
| return url; |
| } |
| |
| public List<Router> getRouters(){ |
| return routers; |
| } |
| |
| protected void setRouters(List<Router> routers){
|
| // copy list
|
| routers = routers == null ? new ArrayList<Router>() : new ArrayList<Router>(routers);
|
| // append url router
|
| String routerkey = url.getParameter(Constants.ROUTER_KEY);
|
| if (routerkey != null && routerkey.length() > 0) {
|
| RouterFactory routerFactory = ExtensionLoader.getExtensionLoader(RouterFactory.class).getExtension(routerkey);
|
| routers.add(routerFactory.getRouter(url));
|
| }
|
| // append mock invoker selector
|
| routers.add(new MockInvokersSelector());
|
| this.routers = routers; |
| } |
|
|
| public boolean isDestroyed() {
|
| return destroyed;
|
| }
|
| |
| public void destroy(){ |
| destroyed = true; |
| }
|
|
|
| protected abstract List<Invoker<T>> doList(Invocation invocation) throws RpcException ;
|
| |
| } |