Dubbo 可以实现的扩展很多, 官方文档在这: https://dubbo.apache.org/zh/docs/v2.7/dev/impls/ (太简单了....)
下面我们实现一个Route Factory, 它会根据参数中的workzone来选择合适的Invoker实例, 可以实现一定程度上的灰度发布.
首先实现一个Route,
package com.cnscud.dubboroom.base.dubbo;
import org.apache.dubbo.common.URL;
import org.apache.dubbo.rpc.Invocation;
import org.apache.dubbo.rpc.Invoker;
import org.apache.dubbo.rpc.RpcException;
import org.apache.dubbo.rpc.cluster.router.AbstractRouter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.ArrayList;
import java.util.List;
/**
* My Router.
*
* @author Felix Zhang 2021-06-21 10:38
* @version 1.0.0
*/
public class MyRouter extends AbstractRouter {
private static Logger logger = LoggerFactory.getLogger(MyRouter.class);
protected static String ZONE_KEY = "workzone";
@Override
public <T> List<Invoker<T>> route(List<Invoker<T>> invokers, URL url, Invocation invocation) throws RpcException {
//自己的逻辑
String workzone = invocation.getAttachment("workzone");
List<Invoker<T>> newInvokerList = new ArrayList<>();
//选择特定服务器
for (Invoker<T> invoker : invokers) {
URL serviceUrl = invoker.getUrl();
logger.info("router serviceUrl: " + serviceUrl.toIdentityString() + " " + serviceUrl.getParameters() + ", port: " + serviceUrl.getPort());
if (serviceUrl.hasParameter(ZONE_KEY) && workzone != null && workzone.equalsIgnoreCase(serviceUrl.getParameter(ZONE_KEY))) {
logger.info("find match invoker for workzone: " + workzone + " ip: " + serviceUrl.getIp() + " port: " + serviceUrl.getPort());
newInvokerList.add(invoker);
}
}
if (newInvokerList.size() > 0) {
return newInvokerList;
}
return invokers;
}
}
然后实现一个 Factory,
package com.cnscud.dubboroom.base.dubbo;
import org.apache.dubbo.common.URL;
import org.apache.dubbo.rpc.cluster.CacheableRouterFactory;
import org.apache.dubbo.rpc.cluster.Router;
/**
* My Router Factory.
*
* @author Felix Zhang 2021-06-21 10:48
* @version 1.0.0
*/
public class MyRouterFactory extends CacheableRouterFactory {
@Override
protected Router createRouter(URL url) {
return new MyRouter();
}
}
在配置文件里声明:
声明 Route Factory: 文件名: META-INF/dubbo/org.apache.dubbo.rpc.cluster.RouterFactory
myrouter=com.cnscud.dubboroom.base.dubbo.MyRouterFactory
然后在dubbo的配置里声明使用:
dubbo:
consumer:
router: myrouter
这样就可以生效了......
(仅供参考!!)