use of com.linkedin.d2.balancer.LoadBalancerState.NullStateListenerCallback in project rest.li by linkedin.
the class SimpleLoadBalancerStateTest method testUpdatePartitionDataMap.
@Test(groups = { "small", "back-end" })
public void testUpdatePartitionDataMap() {
reset();
URI uri = URI.create("http://cluster-1/test");
List<String> schemes = new ArrayList<String>();
Map<Integer, PartitionData> partitionDataMap = new HashMap<Integer, PartitionData>(1);
Map<URI, Map<Integer, PartitionData>> uriData = new HashMap<URI, Map<Integer, PartitionData>>();
uriData.put(uri, partitionDataMap);
schemes.add("http");
_state.listenToCluster("cluster-1", new NullStateListenerCallback());
_state.listenToService("service-1", new NullStateListenerCallback());
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-1", "/test", Arrays.asList("random"), Collections.<String, Object>emptyMap(), null, null, schemes, null));
// first we announce uri with empty partition data map
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
TrackerClient client = _state.getClient("service-1", uri);
assertNotNull(client);
assertEquals(client.getUri(), uri);
// tracker client should see empty partition data map
assertTrue(client.getParttitionDataMap().isEmpty());
// then we update this uri to have a non-empty partition data map
partitionDataMap.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
TrackerClient updatedClient = _state.getClient("service-1", uri);
assertNotNull(updatedClient);
// should have got a different tracker client
assertNotSame(client, updatedClient);
assertEquals(updatedClient.getUri(), uri);
// this updated client should have updated partition data map
assertFalse(updatedClient.getParttitionDataMap().isEmpty());
assertEquals(updatedClient.getParttitionDataMap(), partitionDataMap);
}
use of com.linkedin.d2.balancer.LoadBalancerState.NullStateListenerCallback in project rest.li by linkedin.
the class SimpleLoadBalancerStateTest method testGetClientAfterServiceMetadataChange.
@Test(groups = { "small", "back-end" })
public void testGetClientAfterServiceMetadataChange() {
reset();
URI uri = URI.create("http://cluster-1/test");
List<String> schemes = new ArrayList<String>();
Map<Integer, PartitionData> partitionData = new HashMap<Integer, PartitionData>(1);
partitionData.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
Map<URI, Map<Integer, PartitionData>> uriData = new HashMap<URI, Map<Integer, PartitionData>>();
uriData.put(uri, partitionData);
schemes.add("http");
assertNull(_state.getClient("service-1", uri));
// set up state
_state.listenToCluster("cluster-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
_state.listenToService("service-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-1", "/test", Arrays.asList("random"), Collections.<String, Object>emptyMap(), null, null, schemes, null));
assertNull(_state.getClient("service-1", uri));
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
TrackerClient client = _state.getClient("service-1", uri);
assertNotNull(client);
assertEquals(client.getUri(), uri);
//now we publish an event that tells service-1 changes cluster. Now it's hosted in cluster-2
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-2", "/test", Arrays.asList("random"), Collections.<String, Object>emptyMap(), null, null, schemes, null));
//this time, since we haven't listened to cluster-2 and there's no uri in cluster-2, we get no client.
assertNull(_state.getClient("service-1", uri));
//we shouldn't be affected by any update to cluster-1 uris because now service-1 listen to cluster-2
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
assertNull(_state.getClient("service-1", uri));
//now let's create URI for cluster 2 and make the state listen to cluster_2
_state.listenToCluster("cluster-2", new NullStateListenerCallback());
URI uri2 = URI.create("http://cluster-2/test");
Map<Integer, PartitionData> partitionData2 = new HashMap<Integer, PartitionData>(1);
partitionData2.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
Map<URI, Map<Integer, PartitionData>> uriData2 = new HashMap<URI, Map<Integer, PartitionData>>();
uriData2.put(uri2, partitionData2);
//if we start publishing new event to cluster-2 then we should get trackerClient
_uriRegistry.put("cluster-2", new UriProperties("cluster-2", uriData2));
client = _state.getClient("service-1", uri2);
assertNotNull(client);
assertEquals(client.getUri(), uri2);
//just to make sure that any event from cluster-1 doesn't affect us anymore
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
client = _state.getClient("service-1", uri2);
assertNotNull(client);
assertEquals(client.getUri(), uri2);
//if we pass uri from cluster-1, we also get no tracker client
client = _state.getClient("service-1", uri);
assertNull(client);
}
use of com.linkedin.d2.balancer.LoadBalancerState.NullStateListenerCallback in project rest.li by linkedin.
the class SimpleLoadBalancerStateTest method testGetSSLClient.
@Test(groups = { "small", "back-end" })
public void testGetSSLClient() throws URISyntaxException {
reset(true);
URI uri = URI.create("https://cluster-1/test");
List<String> schemes = new ArrayList<String>();
Map<Integer, PartitionData> partitionData = new HashMap<Integer, PartitionData>(1);
partitionData.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
Map<URI, Map<Integer, PartitionData>> uriData = new HashMap<URI, Map<Integer, PartitionData>>();
uriData.put(uri, partitionData);
schemes.add("https");
assertNull(_state.getClient("service-1", uri));
// set up state
_state.listenToCluster("cluster-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
_state.listenToService("service-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
Map<String, Object> transportClientProperties = new HashMap<String, Object>();
transportClientProperties.put(HttpClientFactory.HTTP_SSL_CONTEXT, _sslContext);
transportClientProperties.put(HttpClientFactory.HTTP_SSL_PARAMS, _sslParameters);
transportClientProperties = Collections.unmodifiableMap(transportClientProperties);
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-1", "/test", Arrays.asList("random"), Collections.<String, Object>emptyMap(), transportClientProperties, null, schemes, null));
assertNull(_state.getClient("service-1", uri));
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
TrackerClient client = _state.getClient("service-1", uri);
assertNotNull(client);
assertEquals(client.getUri(), uri);
}
use of com.linkedin.d2.balancer.LoadBalancerState.NullStateListenerCallback in project rest.li by linkedin.
the class SimpleLoadBalancerStateTest method testVersion.
@Test(groups = { "small", "back-end" })
public void testVersion() throws URISyntaxException {
reset();
int expectedVersion = 0;
URI uri = URI.create("http://cluster-1/test");
List<String> schemes = new ArrayList<String>();
Map<Integer, PartitionData> partitionData = new HashMap<Integer, PartitionData>(1);
partitionData.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
Map<URI, Map<Integer, PartitionData>> uriData = new HashMap<URI, Map<Integer, PartitionData>>();
uriData.put(uri, partitionData);
schemes.add("http");
// set up state
_state.listenToCluster("cluster-1", new NullStateListenerCallback());
// one for uri properties onInit, and one for cluster properties onInit
expectedVersion += 2;
assertEquals(_state.getVersion(), expectedVersion);
_state.listenToService("service-1", new NullStateListenerCallback());
// one for service onInit
++expectedVersion;
assertEquals(_state.getVersion(), expectedVersion);
_clusterRegistry.put("cluster-1", new ClusterProperties("cluster-1", schemes));
// one for cluster onAdd
++expectedVersion;
assertEquals(_state.getVersion(), expectedVersion);
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
// one for uri onAdd
++expectedVersion;
assertEquals(_state.getVersion(), expectedVersion);
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-1", "/test", Arrays.asList("random")));
// one for service onAdd
++expectedVersion;
assertEquals(_state.getVersion(), expectedVersion);
// this shouldn't change the version
_state.getClient("cluster-1", uri);
assertEquals(_state.getVersion(), expectedVersion);
// this shouldn't change the version
_state.getStrategy("service-1", "http");
assertEquals(_state.getVersion(), expectedVersion);
}
use of com.linkedin.d2.balancer.LoadBalancerState.NullStateListenerCallback in project rest.li by linkedin.
the class SimpleLoadBalancerStateTest method testListValueInTransportClientProperties.
@Test(groups = { "small", "back-end" })
public void testListValueInTransportClientProperties() throws URISyntaxException {
reset();
URI uri = URI.create("http://cluster-1/test");
List<String> schemes = new ArrayList<String>();
Map<Integer, PartitionData> partitionData = new HashMap<Integer, PartitionData>(1);
partitionData.put(DefaultPartitionAccessor.DEFAULT_PARTITION_ID, new PartitionData(1d));
Map<URI, Map<Integer, PartitionData>> uriData = new HashMap<URI, Map<Integer, PartitionData>>();
uriData.put(uri, partitionData);
schemes.add("http");
assertNull(_state.getClient("service-1", uri));
// set up state
_state.listenToCluster("cluster-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
_state.listenToService("service-1", new NullStateListenerCallback());
assertNull(_state.getClient("service-1", uri));
Map<String, Object> transportClientProperties = new HashMap<String, Object>();
List<String> allowedClientOverrideKeys = new ArrayList<String>();
allowedClientOverrideKeys.add(PropertyKeys.HTTP_REQUEST_TIMEOUT);
allowedClientOverrideKeys.add(PropertyKeys.HTTP_RESPONSE_COMPRESSION_OPERATIONS);
transportClientProperties.put(PropertyKeys.ALLOWED_CLIENT_OVERRIDE_KEYS, allowedClientOverrideKeys);
List<String> compressionOperations = new ArrayList<String>();
compressionOperations.add("get");
compressionOperations.add("batch_get");
compressionOperations.add("get_all");
transportClientProperties.put(PropertyKeys.HTTP_RESPONSE_COMPRESSION_OPERATIONS, compressionOperations);
transportClientProperties = Collections.unmodifiableMap(transportClientProperties);
_serviceRegistry.put("service-1", new ServiceProperties("service-1", "cluster-1", "/test", Arrays.asList("random"), Collections.<String, Object>emptyMap(), transportClientProperties, null, schemes, null));
assertNull(_state.getClient("service-1", uri));
_uriRegistry.put("cluster-1", new UriProperties("cluster-1", uriData));
TrackerClient client = _state.getClient("service-1", uri);
assertNotNull(client);
assertEquals(client.getUri(), uri);
}
Aggregations