TCP: do not disconnect if ota client is connected

Author: pennam

src/ArduinoIoTCloudTCP.cpp

@@ -319,10 +319,22 @@ ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_ConnectMqttBroker()
 
 ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_Connected()
 {
-  if (!_mqttClient.connected() || !_thing.connected() || !_device.connected())
-  {
-    return State::Disconnect;
+#if OTA_ENABLED
+  if(_get_ota_confirmation != nullptr &&
+      _ota.getState() == OTACloudProcessInterface::State::OtaAvailable &&
+      _get_ota_confirmation()) {
+    _ota.approveOta();
+  }
+  _ota.update();
+
+  if (!_otaClient.connected()) {
+#endif
+    if (!_mqttClient.connected() || !_thing.connected() || !_device.connected()) {
+      return State::Disconnect;
+    }
+#if OTA_ENABLED
   }
+#endif
 
   /* Check for new data from the MQTT client. */
   _mqttClient.poll();
@@ -338,16 +350,6 @@ ArduinoIoTCloudTCP::State ArduinoIoTCloudTCP::handle_Connected()
   /* Call CloudDevice process to get configuration */
   _device.update();
 
-#if  OTA_ENABLED
-  if(_get_ota_confirmation != nullptr &&
-      _ota.getState() == OTACloudProcessInterface::State::OtaAvailable &&
-      _get_ota_confirmation()) {
-    _ota.approveOta();
-  }
-
-  _ota.update();
-#endif // OTA_ENABLED
-
 
   if (_device.isAttached()) {
     /* Call CloudThing process to synchronize properties */