fgerlits commented on code in PR #1432: URL: https://github.com/apache/nifi-minifi-cpp/pull/1432#discussion_r1041264186
########## extensions/mqtt/processors/AbstractMQTTProcessor.cpp: ########## @@ -163,23 +200,259 @@ void AbstractMQTTProcessor::reconnect() { } logger_->log_info("Reconnecting to %s", uri_); - int ret = MQTTAsync_connect(client_, &conn_opts); + if (MQTTAsync_isConnected(client_)) { + logger_->log_debug("Already connected to %s, no need to reconnect", uri_); + return; + } + const int ret = MQTTAsync_connect(client_, &conn_opts); + MQTTProperties_free(&connect_props); if (ret != MQTTASYNC_SUCCESS) { - logger_->log_error("Failed to reconnect to MQTT broker %s (%d)", uri_, ret); + logger_->log_error("MQTTAsync_connect failed to MQTT broker %s with error code [%d]", uri_, ret); + return; } + + // wait until connection succeeds or fails + connect_finished_task.get_future().get(); +} + +void AbstractMQTTProcessor::setMqtt5ConnectOptions(MQTTAsync_connectOptions& conn_opts, MQTTProperties& connect_props, MQTTProperties& will_props) const { + conn_opts.cleanstart = getCleanStart(); + + { + MQTTProperty property; + property.identifier = MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL; + property.value.integer4 = gsl::narrow<int>(getSessionExpiryInterval().count()); + MQTTProperties_add(&connect_props, &property); + } + + if (!last_will_content_type_.empty()) { + MQTTProperty property; + property.identifier = MQTTPROPERTY_CODE_CONTENT_TYPE; + property.value.data.len = last_will_content_type_.length(); + property.value.data.data = const_cast<char*>(last_will_content_type_.data()); + MQTTProperties_add(&will_props, &property); + } + + conn_opts.willProperties = &will_props; + + setMqtt5ConnectOptionsImpl(connect_props); +} + +void AbstractMQTTProcessor::onTrigger(const std::shared_ptr<core::ProcessContext>& context, const std::shared_ptr<core::ProcessSession>& session) { + // read lock + std::shared_lock client_lock{client_mutex_}; + if (client_ == nullptr) { + // we are shutting down + return; + } + + // reconnect if needed + reconnect(); + + if (!MQTTAsync_isConnected(client_)) { + logger_->log_error("Could not work with MQTT broker because disconnected to %s", uri_); + yield(); + return; + } + + onTriggerImpl(context, session); } void AbstractMQTTProcessor::freeResources() { - if (client_ && MQTTAsync_isConnected(client_)) { - MQTTAsync_disconnectOptions disconnect_options = MQTTAsync_disconnectOptions_initializer; - disconnect_options.context = this; - disconnect_options.onSuccess = disconnectionSuccess; - disconnect_options.onFailure = disconnectionFailure; - disconnect_options.timeout = gsl::narrow<int>(std::chrono::milliseconds{connection_timeout_}.count()); - MQTTAsync_disconnect(client_, &disconnect_options); + // write lock + std::lock_guard client_lock{client_mutex_}; + + if (!client_) { + return; } - if (client_) { - MQTTAsync_destroy(&client_); + + disconnect(); + + MQTTAsync_destroy(&client_); +} + +void AbstractMQTTProcessor::disconnect() { + if (!MQTTAsync_isConnected(client_)) { + return; + } + + MQTTAsync_disconnectOptions disconnect_options = MQTTAsync_disconnectOptions_initializer; + std::packaged_task<void(MQTTAsync_successData*, MQTTAsync_successData5*, MQTTAsync_failureData*, MQTTAsync_failureData5*)> disconnect_finished_task( + [this] (MQTTAsync_successData* success_data, MQTTAsync_successData5* success_data_5, MQTTAsync_failureData* failure_data, MQTTAsync_failureData5* failure_data_5) { + onDisconnectFinished(success_data, success_data_5, failure_data, failure_data_5); + }); + disconnect_options.context = &disconnect_finished_task; + + if (mqtt_version_.value() == MqttVersions::V_5_0) { + disconnect_options.onSuccess5 = connectionSuccess5; + disconnect_options.onFailure5 = connectionFailure5; + } else { + disconnect_options.onSuccess = connectionSuccess; + disconnect_options.onFailure = connectionFailure; + } + + disconnect_options.timeout = gsl::narrow<int>(std::chrono::milliseconds{connection_timeout_}.count()); + + const int ret = MQTTAsync_disconnect(client_, &disconnect_options); + if (ret != MQTTASYNC_SUCCESS) { + logger_->log_error("MQTTAsync_disconnect failed to MQTT broker %s with error code [%d]", uri_, ret); + return; + } + + // wait until connection succeeds or fails + disconnect_finished_task.get_future().get(); +} + +void AbstractMQTTProcessor::setBrokerLimits(MQTTAsync_successData5* response) { + auto readProperty = [response] (MQTTPropertyCodes property_code, auto& out_var) { + const int value = MQTTProperties_getNumericValue(&response->properties, property_code); + if (value != PAHO_MQTT_C_FAILURE_CODE) { + if constexpr (std::is_same_v<decltype(out_var), std::optional<std::chrono::seconds>&>) { + out_var = std::chrono::seconds(value); + } else { + out_var = gsl::narrow<typename std::remove_reference_t<decltype(out_var)>::value_type>(value); + } + } else { + out_var.reset(); + } + }; + + readProperty(MQTTPROPERTY_CODE_RETAIN_AVAILABLE, retain_available_); + readProperty(MQTTPROPERTY_CODE_WILDCARD_SUBSCRIPTION_AVAILABLE, wildcard_subscription_available_); + readProperty(MQTTPROPERTY_CODE_SHARED_SUBSCRIPTION_AVAILABLE, shared_subscription_available_); + + readProperty(MQTTPROPERTY_CODE_TOPIC_ALIAS_MAXIMUM, broker_topic_alias_maximum_); + readProperty(MQTTPROPERTY_CODE_RECEIVE_MAXIMUM, broker_receive_maximum_); + readProperty(MQTTPROPERTY_CODE_MAXIMUM_QOS, maximum_qos_); + readProperty(MQTTPROPERTY_CODE_MAXIMUM_PACKET_SIZE, maximum_packet_size_); + + readProperty(MQTTPROPERTY_CODE_SESSION_EXPIRY_INTERVAL, maximum_session_expiry_interval_); + readProperty(MQTTPROPERTY_CODE_SERVER_KEEP_ALIVE, server_keep_alive_); +} + +void AbstractMQTTProcessor::checkBrokerLimits() { + try { + if (server_keep_alive_.has_value() && server_keep_alive_ < keep_alive_interval_) { + std::ostringstream os; + os << "Set Keep Alive Interval (" << keep_alive_interval_.count() << " s) is longer then maximum supported by broker (" << server_keep_alive_->count() << " s)"; + throw minifi::Exception(ExceptionType::PROCESS_SCHEDULE_EXCEPTION, os.str()); + } + + if (maximum_qos_.has_value() && qos_.value() > maximum_qos_) { + std::ostringstream os; + os << "Set QoS (" << qos_.value() << ") is higher than maximum supported by broker (" << *maximum_qos_ << ")"; + throw minifi::Exception(ExceptionType::PROCESS_SCHEDULE_EXCEPTION, os.str()); + } + + checkBrokerLimitsImpl(); + } + catch (...) { + disconnect(); + throw; + } +} + +void AbstractMQTTProcessor::connectionLost(void *context, char* cause) { + auto* processor = reinterpret_cast<AbstractMQTTProcessor*>(context); + processor->onConnectionLost(cause); +} + + +void AbstractMQTTProcessor::connectionSuccess(void* context, MQTTAsync_successData* response) { + auto* task = reinterpret_cast<std::packaged_task<void(MQTTAsync_successData*, MQTTAsync_successData5*, MQTTAsync_failureData*, MQTTAsync_failureData5*)>*>(context); Review Comment: yeah, I meant `using`, but "use a `using`" sounds weird) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@nifi.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org