Example usage for java.util.concurrent.atomic AtomicInteger get

List of usage examples for java.util.concurrent.atomic AtomicInteger get

Introduction

In this page you can find the example usage for java.util.concurrent.atomic AtomicInteger get.

Prototype

public final int get() 

Source Link

Document

Returns the current value, with memory effects as specified by VarHandle#getVolatile .

Usage

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(" &gt; ")
            .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));
}