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

Example 11 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project spring-cloud-tencent by Tencent.

the class PolarisServiceDiscovery method getInstances.

/**
 * Return all instances for the given service.
 * @param serviceId id of service
 * @return list of instances
 * @throws PolarisException polarisException
 */
public List<ServiceInstance> getInstances(String serviceId) throws PolarisException {
    List<ServiceInstance> instances = new ArrayList<>();
    InstancesResponse filteredInstances = polarisDiscoveryHandler.getHealthyInstances(serviceId);
    ServiceInstances serviceInstances = filteredInstances.toServiceInstances();
    for (Instance instance : serviceInstances.getInstances()) {
        instances.add(new PolarisServiceInstance(instance));
    }
    return instances;
}

Example 12 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project spring-cloud-tencent by Tencent.

the class PolarisServerList method getServers.

private List<Server> getServers() {
    InstancesResponse allInstances = polarisDiscoveryHandler.getHealthyInstances(serviceId);
    ServiceInstances serviceInstances = allInstances.toServiceInstances();
    List<Server> polarisServers = new ArrayList<>();
    for (Instance instance : serviceInstances.getInstances()) {
        polarisServers.add(new PolarisServer(serviceInstances, instance));
    }
    return polarisServers;
}

Example 13 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project spring-cloud-tencent by Tencent.

the class PolarisLoadBalancerCompositeRuleTest method testBuildNearbyRouteRequest.

@Test
public void testBuildNearbyRouteRequest() {
    when(polarisNearByRouterProperties.isEnabled()).thenReturn(true);
    when(polarisLoadBalancerProperties.getStrategy()).thenReturn("");
    try (MockedStatic<ApplicationContextAwareUtils> mockedApplicationContextAwareUtils = Mockito.mockStatic(ApplicationContextAwareUtils.class)) {
        mockedApplicationContextAwareUtils.when(() -> ApplicationContextAwareUtils.getProperties(anyString())).thenReturn(testCallerService);
        setTransitiveMetadata();
        PolarisLoadBalancerCompositeRule compositeRule = new PolarisLoadBalancerCompositeRule(routerAPI, polarisLoadBalancerProperties, polarisNearByRouterProperties, polarisMetadataRouterProperties, polarisRuleBasedRouterProperties, config);
        ServiceInstances serviceInstances = assembleServiceInstances();
        PolarisRouterContext routerContext = assembleRouterContext();
        ProcessRoutersRequest request = compositeRule.buildProcessRoutersRequest(serviceInstances, routerContext);
        Map<String, String> routerMetadata = request.getRouterMetadata(NearbyRouter.ROUTER_TYPE_NEAR_BY);
        Assert.assertEquals(0, request.getRouterMetadata(MetadataRouter.ROUTER_TYPE_METADATA).size());
        Assert.assertEquals(1, routerMetadata.size());
        Assert.assertEquals("true", routerMetadata.get(NearbyRouter.ROUTER_ENABLED));
        Assert.assertEquals(1, request.getRouterMetadata(RuleBasedRouter.ROUTER_TYPE_RULE_BASED).size());
        Assert.assertEquals("false", request.getRouterMetadata(RuleBasedRouter.ROUTER_TYPE_RULE_BASED).get(RuleBasedRouter.ROUTER_ENABLED));
    }
}

Example 14 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project spring-cloud-tencent by Tencent.

the class PolarisLoadBalancer method getReachableServers.

@Override
public List<Server> getReachableServers() {
    ServiceInstances serviceInstances;
    if (polarisLoadBalancerProperties.getDiscoveryType().equals(ContextConstant.POLARIS)) {
        serviceInstances = getPolarisDiscoveryServiceInstances();
    } else {
        serviceInstances = getExtendDiscoveryServiceInstances();
    }
    if (serviceInstances == null || CollectionUtils.isEmpty(serviceInstances.getInstances())) {
        return Collections.emptyList();
    }
    List<Server> servers = new LinkedList<>();
    for (Instance instance : serviceInstances.getInstances()) {
        servers.add(new PolarisServer(serviceInstances, instance));
    }
    return servers;
}

Example 15 with ServiceInstances

use of com.tencent.polaris.api.pojo.ServiceInstances in project spring-cloud-tencent by Tencent.

the class PolarisLoadBalancer method getExtendDiscoveryServiceInstances.

private ServiceInstances getExtendDiscoveryServiceInstances() {
    List<Server> allServers = super.getAllServers();
    if (CollectionUtils.isEmpty(allServers)) {
        return null;
    }
    ServiceInstances serviceInstances;
    if (StringUtils.isBlank(name)) {
        throw new IllegalStateException("PolarisLoadBalancer only Server with AppName or ServiceIdForDiscovery attribute");
    }
    ServiceKey serviceKey = new ServiceKey(MetadataContext.LOCAL_NAMESPACE, name);
    List<Instance> instances = new ArrayList<>(8);
    for (Server server : allServers) {
        DefaultInstance instance = new DefaultInstance();
        instance.setNamespace(MetadataContext.LOCAL_NAMESPACE);
        instance.setService(name);
        instance.setHealthy(server.isAlive());
        instance.setProtocol(server.getScheme());
        instance.setId(server.getId());
        instance.setHost(server.getHost());
        instance.setPort(server.getPort());
        instance.setZone(server.getZone());
        instance.setWeight(100);
        instances.add(instance);
    }
    serviceInstances = new DefaultServiceInstances(serviceKey, instances);
    return serviceInstances;
}