List of usage examples for java.util.concurrent.atomic AtomicInteger get
public final int get()
From source file:com.cloudera.livy.client.local.rpc.TestRpc.java
@Test public void testCloseListener() throws Exception { RpcServer server = autoClose(new RpcServer(emptyConfig)); Rpc[] rpcs = createRpcConnection(server); Rpc client = rpcs[1];//from w w w. j av a2 s . com final AtomicInteger closeCount = new AtomicInteger(); client.addListener(new Rpc.Listener() { @Override public void rpcClosed(Rpc rpc) { closeCount.incrementAndGet(); } }); client.close(); client.close(); assertEquals(1, closeCount.get()); }
From source file:org.apache.bookkeeper.metadata.etcd.helpers.HelpersTest.java
private void testKeyIterator(int numKeys, int batchSize) throws Exception { for (int i = 0; i < numKeys; i++) { String key = getKey(scope, i); ByteSequence keyBs = ByteSequence.fromString(key); result(etcdClient.getKVClient().put(keyBs, keyBs)); }/* w w w .j a va2 s . c o m*/ KeyStream<Integer> ks = openKeyStream(batchSize); KeyIterator<Integer> ki = new KeyIterator<>(ks); AtomicInteger numReceived = new AtomicInteger(0); while (ki.hasNext()) { List<Integer> values = ki.next(); log.info("Received values : {}", values); if (values.isEmpty()) { break; } for (int value : values) { assertEquals(numReceived.getAndIncrement(), value); } } assertEquals(numKeys, numReceived.get()); }
From source file:org.apache.bookkeeper.metadata.etcd.helpers.HelpersTest.java
private void testKeyStream(int numKeys, int batchSize) throws Exception { for (int i = 0; i < numKeys; i++) { String key = getKey(scope, i); ByteSequence keyBs = ByteSequence.fromBytes(key.getBytes(UTF_8)); result(etcdClient.getKVClient().put(keyBs, keyBs)); }//from w ww. ja v a 2 s. c o m KeyStream<Integer> ks = openKeyStream(batchSize); AtomicInteger numReceived = new AtomicInteger(0); while (true) { List<Integer> values = result(ks.readNext()); log.info("Received values : {}", values); if (values.isEmpty()) { break; } for (int value : values) { assertEquals(numReceived.getAndIncrement(), value); } } assertEquals(numKeys, numReceived.get()); }
From source file:anhttpclient.AnhttpclientTest.java
@Test public void testCookies() throws Exception { final String cookieName1 = "PREF"; final String cookieValue1 = "ID=120db22e4e2810cb:FF=0:TM=1318488512:LM=1318488512:S=tTZoDOAsNIbsiH4R"; final String cookieParams1 = "expires=Sat, 12-Oct-2099 06:48:32 GMT; path=/; domain=localhost"; final String cookieName2 = "NID"; final String cookieValue2 = "52=fa8SxaS_YoBUQ7jLIy9bp14biO"; final String cookieParams2 = "expires=Fri, 13-Apr-2099 06:48:32 GMT; path=/; domain=localhost; HttpOnly"; final AtomicInteger requestCount = new AtomicInteger(0); server.addHandler("/cookies", new ByteArrayHandlerAdapter() { public byte[] getResponseAsByteArray(HttpRequestContext httpRequestContext) { requestCount.addAndGet(1);/*from w w w.j a v a2 s . co m*/ if (requestCount.get() == 1) { setResponseHeader("Set-Cookie", cookieName1 + "=" + cookieValue1 + "; " + cookieParams1, httpRequestContext); } else if (requestCount.get() == 2) { assertEquals(cookieName1 + "=" + cookieValue1, httpRequestContext.getRequestHeaders().get("Cookie").get(0)); setResponseHeader("Set-Cookie", cookieName2 + "=" + cookieValue2 + "; " + cookieParams2, httpRequestContext); } else { //Sends only one cookie somewhy //assertEquals(cookieName1 + "; " + cookieName2, httpRequestContext.getRequestHeaders().get("Cookie")); } return "OK".getBytes(); } }); WebRequest req = new HttpGetWebRequest(server.getBaseUrl() + "/cookies"); wb.getResponse(req); Cookie cookie1 = wb.getCookieByName(cookieName1); assertEquals("Invalid cookie 1", cookieName1, cookie1.getName()); assertEquals("Invalid cookie 1", cookieValue1, cookie1.getValue()); wb.getResponse(req); cookie1 = wb.getCookieByName(cookieName1); assertEquals("Invalid cookie 1", cookieName1, cookie1.getName()); assertEquals("Invalid cookie 1", cookieValue1, cookie1.getValue()); Cookie cookie2 = wb.getCookieByName(cookieName2); assertEquals("Invalid cookie 2", cookieName2, cookie2.getName()); assertEquals("Invalid cookie2", cookieValue2, cookie2.getValue()); wb.getResponse(req); }
From source file:com.android.tools.swing.ui.NavigationComponent.java
private void updateText() { if (myItemStack.isEmpty()) { setText(""); return;// ww w .j av a 2s .c om } if (myItemStack.size() == 1 && hasRootItem && !myDisplaySingleRoot) { setText(""); return; } final AtomicInteger id = new AtomicInteger(myItemStack.size() - 1); String text = Joiner.on(" > ") .join(Iterators.transform(myItemStack.descendingIterator(), new Function<T, String>() { @Override public String apply(T input) { // Do not display link for the last element. if (id.get() == 0) { return input.getDisplayText(); } return String.format("<a href=\"%d\">%s</a>", id.getAndDecrement(), input.getDisplayText()); } @Override public boolean equals(Object object) { return false; } })); setText(text); }
From source file:com.amazonaws.http.DelegatingDnsResolverTest.java
@Test public void testDelegatingDnsResolverCallsResolveOnDelegate() throws Exception { final AtomicInteger timesCalled = new AtomicInteger(); DnsResolver delegate = new DnsResolver() { @Override/*from ww w . j av a 2s . c om*/ public InetAddress[] resolve(String host) throws UnknownHostException { timesCalled.incrementAndGet(); return new InetAddress[0]; } }; org.apache.http.conn.DnsResolver resolver = new DelegatingDnsResolver(delegate); resolver.resolve("localhost"); assertEquals("Delegate Resolver should have been executed", 1, timesCalled.get()); }
From source file:de.bund.bfr.math.LeastSquaresOptimization.java
@Override public Result optimize(int nParameterSpace, int nOptimizations, boolean stopWhenSuccessful, Map<String, Double> minStartValues, Map<String, Double> maxStartValues, int maxIterations, DoubleConsumer progressListener, ExecutionContext exec) throws CanceledExecutionException { if (exec != null) { exec.checkCanceled();/*from w ww . j a v a 2 s. c o m*/ } progressListener.accept(0.0); List<ParamRange> ranges = MathUtils.getParamRanges(parameters, minStartValues, maxStartValues, nParameterSpace); RealVector targetVector = new ArrayRealVector(Doubles.toArray(targetValues)); List<StartValues> startValuesList = MathUtils.createStartValuesList(ranges, nOptimizations, values -> targetVector .getDistance(new ArrayRealVector(optimizerFunction.value(Doubles.toArray(values)))), progress -> progressListener.accept(0.5 * progress), exec); LevenbergMarquardtOptimizer optimizer = new LevenbergMarquardtOptimizer(); Result result = new Result(); AtomicInteger count = new AtomicInteger(0); for (StartValues startValues : startValuesList) { if (exec != null) { exec.checkCanceled(); } progressListener.accept(0.5 * count.get() / startValuesList.size() + 0.5); try { LeastSquaresBuilder builder = createLeastSquaresBuilder(startValues.getValues(), maxIterations); builder.checker((iteration, previous, current) -> { double currentProgress = (double) iteration / (double) maxIterations; if (exec != null) { try { exec.checkCanceled(); } catch (CanceledExecutionException e) { return true; } } progressListener.accept(0.5 * (count.get() + currentProgress) / startValuesList.size() + 0.5); return iteration == maxIterations; }); LeastSquaresOptimizer.Optimum optimizerResults = optimizer.optimize(builder.build()); if (exec != null) { exec.checkCanceled(); } double cost = optimizerResults.getCost(); if (result.sse == null || cost * cost < result.sse) { result = getResults(optimizerResults); if (result.sse == 0.0) { break; } if (result.r2 != null && result.r2 > 0.0 && stopWhenSuccessful) { break; } } } catch (TooManyEvaluationsException | TooManyIterationsException | ConvergenceException e) { } count.incrementAndGet(); } return result; }
From source file:com.cloudera.livy.rsc.rpc.TestRpc.java
@Test public void testCloseListener() throws Exception { RpcServer server = autoClose(new RpcServer(emptyConfig)); Rpc[] rpcs = createRpcConnection(server); Rpc client = rpcs[1];// w w w. java2 s . c o m final AtomicInteger closeCount = new AtomicInteger(); Utils.addListener(client.getChannel().closeFuture(), new FutureListener<Void>() { @Override public void onSuccess(Void unused) { closeCount.incrementAndGet(); } }); client.close(); client.close(); assertEquals(1, closeCount.get()); }
From source file:com.palantir.docker.compose.DockerComposeRuleShould.java
@Test public void pass_wait_for_service_when_check_is_true() throws IOException, InterruptedException { AtomicInteger timesCheckCalled = new AtomicInteger(0); withComposeExecutableReturningContainerFor("db"); HealthCheck<Container> checkCalledOnce = (container) -> SuccessOrFailure .fromBoolean(timesCheckCalled.incrementAndGet() == 1, "not called once yet"); DockerComposeRule.builder().from(rule).waitingForService("db", checkCalledOnce).build().before(); assertThat(timesCheckCalled.get(), is(1)); }
From source file:com.palantir.docker.compose.DockerComposeRuleShould.java
@Test public void pass_wait_for_service_when_check_is_true_after_being_false() throws IOException, InterruptedException { AtomicInteger timesCheckCalled = new AtomicInteger(0); withComposeExecutableReturningContainerFor("db"); HealthCheck<Container> checkCalledTwice = (container) -> SuccessOrFailure .fromBoolean(timesCheckCalled.incrementAndGet() == 2, "not called twice yet"); DockerComposeRule.builder().from(rule).waitingForService("db", checkCalledTwice).build().before(); assertThat(timesCheckCalled.get(), is(2)); }