Search in sources :

Example 1 with VideoUploadRequestMessage

use of com.aws.iot.edgeconnectorforkvs.model.VideoUploadRequestMessage 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);
    }
}
Also used : VideoUploadRequestMessage(com.aws.iot.edgeconnectorforkvs.model.VideoUploadRequestMessage) EdgeConnectorForKVSException(com.aws.iot.edgeconnectorforkvs.model.exceptions.EdgeConnectorForKVSException) MQTTMessage(software.amazon.awssdk.aws.greengrass.model.MQTTMessage) SubscribeToIoTCoreResponse(software.amazon.awssdk.aws.greengrass.model.SubscribeToIoTCoreResponse) IoTCoreMessage(software.amazon.awssdk.aws.greengrass.model.IoTCoreMessage) AssetPropertyValue(software.amazon.awssdk.services.iotsitewise.model.AssetPropertyValue) SubscribeToIoTCoreRequest(software.amazon.awssdk.aws.greengrass.model.SubscribeToIoTCoreRequest) SubscribeToIoTCoreResponseHandler(software.amazon.awssdk.aws.greengrass.SubscribeToIoTCoreResponseHandler) List(java.util.List) ExecutionException(java.util.concurrent.ExecutionException)

Example 2 with VideoUploadRequestMessage

use of com.aws.iot.edgeconnectorforkvs.model.VideoUploadRequestMessage in project aws-iot-greengrass-edge-connector-for-kinesis-video-stream by awslabs.

the class JSONUtils method jsonToVideoUplaodRequestMessage.

public static VideoUploadRequestMessage jsonToVideoUplaodRequestMessage(String payload) {
    Gson gson = new Gson();
    VideoUploadRequestMessage videoUploadRequestMessage = gson.fromJson(payload, VideoUploadRequestMessage.class);
    return videoUploadRequestMessage;
}
Also used : VideoUploadRequestMessage(com.aws.iot.edgeconnectorforkvs.model.VideoUploadRequestMessage) Gson(com.google.gson.Gson)

Aggregations

VideoUploadRequestMessage (com.aws.iot.edgeconnectorforkvs.model.VideoUploadRequestMessage)2 EdgeConnectorForKVSException (com.aws.iot.edgeconnectorforkvs.model.exceptions.EdgeConnectorForKVSException)1 Gson (com.google.gson.Gson)1 List (java.util.List)1 ExecutionException (java.util.concurrent.ExecutionException)1 SubscribeToIoTCoreResponseHandler (software.amazon.awssdk.aws.greengrass.SubscribeToIoTCoreResponseHandler)1 IoTCoreMessage (software.amazon.awssdk.aws.greengrass.model.IoTCoreMessage)1 MQTTMessage (software.amazon.awssdk.aws.greengrass.model.MQTTMessage)1 SubscribeToIoTCoreRequest (software.amazon.awssdk.aws.greengrass.model.SubscribeToIoTCoreRequest)1 SubscribeToIoTCoreResponse (software.amazon.awssdk.aws.greengrass.model.SubscribeToIoTCoreResponse)1 AssetPropertyValue (software.amazon.awssdk.services.iotsitewise.model.AssetPropertyValue)1