Examples with RekognitionAsyncClient software.amazon.awssdk.services.rekognition.RekognitionAsyncClient used on opensource projects

Example 1 with RekognitionAsyncClient

use of software.amazon.awssdk.services.rekognition.RekognitionAsyncClient in project aws-doc-sdk-examples by awsdocs.

the class AnalyzePhotos method DetectLabels.

public ArrayList DetectLabels(byte[] bytes, String key) {
    Region region = Region.US_EAST_2;
    RekognitionAsyncClient rekAsyncClient = RekognitionAsyncClient.builder().credentialsProvider(EnvironmentVariableCredentialsProvider.create()).region(region).build();
    try {
        final AtomicReference<ArrayList<WorkItem>> reference = new AtomicReference<>();
        SdkBytes sourceBytes = SdkBytes.fromByteArray(bytes);
        // Create an Image object for the source image.
        Image souImage = Image.builder().bytes(sourceBytes).build();
        DetectLabelsRequest detectLabelsRequest = DetectLabelsRequest.builder().image(souImage).maxLabels(10).build();
        CompletableFuture<DetectLabelsResponse> futureGet = rekAsyncClient.detectLabels(detectLabelsRequest);
        futureGet.whenComplete((resp, err) -> {
            try {
                if (resp != null) {
                    List<Label> labels = resp.labels();
                    System.out.println("Detected labels for the given photo");
                    ArrayList list = new ArrayList<WorkItem>();
                    WorkItem item;
                    for (Label label : labels) {
                        item = new WorkItem();
                        // identifies the photo
                        item.setKey(key);
                        item.setConfidence(label.confidence().toString());
                        item.setName(label.name());
                        list.add(item);
                    }
                    reference.set(list);
                } else {
                    err.printStackTrace();
                }
            } finally {
                // Only close the client when you are completely done with it
                rekAsyncClient.close();
            }
        });
        futureGet.join();
        // Use the AtomicReference object to return the ArrayList<WorkItem> collection.
        return reference.get();
    } catch (RekognitionException e) {
        System.out.println(e.getMessage());
        System.exit(1);
    }
    return null;
}