Examples with ServiceInstances com.tencent.polaris.api.pojo.ServiceInstances used on opensource projects

Example 21 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances 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());
    }
}

Example 22 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project polaris-java by polarismesh.

the class APIFacade method getInstances.

public static List<?> getInstances(String namespace, String service, Map<String, String> srcLabels, Map<String, String> dstLabels) {
    if (!inited.get()) {
        LOGGER.info("polaris not inited, updateServiceCallResult fail");
        return null;
    }
    GetInstancesRequest getInstancesRequest = new GetInstancesRequest();
    getInstancesRequest.setNamespace(namespace);
    getInstancesRequest.setService(service);
    if (MapUtils.isNotEmpty(srcLabels)) {
        ServiceInfo serviceInfo = new ServiceInfo();
        serviceInfo.setMetadata(srcLabels);
        getInstancesRequest.setServiceInfo(serviceInfo);
    }
    if (MapUtils.isNotEmpty(dstLabels)) {
        getInstancesRequest.setMetadata(dstLabels);
    }
    InstancesResponse instancesResp = consumerAPI.getInstances(getInstancesRequest);
    ServiceInstances serviceInstances = instancesResp.toServiceInstances();
    return serviceInstances.getInstances();
}

Example 23 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project polaris-java by polarismesh.

the class SyncFlow method commonSyncGetOneInstance.

/**
 * 获取单个实例
 *
 * @param request 请求对象
 * @return 实例应答
 * @throws PolarisException 异常
 */
public InstancesResponse commonSyncGetOneInstance(CommonInstancesRequest request) throws PolarisException {
    syncGetServiceInstances(request);
    ServiceInstances dstInstances = request.getDstInstances();
    if (CollectionUtils.isEmpty(dstInstances.getInstances())) {
        return new InstancesResponse(dstInstances);
    }
    ServiceInstances routerInstances = BaseFlow.processServiceRouters(request.getRouteInfo(), request.getDstInstances(), extensions.getConfigRouterChainGroup());
    LoadBalancer loadBalancer = extensions.getLoadBalancer();
    Instance instance = BaseFlow.processLoadBalance(loadBalancer, request.getCriteria(), routerInstances);
    return new InstancesResponse(dstInstances, instance);
}

Example 24 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project polaris-java by polarismesh.

the class SyncFlow method commonSyncGetInstances.

/**
 * 获取多个实例列表
 *
 * @param request 请求对象
 * @return 实例应答
 * @throws PolarisException 异常
 */
public InstancesResponse commonSyncGetInstances(CommonInstancesRequest request) throws PolarisException {
    syncGetServiceInstances(request);
    ServiceInstances dstInstances = request.getDstInstances();
    if (CollectionUtils.isEmpty(dstInstances.getInstances())) {
        return new InstancesResponse(dstInstances);
    }
    ServiceInstances routerInstances = BaseFlow.processServiceRouters(request.getRouteInfo(), request.getDstInstances(), extensions.getConfigRouterChainGroup());
    return new InstancesResponse(routerInstances);
}

Example 25 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances 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);
}