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

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

Introduction

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

Prototype

public final boolean get() 

Source Link

Document

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

Usage

From source file:com.alibaba.wasp.meta.FMetaServicesImplWithoutRetry.java

public boolean isTableAvailable(final byte[] tableName) throws IOException {
    final AtomicBoolean available = new AtomicBoolean(true);
    final AtomicInteger entityGroupCount = new AtomicInteger(0);
    FMetaScanner.MetaScannerVisitor visitor = new FMetaScanner.MetaScannerVisitorBase() {
        @Override/* w  ww  .  j  a va  2 s  . c  om*/
        public boolean processRow(Result row) throws IOException {
            EntityGroupInfo info = FMetaScanner.getEntityGroupInfo(row);
            if (info != null) {
                if (Bytes.equals(tableName, info.getTableName())) {
                    ServerName sn = ServerName.getServerName(row);
                    if (sn == null) {
                        available.set(false);
                        return false;
                    }
                    entityGroupCount.incrementAndGet();
                }
            }
            return true;
        }
    };
    FMetaScanner.metaScan(getConf(), visitor);
    return available.get() && (entityGroupCount.get() > 0);
}

From source file:com.jkoolcloud.tnt4j.streams.parsers.GenericActivityParser.java

/**
 * Gets field value from raw data location and formats it according locator definition.
 *
 * @param locator//from  w w w . j ava2 s .c  o  m
 *            activity field locator
 * @param cData
 *            parsing context data package
 * @return value formatted based on locator definition or {@code null} if locator is not defined
 * @throws ParseException
 *             if exception occurs applying locator format properties to specified value
 * @see ActivityFieldLocator#formatValue(Object)
 */
protected Object getLocatorValue(ActivityFieldLocator locator, ActivityContext cData) throws ParseException {
    Object val = null;
    if (locator != null) {
        String locStr = locator.getLocator();
        AtomicBoolean formattingNeeded = new AtomicBoolean(true);
        if (StringUtils.isNotEmpty(locStr)) {
            if (locator.getBuiltInType() == ActivityFieldLocatorType.StreamProp) {
                val = cData.getStream().getProperty(locStr);
            } else if (locator.getBuiltInType() == ActivityFieldLocatorType.Cache) {
                val = Utils.simplifyValue(StreamsCache.getValue(cData.getActivity(), locStr, getName()));
            } else if (locator.getBuiltInType() == ActivityFieldLocatorType.Activity) {
                val = cData.getActivity().getFieldValue(locator.getLocator());
            } else {
                val = resolveLocatorValue(locator, cData, formattingNeeded);
                // logger().log(val == null && !locator.isOptional() ? OpLevel.WARNING : OpLevel.TRACE,
                logger().log(OpLevel.TRACE, StreamsResources.getString(StreamsResources.RESOURCE_BUNDLE_NAME,
                        "ActivityParser.locator.resolved"), locStr, toString(val));
            }
        }

        if (formattingNeeded.get()) {
            val = locator.formatValue(val);
        }

        try {
            val = locator.transformValue(val);
        } catch (Exception exc) {
            logger().log(OpLevel.WARNING, StreamsResources.getString(StreamsResources.RESOURCE_BUNDLE_NAME,
                    "ActivityParser.transformation.failed"), locStr, toString(val), exc);
        }

        try {
            boolean filteredOut = locator.filterValue(val);

            if (filteredOut) {
                val = null;
            }
        } catch (Exception exc) {
            logger().log(OpLevel.WARNING, StreamsResources.getString(StreamsResources.RESOURCE_BUNDLE_NAME,
                    "ActivityParser.field.filtering.failed"), locStr, toString(val), exc);
        }
    }
    return val;
}

From source file:com.facebook.buck.artifact_cache.ArtifactUploaderTest.java

@Test
public void testPerformUploadToArtifactCache() throws IOException {

    FakeProjectFilesystem filesystem = new FakeProjectFilesystem();

    byte[] contents = "contents".getBytes();

    Path file = Paths.get("file");
    filesystem.writeBytesToPath(contents, file);

    Path dirFile = Paths.get("dir", "file");
    filesystem.createParentDirs(dirFile);
    filesystem.writeBytesToPath(contents, dirFile);

    Path metadataFile = Paths.get("buck-out", "bin", "foo", ".bar", "metadata", "artifact", "metadata");
    filesystem.createParentDirs(metadataFile);
    filesystem.writeBytesToPath(contents, metadataFile);

    Path dir = Paths.get("buck-out", "bin", "foo", ".bar/");
    filesystem.mkdirs(dir);//from ww w. j a v  a 2 s .co  m

    AtomicBoolean stored = new AtomicBoolean(false);
    ArtifactCache cache = new NoopArtifactCache() {
        @Override
        public CacheReadMode getCacheReadMode() {
            return CacheReadMode.READWRITE;
        }

        @Override
        public ListenableFuture<Void> store(ArtifactInfo info, BorrowablePath output) {
            stored.set(true);

            // Verify the build metadata.
            assertThat(info.getMetadata().get("build-metadata"), Matchers.equalTo("build-metadata"));

            // Unarchive file.
            final ImmutableMap<String, byte[]> archiveContents;
            try {
                archiveContents = TarInspector.readTarZst(output.getPath());
            } catch (IOException | CompressorException e) {
                fail(e.getMessage());
                return Futures.immediateFuture(null);
            }

            // Verify archive contents.
            assertEquals(ImmutableSet.of("buck-out/bin/foo/.bar/", "dir/file", "file",
                    "buck-out/bin/foo/.bar/metadata/artifact/metadata"), archiveContents.keySet());
            assertArrayEquals(contents, archiveContents.get("file"));
            assertArrayEquals(contents, archiveContents.get("dir/file"));
            assertArrayEquals(contents,
                    archiveContents.get("buck-out/bin/foo/.bar/metadata/artifact/metadata"));
            return Futures.immediateFuture(null);
        }
    };

    ArtifactUploader.performUploadToArtifactCache(ImmutableSet.of(new RuleKey("aa")), cache,
            BuckEventBusForTests.newInstance(),
            ImmutableMap.of("metadata", "metadata", "build-metadata", "build-metadata"),
            ImmutableSortedSet.of(dir, file, dirFile, metadataFile), BUILD_TARGET, filesystem, 1000);

    assertTrue(stored.get());
}

From source file:io.pravega.client.stream.impl.ControllerImplTest.java

@Test
public void testParallelCreateStream() throws Exception {
    final ExecutorService executorService = Executors.newFixedThreadPool(10);
    Semaphore createCount = new Semaphore(-19);
    AtomicBoolean success = new AtomicBoolean(true);
    for (int i = 0; i < 10; i++) {
        executorService.submit(() -> {
            for (int j = 0; j < 2; j++) {
                try {
                    CompletableFuture<Boolean> createStreamStatus;
                    createStreamStatus = controllerClient
                            .createStream(StreamConfiguration.builder().streamName("streamparallel")
                                    .scope("scope1").scalingPolicy(ScalingPolicy.fixed(1)).build());
                    log.info("{}", createStreamStatus.get());
                    assertTrue(createStreamStatus.get());
                    createCount.release();
                } catch (Exception e) {
                    log.error("Exception when creating stream: {}", e);

                    // Don't wait for other threads to complete.
                    success.set(false);/*  w  w w  .j  a va  2 s.  c om*/
                    createCount.release(20);
                }
            }
        });
    }
    createCount.acquire();
    executorService.shutdownNow();
    assertTrue(success.get());
}

From source file:com.facebook.RequestTests.java

@LargeTest
public void testShareOpenGraphContentWithBadType() throws Exception {
    ShareOpenGraphObject ogObject = new ShareOpenGraphObject.Builder().putString("og:title", "a title")
            .putString("og:type", TEST_OG_OBJECT_TYPE).putString("og:description", "a description").build();

    ShareOpenGraphAction ogAction = new ShareOpenGraphAction.Builder()
            .setActionType(TEST_OG_ACTION_TYPE + "bad").putObject("test", ogObject).build();

    ShareOpenGraphContent content = new ShareOpenGraphContent.Builder().setAction(ogAction)
            .setPreviewPropertyName("test").build();

    final ShareApi shareApi = new ShareApi(content);
    final AtomicReference<String> actionId = new AtomicReference<>(null);
    final AtomicBoolean errorOccurred = new AtomicBoolean(false);

    getActivity().runOnUiThread(new Runnable() {
        @Override/*  ww  w  .j a  v a  2  s  . c o m*/
        public void run() {
            shareApi.share(new FacebookCallback<Sharer.Result>() {
                @Override
                public void onSuccess(Sharer.Result result) {
                    actionId.set(result.getPostId());
                    notifyShareFinished();
                }

                @Override
                public void onCancel() {
                    notifyShareFinished();
                }

                @Override
                public void onError(FacebookException error) {
                    errorOccurred.set(true);
                    notifyShareFinished();
                }

                private void notifyShareFinished() {
                    synchronized (shareApi) {
                        shareApi.notifyAll();
                    }
                }
            });
        }
    });

    synchronized (shareApi) {
        shareApi.wait(REQUEST_TIMEOUT_MILLIS);
    }
    assertNull(actionId.get());
    assertTrue(errorOccurred.get());
}

From source file:org.apache.hadoop.yarn.server.nodemanager.TestNodeStatusUpdater.java

@Test
public void testConcurrentAccessToSystemCredentials() {
    final Map<ApplicationId, ByteBuffer> testCredentials = new HashMap<>();
    ByteBuffer byteBuffer = ByteBuffer.wrap(new byte[300]);
    ApplicationId applicationId = ApplicationId.newInstance(123456, 120);
    testCredentials.put(applicationId, byteBuffer);

    final List<Throwable> exceptions = Collections.synchronizedList(new ArrayList<Throwable>());

    final int NUM_THREADS = 10;
    final CountDownLatch allDone = new CountDownLatch(NUM_THREADS);
    final ExecutorService threadPool = Executors.newFixedThreadPool(NUM_THREADS);

    final AtomicBoolean stop = new AtomicBoolean(false);

    try {/* w w  w  .j  av  a  2s. c  o m*/
        for (int i = 0; i < NUM_THREADS; i++) {
            threadPool.submit(new Runnable() {
                @Override
                public void run() {
                    try {
                        for (int i = 0; i < 100 && !stop.get(); i++) {
                            NodeHeartbeatResponse nodeHeartBeatResponse = newNodeHeartbeatResponse(0,
                                    NodeAction.NORMAL, null, null, null, null, 0);
                            nodeHeartBeatResponse.setSystemCredentialsForApps(testCredentials);
                            NodeHeartbeatResponseProto proto = ((NodeHeartbeatResponsePBImpl) nodeHeartBeatResponse)
                                    .getProto();
                            Assert.assertNotNull(proto);
                        }
                    } catch (Throwable t) {
                        exceptions.add(t);
                        stop.set(true);
                    } finally {
                        allDone.countDown();
                    }
                }
            });
        }

        int testTimeout = 2;
        Assert.assertTrue("Timeout waiting for more than " + testTimeout + " " + "seconds",
                allDone.await(testTimeout, TimeUnit.SECONDS));
    } catch (InterruptedException ie) {
        exceptions.add(ie);
    } finally {
        threadPool.shutdownNow();
    }
    Assert.assertTrue("Test failed with exception(s)" + exceptions, exceptions.isEmpty());
}

From source file:org.pentaho.di.trans.steps.httppost.HTTPPOSTIT.java

public void testServerReturnsCorrectlyEncodedParams(String testString, String testCharset) throws Exception {
    AtomicBoolean testStatus = new AtomicBoolean();
    startHttpServer(getEncodingCheckingHandler(testString, testCharset, testStatus));
    HTTPPOSTData data = new HTTPPOSTData();
    RowMeta meta = new RowMeta();
    meta.addValueMeta(new ValueMetaString("fieldName"));
    HTTPPOSTHandler httpPost = new HTTPPOSTHandler(stepMockHelper.stepMeta, data, 0, stepMockHelper.transMeta,
            stepMockHelper.trans, false);
    RowMetaInterface inputRowMeta = mock(RowMetaInterface.class);
    httpPost.setInputRowMeta(inputRowMeta);
    httpPost.row = new Object[] { testString };
    when(inputRowMeta.clone()).thenReturn(inputRowMeta);
    when(inputRowMeta.getString(httpPost.row, 0)).thenReturn(testString);
    when(stepMockHelper.processRowsStepMetaInterface.getUrl()).thenReturn(HTTP_LOCALHOST_9998);
    when(stepMockHelper.processRowsStepMetaInterface.getQueryField()).thenReturn(new String[] {});
    when(stepMockHelper.processRowsStepMetaInterface.getArgumentField())
            .thenReturn(new String[] { "testBodyField" });
    when(stepMockHelper.processRowsStepMetaInterface.getArgumentParameter())
            .thenReturn(new String[] { "testBodyParam" });
    when(stepMockHelper.processRowsStepMetaInterface.getArgumentHeader()).thenReturn(new boolean[] { false });
    when(stepMockHelper.processRowsStepMetaInterface.getFieldName()).thenReturn("ResultFieldName");
    when(stepMockHelper.processRowsStepMetaInterface.getEncoding()).thenReturn(testCharset);
    httpPost.init(stepMockHelper.processRowsStepMetaInterface, data);
    Assert.assertTrue(httpPost.processRow(stepMockHelper.processRowsStepMetaInterface, data));
    Assert.assertTrue(testStatus.get(), "Test failed");
}

From source file:org.apache.james.protocols.smtp.AbstractSMTPServerTest.java

@Test
public void testDisconnectHandler() throws Exception {

    final AtomicBoolean called = new AtomicBoolean(false);
    DisconnectHandler<SMTPSession> handler = new DisconnectHandler<SMTPSession>() {

        @Override//from ww w  . j ava2 s. c  o  m
        public void init(Configuration config) throws ConfigurationException {

        }

        @Override
        public void destroy() {

        }

        public void onDisconnect(SMTPSession session) {
            called.set(true);
        }
    };

    InetSocketAddress address = new InetSocketAddress("127.0.0.1", TestUtils.getFreePort());

    ProtocolServer server = null;
    try {
        server = createServer(createProtocol(handler), address);
        server.bind();

        SMTPClient client = createClient();
        client.connect(address.getAddress().getHostAddress(), address.getPort());
        assertTrue("Reply=" + client.getReplyString(), SMTPReply.isPositiveCompletion(client.getReplyCode()));

        client.disconnect();

        Thread.sleep(1000);
        assertTrue(called.get());

    } finally {
        if (server != null) {
            server.unbind();
        }
    }

}

From source file:org.apache.pulsar.broker.service.persistent.PersistentTopic.java

private boolean hasLocalProducers() {
    AtomicBoolean foundLocal = new AtomicBoolean(false);
    producers.forEach(producer -> {/*from w  ww. j  a  va 2s  .  c om*/
        if (!producer.isRemote()) {
            foundLocal.set(true);
        }
    });

    return foundLocal.get();
}

From source file:org.apache.pulsar.broker.service.persistent.PersistentTopic.java

private boolean hasRemoteProducers() {
    AtomicBoolean foundRemote = new AtomicBoolean(false);
    producers.forEach(producer -> {//w w w.  j  a v a 2  s .c  o  m
        if (producer.isRemote()) {
            foundRemote.set(true);
        }
    });

    return foundRemote.get();
}