use of com.tencent.polaris.api.rpc.ServiceCallResult in project polaris-java by polarismesh.
the class CircuitBreakerTest method testCircuitBreakByErrorRate.
@Test
public void testCircuitBreakByErrorRate() {
Configuration configuration = TestUtils.configWithEnvAddress();
try (ConsumerAPI consumerAPI = DiscoveryAPIFactory.createConsumerAPIByConfig(configuration)) {
GetInstancesRequest getInstancesRequest = new GetInstancesRequest();
getInstancesRequest.setNamespace(NAMESPACE_TEST);
getInstancesRequest.setService(SERVICE_CIRCUIT_BREAKER);
InstancesResponse instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
Instance instanceToLimit = instances.getInstances()[1];
// report 60 fail in 500ms
for (int i = 0; i < 60; ++i) {
ServiceCallResult result = instanceToResult(instanceToLimit);
result.setDelay(1000L);
if (i % 2 == 0) {
result.setRetCode(0);
result.setRetStatus(RetStatus.RetSuccess);
Utils.sleepUninterrupted(1);
} else {
result.setRetCode(-1);
result.setRetStatus(RetStatus.RetFail);
}
consumerAPI.updateServiceCallResult(result);
Utils.sleepUninterrupted(1);
}
Utils.sleepUninterrupted(1000);
instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT - 1, instances.getInstances().length);
Instance[] instanceArray = instances.getInstances();
boolean exists = false;
for (int i = 0; i < instanceArray.length; ++i) {
if (instanceArray[i].getId().equals(instanceToLimit.getId())) {
exists = true;
}
}
Assert.assertFalse(exists);
Utils.sleepUninterrupted(10000);
// default halfopen pass 3 success
int requestCountAfterHalfOpen = configuration.getConsumer().getCircuitBreaker().getRequestCountAfterHalfOpen();
for (int i = 0; i < requestCountAfterHalfOpen; i++) {
ServiceCallResult result = instanceToResult(instanceToLimit);
result.setRetCode(-1);
result.setRetStatus(RetStatus.RetSuccess);
consumerAPI.updateServiceCallResult(result);
Utils.sleepUninterrupted(200);
consumerAPI.updateServiceCallResult(result);
}
LOG.info("start to test half open to close");
Utils.sleepUninterrupted(1000);
instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
}
}
use of com.tencent.polaris.api.rpc.ServiceCallResult in project polaris-java by polarismesh.
the class CircuitBreakerTest method testUpdateServiceCallResult.
@Test
public void testUpdateServiceCallResult() {
Configuration configuration = TestUtils.configWithEnvAddress();
try (ConsumerAPI consumerAPI = DiscoveryAPIFactory.createConsumerAPIByConfig(configuration)) {
int index = 1;
GetInstancesRequest req = new GetInstancesRequest();
req.setNamespace(NAMESPACE_TEST);
req.setService(SERVICE_CIRCUIT_BREAKER);
InstancesResponse instances = consumerAPI.getInstances(req);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
Instance instanceToLimit = instances.getInstances()[index];
ServiceCallResult result = instanceToResult(instanceToLimit);
result.setRetCode(-1);
result.setDelay(1000L);
result.setRetStatus(RetStatus.RetFail);
consumerAPI.updateServiceCallResult(result);
}
}
use of com.tencent.polaris.api.rpc.ServiceCallResult in project polaris-java by polarismesh.
the class CircuitBreakerTest method testCircuitBreakByErrorCount.
@Test
public void testCircuitBreakByErrorCount() {
Configuration configuration = TestUtils.configWithEnvAddress();
try (ConsumerAPI consumerAPI = DiscoveryAPIFactory.createConsumerAPIByConfig(configuration)) {
Utils.sleepUninterrupted(10000);
Assert.assertNotNull(consumerAPI);
GetInstancesRequest getInstancesRequest = new GetInstancesRequest();
getInstancesRequest.setNamespace(NAMESPACE_TEST);
getInstancesRequest.setService(SERVICE_CIRCUIT_BREAKER);
InstancesResponse instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
Instance instanceToLimit = instances.getInstances()[1];
// report 60 fail in 500ms
for (int i = 0; i < 60; ++i) {
ServiceCallResult result = instanceToResult(instanceToLimit);
result.setRetCode(-1);
result.setDelay(1000L);
result.setRetStatus(RetStatus.RetFail);
consumerAPI.updateServiceCallResult(result);
if (i % 10 == 0) {
Utils.sleepUninterrupted(1);
}
}
Utils.sleepUninterrupted(1000);
instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT - 1, instances.getInstances().length);
Instance[] instanceArray = instances.getInstances();
boolean exists = false;
for (int i = 0; i < instanceArray.length; ++i) {
if (instanceArray[i].getId().equals(instanceToLimit.getId())) {
exists = true;
}
}
Assert.assertFalse(exists);
LOG.info("start to test half open by error rate");
Utils.sleepUninterrupted(10000);
instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
for (Instance instance : instances.getInstances()) {
CircuitBreakerStatus circuitBreakerStatus = instance.getCircuitBreakerStatus();
if (null != circuitBreakerStatus && circuitBreakerStatus.getStatus() == CircuitBreakerStatus.Status.HALF_OPEN) {
LOG.info("half open instance is {}", instance);
}
}
// default halfopen pass 3 success
int requestCountAfterHalfOpen = configuration.getConsumer().getCircuitBreaker().getRequestCountAfterHalfOpen();
for (int i = 0; i < requestCountAfterHalfOpen; i++) {
ServiceCallResult result = instanceToResult(instanceToLimit);
result.setRetCode(-1);
result.setRetStatus(RetStatus.RetSuccess);
consumerAPI.updateServiceCallResult(result);
Utils.sleepUninterrupted(200);
consumerAPI.updateServiceCallResult(result);
}
LOG.info("start to test half open to close");
Utils.sleepUninterrupted(1000);
instances = consumerAPI.getInstances(getInstancesRequest);
Assert.assertEquals(MAX_COUNT, instances.getInstances().length);
}
}
use of com.tencent.polaris.api.rpc.ServiceCallResult in project polaris-java by polarismesh.
the class PrometheusPushHandlerTest method testExpiredDataClean.
@Test
public void testExpiredDataClean() throws InterruptedException {
int count = 5;
StatInfo statInfo = new StatInfo();
ServiceCallResult callResult = mockServiceCallResult();
statInfo.setRouterGauge(callResult);
batchDone(() -> handler.handle(statInfo), count);
// mock push
LOG.info("first mock push finish...");
Thread.sleep(pushInterval + 1000);
Double result = getServiceCallTotalResult(callResult);
Assert.assertEquals(new Double(count), result);
// mock next push
LOG.info("second mock push finish...");
Thread.sleep(pushInterval + 1000);
result = getServiceCallTotalResult(callResult);
Assert.assertEquals(new Double(0), result);
LOG.info("mock sleep {} times end...", REVISION_MAX_SCOPE);
Thread.sleep(pushInterval * REVISION_MAX_SCOPE + 1000);
result = getServiceCallTotalResult(callResult);
Assert.assertNull(result);
// stop handle
handler.stopHandle();
}
use of com.tencent.polaris.api.rpc.ServiceCallResult in project polaris-java by polarismesh.
the class PrometheusPushHandlerTest method testServiceCallTotalStrategy.
@Test
public void testServiceCallTotalStrategy() throws InterruptedException {
StatInfo statInfo = new StatInfo();
ServiceCallResult callResult = mockServiceCallResult();
statInfo.setRouterGauge(callResult);
int count = 10;
batchDone(() -> handler.handle(statInfo), count);
// mock pushing
Thread.sleep(pushInterval + 1000);
handler.stopHandle();
Double result = getServiceCallTotalResult(callResult);
Assert.assertEquals(new Double(count), result);
}
Aggregations