com.alibaba.cloud.nacos.NacosDiscoveryProperties Java Examples

@Test
@SuppressWarnings("unchecked")
public void testEmptyInstancesReturnsEmptyList() throws Exception {
	NacosDiscoveryProperties nacosDiscoveryProperties = mock(
			NacosDiscoveryProperties.class);

	NamingService namingService = mock(NamingService.class);

	when(nacosDiscoveryProperties.namingServiceInstance()).thenReturn(namingService);
	when(namingService.selectInstances(anyString(), eq("DEFAULT"), eq(true)))
			.thenReturn(null);

	NacosServerList serverList = new NacosServerList(nacosDiscoveryProperties);
	List<NacosServer> servers = serverList.getInitialListOfServers();
	assertThat(servers).isEmpty();
}
 

@Test
public void testUpdateServers() throws Exception {
	ArrayList<Instance> instances = new ArrayList<>();

	HashMap<String, String> map = new HashMap<>();
	map.put("instanceNum", "1");
	instances.add(NacosMockTest.serviceInstance("test-service", true, map));

	NacosDiscoveryProperties nacosDiscoveryProperties = mock(
			NacosDiscoveryProperties.class);

	NamingService namingService = mock(NamingService.class);

	when(nacosDiscoveryProperties.namingServiceInstance()).thenReturn(namingService);
	when(nacosDiscoveryProperties.getGroup()).thenReturn("DEFAULT");
	when(namingService.selectInstances(eq("test-service"), eq("DEFAULT"), eq(true)))
			.thenReturn(instances.stream().filter(Instance::isHealthy)
					.collect(Collectors.toList()));

	IClientConfig clientConfig = mock(IClientConfig.class);
	when(clientConfig.getClientName()).thenReturn("test-service");
	NacosServerList serverList = new NacosServerList(nacosDiscoveryProperties);
	serverList.initWithNiwsConfig(clientConfig);

	List<NacosServer> servers = serverList.getUpdatedListOfServers();
	assertThat(servers).hasSize(1);

	assertThat(servers.get(0).getInstance().isHealthy()).isEqualTo(true);
	assertThat(servers.get(0).getInstance().getMetadata().get("instanceNum"))
			.isEqualTo("1");
}
 

@Bean
@ConditionalOnMissingBean
@ConditionalOnProperty(value = "spring.cloud.nacos.discovery.watch.enabled",
		matchIfMissing = true)
public NacosWatch nacosWatch(NacosDiscoveryProperties nacosDiscoveryProperties,
		ObjectProvider<TaskScheduler> taskScheduler) {
	return new NacosWatch(nacosDiscoveryProperties, taskScheduler);
}
 

@Test
@SuppressWarnings("unchecked")
public void testGetServers() throws Exception {

	ArrayList<Instance> instances = new ArrayList<>();
	instances.add(NacosMockTest.serviceInstance("test-service", false,
			Collections.emptyMap()));

	NacosDiscoveryProperties nacosDiscoveryProperties = mock(
			NacosDiscoveryProperties.class);

	NamingService namingService = mock(NamingService.class);

	when(nacosDiscoveryProperties.namingServiceInstance()).thenReturn(namingService);
	when(nacosDiscoveryProperties.getGroup()).thenReturn("DEFAULT");
	when(nacosDiscoveryProperties.getGroup()).thenReturn("DEFAULT");
	when(namingService.selectInstances(eq("test-service"), eq("DEFAULT"), eq(true)))
			.thenReturn(instances);

	IClientConfig clientConfig = mock(IClientConfig.class);
	when(clientConfig.getClientName()).thenReturn("test-service");
	NacosServerList serverList = new NacosServerList(nacosDiscoveryProperties);
	serverList.initWithNiwsConfig(clientConfig);
	List<NacosServer> servers = serverList.getInitialListOfServers();
	assertThat(servers).hasSize(1);

	servers = serverList.getUpdatedListOfServers();
	assertThat(servers).hasSize(1);
}
 

@Test
public void testDefaultInitialization() {
	contextRunner.run(context -> {
		assertThat(context).hasSingleBean(NacosDiscoveryProperties.class);
		assertThat(context).hasSingleBean(NacosServiceDiscovery.class);
	});
}
 

@Test
public void testGetServices() throws NacosException {
	ListView<String> nacosServices = new ListView<>();

	nacosServices.setData(new LinkedList<>());

	nacosServices.getData().add(serviceName + "1");
	nacosServices.getData().add(serviceName + "2");
	nacosServices.getData().add(serviceName + "3");

	NacosDiscoveryProperties nacosDiscoveryProperties = mock(
			NacosDiscoveryProperties.class);

	NamingService namingService = mock(NamingService.class);

	when(nacosDiscoveryProperties.namingServiceInstance()).thenReturn(namingService);
	when(nacosDiscoveryProperties.getGroup()).thenReturn("DEFAULT");
	when(namingService.getServicesOfServer(eq(1), eq(Integer.MAX_VALUE),
			eq("DEFAULT"))).thenReturn(nacosServices);

	NacosServiceDiscovery serviceDiscovery = new NacosServiceDiscovery(
			nacosDiscoveryProperties);

	List<String> services = serviceDiscovery.getServices();

	assertThat(services.size()).isEqualTo(3);
	assertThat(services.contains(serviceName + "1"));
	assertThat(services.contains(serviceName + "2"));
	assertThat(services.contains(serviceName + "3"));
}
 

@Bean
@ConditionalOnEnabledHealthIndicator("nacos-discovery")
public HealthIndicator nacosDiscoveryHealthIndicator(
		NacosDiscoveryProperties nacosDiscoveryProperties) {
	return new NacosDiscoveryHealthIndicator(
			nacosDiscoveryProperties.namingServiceInstance());
}
 

@Bean
@ConditionalOnMissingBean
@ConditionalOnEnabledEndpoint
public NacosDiscoveryEndpoint nacosDiscoveryEndpoint(
		NacosDiscoveryProperties nacosDiscoveryProperties) {
	return new NacosDiscoveryEndpoint(nacosDiscoveryProperties);
}
 

@Bean
@ConditionalOnBean(AutoServiceRegistrationProperties.class)
public NacosRegistration nacosRegistration(
		NacosDiscoveryProperties nacosDiscoveryProperties,
		ApplicationContext context) {
	return new NacosRegistration(nacosDiscoveryProperties, context);
}
 

@Bean
public NacosServiceRegistry nacosServiceRegistry(
		NacosDiscoveryProperties nacosDiscoveryProperties) {
	return new NacosServiceRegistry(nacosDiscoveryProperties);
}
 

@Test
@SuppressWarnings("unchecked")
public void testGetServersWithInstanceStatus() throws Exception {
	ArrayList<Instance> instances = new ArrayList<>();

	HashMap<String, String> map1 = new HashMap<>();
	map1.put("instanceNum", "1");
	HashMap<String, String> map2 = new HashMap<>();
	map2.put("instanceNum", "2");
	instances.add(NacosMockTest.serviceInstance("test-service", false, map1));
	instances.add(NacosMockTest.serviceInstance("test-service", true, map2));

	NacosDiscoveryProperties nacosDiscoveryProperties = mock(
			NacosDiscoveryProperties.class);

	NamingService namingService = mock(NamingService.class);

	when(nacosDiscoveryProperties.namingServiceInstance()).thenReturn(namingService);
	when(nacosDiscoveryProperties.getGroup()).thenReturn("DEFAULT");
	when(namingService.selectInstances(eq("test-service"), eq("DEFAULT"), eq(true)))
			.thenReturn(instances.stream().filter(Instance::isHealthy)
					.collect(Collectors.toList()));

	IClientConfig clientConfig = mock(IClientConfig.class);
	when(clientConfig.getClientName()).thenReturn("test-service");
	NacosServerList serverList = new NacosServerList(nacosDiscoveryProperties);
	serverList.initWithNiwsConfig(clientConfig);
	List<NacosServer> servers = serverList.getInitialListOfServers();
	assertThat(servers).hasSize(1);

	NacosServer nacosServer = servers.get(0);

	assertThat(nacosServer.getMetaInfo().getInstanceId())
			.isEqualTo(instances.get(1).getInstanceId());
	assertThat(nacosServer.getMetadata()).isEqualTo(map2);
	assertThat(nacosServer.getInstance().isHealthy()).isEqualTo(true);
	assertThat(nacosServer.getInstance().getServiceName()).isEqualTo("test-service");
	assertThat(nacosServer.getInstance().getMetadata().get("instanceNum"))
			.isEqualTo("2");

}
 

public NacosDiscoveryEndpoint(NacosDiscoveryProperties nacosDiscoveryProperties) {
	this.nacosDiscoveryProperties = nacosDiscoveryProperties;
}
 

public NacosDiscoveryProperties getNacosDiscoveryProperties() {
	return nacosDiscoveryProperties;
}
 

public NacosRegistration(NacosDiscoveryProperties nacosDiscoveryProperties,
		ApplicationContext context) {
	this.nacosDiscoveryProperties = nacosDiscoveryProperties;
	this.context = context;
}
 

public NacosServiceDiscovery(NacosDiscoveryProperties discoveryProperties) {
    this.discoveryProperties = discoveryProperties;
}
 

public NacosServiceRegistry(NacosDiscoveryProperties nacosDiscoveryProperties) {
	this.nacosDiscoveryProperties = nacosDiscoveryProperties;
}
 

public NacosServerList(NacosDiscoveryProperties discoveryProperties) {
	this.discoveryProperties = discoveryProperties;
}
 

public NacosWatch(NacosDiscoveryProperties properties, TaskScheduler taskScheduler) {
	this.properties = properties;
	this.taskScheduler = taskScheduler;
}
 

public NacosWatch(NacosDiscoveryProperties properties) {
	this(properties, getTaskScheduler());
}
 

public NacosServiceDiscovery(NacosDiscoveryProperties discoveryProperties) {
	this.discoveryProperties = discoveryProperties;
}
 

@Bean
@ConditionalOnMissingBean
public NacosServiceDiscovery nacosServiceDiscovery(
		NacosDiscoveryProperties discoveryProperties) {
	return new NacosServiceDiscovery(discoveryProperties);
}
 

@Bean
@ConditionalOnMissingBean
public NacosDiscoveryProperties nacosProperties() {
	return new NacosDiscoveryProperties();
}
 

NacosConfiguration(NacosDiscoveryProperties nacosDiscoveryProperties) {
	this.namingService = nacosDiscoveryProperties.namingServiceInstance();
	this.nacosDiscoveryProperties = nacosDiscoveryProperties;
	this.listeningServices = new ConcurrentSkipListSet<>();
}
 

@Bean
public NacosServiceDiscovery nacosServiceDiscovery(NacosDiscoveryProperties discoveryProperties){
    return new NacosServiceDiscovery(discoveryProperties);
}
 

/**
 * The constructor with {@link NacosDiscoveryProperties} bean and the optional.
 * {@link TaskScheduler} bean
 * @param properties {@link NacosDiscoveryProperties} bean
 * @param taskScheduler the optional {@link TaskScheduler} bean
 * @since 2.2.0
 */
public NacosWatch(NacosDiscoveryProperties properties,
		ObjectProvider<TaskScheduler> taskScheduler) {
	this(properties, taskScheduler.getIfAvailable(NacosWatch::getTaskScheduler));
}