How to use

getSync

method

in

org.apache.gobblin.restli.throttling.LimiterServerResource

 @Override
 public void run() {
  try {
   while (true) {
    RequestAndCallback requestAndCallback = MockRequester.this.requestAndCallbackQueue.take();
    long nanoTime = System.nanoTime();
    long delayNanos = requestAndCallback.getProcessAfterNanos() - nanoTime;
    if (delayNanos > 0) {
     Thread.sleep(TimeUnit.NANOSECONDS.toMillis(delayNanos));
    }
    try {
     PermitAllocation allocation =
       MockRequester.this.limiterServer.getSync(new ComplexResourceKey<>(requestAndCallback.getRequest(), new EmptyRecord()));
     Response<PermitAllocation> response = Mockito.mock(Response.class);
     Mockito.when(response.getEntity()).thenReturn(allocation);
     requestAndCallback.getCallback().onSuccess(response);
    } catch (RestLiServiceException rexc) {
     RestLiResponseException returnException = Mockito.mock(RestLiResponseException.class);
     Mockito.when(returnException.getStatus()).thenReturn(rexc.getStatus().getCode());
     requestAndCallback.getCallback().onError(returnException);
    }
   }
  } catch (Throwable t) {
   log.error("Error", t);
   throw new RuntimeException(t);
  }
 }
}

@Test
public void testMetrics() throws Exception {
 ThrottlingGuiceServletConfig guiceServletConfig = new ThrottlingGuiceServletConfig();
 guiceServletConfig.initialize(ConfigFactory.empty());
 Injector injector = guiceServletConfig.getInjector();
 LimiterServerResource limiterServer = injector.getInstance(LimiterServerResource.class);
 PermitRequest request = new PermitRequest();
 request.setPermits(10);
 request.setResource("myResource");
 limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord()));
 limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord()));
 limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord()));
 MetricContext metricContext = limiterServer.metricContext;
 Timer timer = metricContext.timer(LimiterServerResource.REQUEST_TIMER_NAME);
 Assert.assertEquals(timer.getCount(), 3);
}

@Test
public void test() {
 ThrottlingGuiceServletConfig guiceServletConfig = new ThrottlingGuiceServletConfig();
 guiceServletConfig.initialize(ConfigFactory.empty());
 Injector injector = guiceServletConfig.getInjector();
 LimiterServerResource limiterServer = injector.getInstance(LimiterServerResource.class);
 PermitRequest request = new PermitRequest();
 request.setPermits(10);
 request.setResource("myResource");
 PermitAllocation allocation = limiterServer.getSync(new ComplexResourceKey<>(request, new EmptyRecord()));
 Assert.assertTrue(allocation.getPermits() >= 10);
}

private PermitAllocation sendRequestToServer(ThrottlingGuiceServletConfig guiceServletConfig, long permits) {
 return guiceServletConfig.getLimiterResource()
   .getSync(new ComplexResourceKey<>(createPermitRequest(10), new EmptyRecord()));
}

res3request.setResource("res3");
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits(), new Long(20));
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits(), new Long(20));
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits(), new Long(20));
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits(), new Long(20));
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits(), new Long(20));
 limiterServer.getSync(new ComplexResourceKey<>(res1request, new EmptyRecord())).getPermits();
 Assert.fail();
} catch (RestLiServiceException exc) {
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res2request, new EmptyRecord())).getPermits(), new Long(20));
Assert.assertEquals(limiterServer.getSync(new ComplexResourceKey<>(res2request, new EmptyRecord())).getPermits(), new Long(20));
 limiterServer.getSync(new ComplexResourceKey<>(res2request, new EmptyRecord())).getPermits();
 Assert.fail();
} catch (RestLiServiceException exc) {
Assert.assertTrue(limiterServer.getSync(new ComplexResourceKey<>(res3request, new EmptyRecord())).getPermits() >= res3request.getPermits());