List of usage examples for java.util.concurrent.atomic AtomicBoolean get
public final boolean get()
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(); }