Examples with SubscribeToIoTCoreResponseHandler software.amazon.awssdk.aws.greengrass.SubscribeToIoTCoreResponseHandler used on opensource projects

Example 1 with SubscribeToIoTCoreResponseHandler

use of software.amazon.awssdk.aws.greengrass.SubscribeToIoTCoreResponseHandler in project aws-iot-greengrass-edge-connector-for-kinesis-video-stream by awslabs.

the class VideoUploadRequestHandler method subscribeToMqttTopic.

/**
 * Subscribes to Specified MQTT Topic for Video Upload Request Events.
 *
 * @param mqttTopic - Name of the MQTT topic where video upload request will be sent
 * @param event     - Callback event for onStart or onError
 */
public void subscribeToMqttTopic(String mqttTopic, VideoUploadRequestEvent event) {
    StreamResponseHandler<IoTCoreMessage> streamResponseHandler;
    streamResponseHandler = new StreamResponseHandler<IoTCoreMessage>() {

        @Override
        public void onStreamEvent(IoTCoreMessage ioTCoreMessage) {
            log.info("onStreamEvent");
            MQTTMessage mqttMessage = ioTCoreMessage.getMessage();
            if (mqttMessage == null) {
                log.error("Empty MQTT Message Received");
                return;
            }
            String payload = new String(mqttMessage.getPayload(), StandardCharsets.UTF_8);
            VideoUploadRequestMessage videoUploadRequestMessage = JSONUtils.jsonToVideoUplaodRequestMessage(payload);
            List<AssetPropertyValue> propertyValueEntry = videoUploadRequestMessage.getPayload().getValues();
            if (propertyValueEntry.size() == 1) {
                AssetPropertyValue assetPropertyValue = propertyValueEntry.get(0);
                long propertyUpdateTimestamp = assetPropertyValue.timestamp().timeInSeconds();
                String value = assetPropertyValue.value().stringValue();
                long startTimestamp = 0;
                long endTimestamp = 0;
                boolean isLive = false;
                if (value.equalsIgnoreCase(MQTT_LIVE_VIDEO_UPLOAD_REQUEST_KEY)) {
                    // Live Uploading Request
                    isLive = true;
                    log.info("Live Streaming Request Received");
                } else {
                    // On-Demand Video Uploading Request
                    String[] timestamps = value.split("-");
                    if (timestamps.length == 2) {
                        try {
                            startTimestamp = Long.parseLong(timestamps[0]);
                            endTimestamp = Long.parseLong(timestamps[1]);
                            log.info("On-Demand Streaming Request Received for Time Range " + timestamps[0] + "-" + timestamps[1]);
                        } catch (NumberFormatException ex) {
                            log.error("Invalid VideoUploadRequest Event Received. " + ex.getMessage());
                        }
                    } else {
                        log.error("Invalid VideoUploadRequest Event Received. Single hyphen required");
                    }
                }
                event.onStart(isLive, propertyUpdateTimestamp, startTimestamp, endTimestamp);
            } else {
                log.error("Invalid VideoUploadRequest Event Received. Single value required");
            }
        }

        @Override
        public boolean onStreamError(Throwable throwable) {
            log.info("onStream Error: " + throwable.getMessage());
            event.onError(throwable.getMessage());
            // Handle error.
            return false;
        }

        @Override
        public void onStreamClosed() {
            log.info("onStream Closed Called");
        }
    };
    try {
        SubscribeToIoTCoreRequest subscribeToIoTCoreRequest = new SubscribeToIoTCoreRequest();
        subscribeToIoTCoreRequest.setTopicName(mqttTopic);
        subscribeToIoTCoreRequest.setQos(QOS.AT_MOST_ONCE);
        SubscribeToIoTCoreResponseHandler operationResponseHandler = greengrassCoreIPCClient.subscribeToIoTCore(subscribeToIoTCoreRequest, Optional.of(streamResponseHandler));
        SubscribeToIoTCoreResponse resp = operationResponseHandler.getResponse().get();
        log.info("Subscribe to MQTT Response: " + resp.toString());
    } catch (ExecutionException ex) {
        final String errorMessage = String.format("Could not Subscribe to MQTT topic %s. %s", mqttTopic, ex.getMessage());
        log.error(errorMessage);
        throw new EdgeConnectorForKVSException(errorMessage, ex);
    } catch (InterruptedException ex) {
        final String errorMessage = String.format("Could not Subscribe to MQTT topic %s. %s", mqttTopic, ex.getMessage());
        log.error(errorMessage);
        // restore interrupted state
        Thread.currentThread().interrupt();
        throw new EdgeConnectorForKVSException(errorMessage, ex);
    }
}

Example 2 with SubscribeToIoTCoreResponseHandler

use of software.amazon.awssdk.aws.greengrass.SubscribeToIoTCoreResponseHandler in project aws-greengrass-nucleus by aws-greengrass.

the class IPCMqttProxyTest method GIVEN_MqttProxyEventStreamClient_WHEN_called_subscribe_THEN_subscribed_and_message_received.

@Test
void GIVEN_MqttProxyEventStreamClient_WHEN_called_subscribe_THEN_subscribed_and_message_received() throws Exception {
    CountDownLatch messageLatch = new CountDownLatch(1);
    GreengrassCoreIPCClient greengrassCoreIPCClient = new GreengrassCoreIPCClient(clientConnection);
    SubscribeToIoTCoreRequest subscribeToIoTCoreRequest = new SubscribeToIoTCoreRequest();
    subscribeToIoTCoreRequest.setQos(QOS.AT_LEAST_ONCE);
    subscribeToIoTCoreRequest.setTopicName(TEST_SUBSCRIBE_TOPIC);
    StreamResponseHandler<IoTCoreMessage> streamResponseHandler = new StreamResponseHandler<IoTCoreMessage>() {

        @Override
        public void onStreamEvent(IoTCoreMessage streamEvent) {
            if (Arrays.equals(streamEvent.getMessage().getPayload(), TEST_PAYLOAD) && streamEvent.getMessage().getTopicName().equals(TEST_SUBSCRIBE_TOPIC)) {
                messageLatch.countDown();
            }
        }

        @Override
        public boolean onStreamError(Throwable error) {
            logger.atError().cause(error).log("Subscribe stream errored");
            return false;
        }

        @Override
        public void onStreamClosed() {
        }
    };
    SubscribeToIoTCoreResponseHandler responseHandler = greengrassCoreIPCClient.subscribeToIoTCore(subscribeToIoTCoreRequest, Optional.of(streamResponseHandler));
    responseHandler.getResponse().get(TIMEOUT_FOR_MQTTPROXY_SECONDS, TimeUnit.SECONDS);
    ArgumentCaptor<SubscribeRequest> subscribeRequestArgumentCaptor = ArgumentCaptor.forClass(SubscribeRequest.class);
    verify(mqttClient).subscribe(subscribeRequestArgumentCaptor.capture());
    SubscribeRequest capturedSubscribeRequest = subscribeRequestArgumentCaptor.getValue();
    assertThat(capturedSubscribeRequest.getTopic(), is(TEST_SUBSCRIBE_TOPIC));
    assertThat(capturedSubscribeRequest.getQos(), is(QualityOfService.AT_LEAST_ONCE));
    Consumer<MqttMessage> callback = capturedSubscribeRequest.getCallback();
    MqttMessage message = new MqttMessage(TEST_SUBSCRIBE_TOPIC, TEST_PAYLOAD);
    callback.accept(message);
    assertTrue(messageLatch.await(TIMEOUT_FOR_MQTTPROXY_SECONDS, TimeUnit.SECONDS));
    // close stream -> unsubscribe
    responseHandler.closeStream();
    Thread.sleep(500);
    ArgumentCaptor<UnsubscribeRequest> unsubscribeRequestArgumentCaptor = ArgumentCaptor.forClass(UnsubscribeRequest.class);
    verify(mqttClient).unsubscribe(unsubscribeRequestArgumentCaptor.capture());
    UnsubscribeRequest capturedUnsubscribeRequest = unsubscribeRequestArgumentCaptor.getValue();
    assertThat(capturedUnsubscribeRequest.getTopic(), is(TEST_SUBSCRIBE_TOPIC));
    assertThat(capturedUnsubscribeRequest.getCallback(), is(callback));
}