use of com.tencent.polaris.router.api.rpc.ProcessRoutersRequest.RouterNamesGroup in project polaris-java by polarismesh.
the class RouterExample method main.
public static void main(String[] args) throws Exception {
InitResult initResult = ExampleUtils.initConsumerConfiguration(args);
String namespace = initResult.getNamespace();
String service = initResult.getService();
try (SDKContext sdkContext = SDKContext.initContext()) {
ConsumerAPI consumerAPI = DiscoveryAPIFactory.createConsumerAPIByContext(sdkContext);
RouterAPI routerAPI = RouterAPIFactory.createRouterAPIByContext(sdkContext);
// 1. 拉取全量服务实例
GetAllInstancesRequest getAllInstancesRequest = new GetAllInstancesRequest();
getAllInstancesRequest.setNamespace(namespace);
getAllInstancesRequest.setService(service);
InstancesResponse allInstanceResp = consumerAPI.getAllInstance(getAllInstancesRequest);
ServiceInstances dstInstances = allInstanceResp.toServiceInstances();
// 2. 执行服务路由
ProcessRoutersRequest processRoutersRequest = new ProcessRoutersRequest();
// 被调服务
System.out.printf("instances count before routing is %s%n", dstInstances.getInstances().size());
// 主调方信息
ServiceInfo srcSourceInfo = new ServiceInfo();
Map<String, String> labels = new HashMap<>();
labels.put("env", "test");
srcSourceInfo.setMetadata(labels);
RouterNamesGroup routerNamesGroup = new RouterNamesGroup();
List<String> coreRouters = new ArrayList<>();
coreRouters.add(ServiceRouterConfig.DEFAULT_ROUTER_RULE);
coreRouters.add(ServiceRouterConfig.DEFAULT_ROUTER_METADATA);
coreRouters.add(ServiceRouterConfig.DEFAULT_ROUTER_NEARBY);
// 设置走规则路由
routerNamesGroup.setCoreRouters(coreRouters);
processRoutersRequest.setDstInstances(dstInstances);
processRoutersRequest.setSourceService(srcSourceInfo);
processRoutersRequest.setRouters(routerNamesGroup);
ProcessRoutersResponse processRoutersResponse = routerAPI.processRouters(processRoutersRequest);
System.out.printf("instances count after routing is %s%n", processRoutersResponse.getServiceInstances().getInstances().size());
// 3. 执行负载均衡
ProcessLoadBalanceRequest processLoadBalanceRequest = new ProcessLoadBalanceRequest();
processLoadBalanceRequest.setDstInstances(processRoutersResponse.getServiceInstances());
processLoadBalanceRequest.setLbPolicy(LoadBalanceConfig.LOAD_BALANCE_WEIGHTED_RANDOM);
ProcessLoadBalanceResponse processLoadBalanceResponse = routerAPI.processLoadBalance(processLoadBalanceRequest);
System.out.printf("instances after lb is %s%n", processLoadBalanceResponse.getTargetInstance());
}
}
use of com.tencent.polaris.router.api.rpc.ProcessRoutersRequest.RouterNamesGroup in project polaris-java by polarismesh.
the class DefaultRouterAPI method processRouters.
@Override
public ProcessRoutersResponse processRouters(ProcessRoutersRequest request) throws PolarisException {
checkAvailable("RouterAPI");
RouterValidator.validateProcessRouterRequest(request);
RouterNamesGroup requestRouterGroup = request.getRouters();
List<ServiceRouter> beforeRouters;
if (null != requestRouterGroup && CollectionUtils.isNotEmpty(requestRouterGroup.getBeforeRouters())) {
beforeRouters = Extensions.loadServiceRouters(requestRouterGroup.getBeforeRouters(), extensions.getPlugins(), true);
} else {
beforeRouters = extensions.getConfigRouterChainGroup().getBeforeRouters();
}
List<ServiceRouter> afterRouters;
if (null != requestRouterGroup && CollectionUtils.isNotEmpty(requestRouterGroup.getAfterRouters())) {
afterRouters = Extensions.loadServiceRouters(requestRouterGroup.getAfterRouters(), extensions.getPlugins(), true);
} else {
afterRouters = extensions.getConfigRouterChainGroup().getAfterRouters();
}
List<ServiceRouter> coreRouters;
if (null != requestRouterGroup && CollectionUtils.isNotEmpty(requestRouterGroup.getCoreRouters())) {
coreRouters = Extensions.loadServiceRouters(requestRouterGroup.getCoreRouters(), extensions.getPlugins(), true);
} else {
coreRouters = extensions.getConfigRouterChainGroup().getCoreRouters();
}
ServiceInstances dstInstances = request.getDstInstances();
RouteInfo routeInfo = new RouteInfo(request.getSourceService(), dstInstances, request.getMethod());
// 获取路由规则
DefaultFlowControlParam engineFlowControlParam = new DefaultFlowControlParam();
BaseFlow.buildFlowControlParam(request, config, engineFlowControlParam);
Set<ServiceEventKey> routerKeys = new HashSet<>();
ServiceEventKey dstSvcEventKey = new ServiceEventKey(new ServiceKey(dstInstances.getNamespace(), dstInstances.getService()), EventType.ROUTING);
routerKeys.add(dstSvcEventKey);
ServiceEventKey srcSvcEventKey = null;
if (null != routeInfo.getSourceService() && StringUtils.isNotBlank(routeInfo.getSourceService().getNamespace()) && StringUtils.isNotBlank(routeInfo.getSourceService().getService())) {
srcSvcEventKey = new ServiceEventKey(new ServiceKey(routeInfo.getSourceService().getNamespace(), routeInfo.getSourceService().getService()), EventType.ROUTING);
routerKeys.add(srcSvcEventKey);
}
DefaultServiceEventKeysProvider svcKeysProvider = new DefaultServiceEventKeysProvider();
svcKeysProvider.setSvcEventKeys(routerKeys);
ResourcesResponse resourcesResponse = BaseFlow.syncGetResources(extensions, false, svcKeysProvider, engineFlowControlParam);
routeInfo.setDestRouteRule(resourcesResponse.getServiceRule(dstSvcEventKey));
if (null != srcSvcEventKey) {
routeInfo.setSourceRouteRule(resourcesResponse.getServiceRule(srcSvcEventKey));
}
// 执行路由
DefaultRouterChainGroup routerChainGroup = new DefaultRouterChainGroup(beforeRouters, coreRouters, afterRouters);
ServiceInstances svcInstances = BaseFlow.processServiceRouters(routeInfo, dstInstances, routerChainGroup);
return new ProcessRoutersResponse(svcInstances);
}
Aggregations