add on disconnect callback

Author: xiazhvera

samples/mqtt/mqtt5_aws_websocket.py

@@ -76,7 +76,7 @@ def on_lifecycle_attempting_connect(lifecycle_attempting_connect_data: mqtt5.Lif
 # Callback for the lifecycle event Connection Success
 def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.LifecycleConnectSuccessData):
     connack_packet = lifecycle_connect_success_data.connack_packet
-    print("Lifecycle Connection Success with reason_code:{}\n".format(
+    print("Lifecycle Connection Success with reason code:{}\n".format(
         repr(connack_packet.reason_code)))
     connection_success_event.set()
 
@@ -87,6 +87,12 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         lifecycle_connection_failure.exception))
 
 
+# Callback for the lifecycle event Disconnection
+def on_lifecycle_disconnection(lifecycle_disconnect_data: mqtt5.LifecycleDisconnectData):
+    print("Lifecycle Disconnected with reason code:{}".format(
+        lifecycle_disconnect_data.disconnect_packet.reason_code if lifecycle_disconnect_data.disconnect_packet else "None"))
+
+
 if __name__ == '__main__':
     print("\nStarting MQTT5 Websocket Sample\n")
     message_count = args.input_count
@@ -107,6 +113,7 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         on_lifecycle_attempting_connect=on_lifecycle_attempting_connect,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
         on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         client_id=args.input_clientId)
 
 

samples/mqtt/mqtt5_custom_auth_signed.py

@@ -86,7 +86,7 @@ def on_lifecycle_attempting_connect(lifecycle_attempting_connect_data: mqtt5.Lif
 # Callback for the lifecycle event Connection Success
 def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.LifecycleConnectSuccessData):
     connack_packet = lifecycle_connect_success_data.connack_packet
-    print("Lifecycle Connection Success with reason_code:{}\n".format(
+    print("Lifecycle Connection Success with reason code:{}\n".format(
         repr(connack_packet.reason_code)))
     connection_success_event.set()
 
@@ -97,6 +97,12 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         lifecycle_connection_failure.exception))
 
 
+# Callback for the lifecycle event Disconnection
+def on_lifecycle_disconnection(lifecycle_disconnect_data: mqtt5.LifecycleDisconnectData):
+    print("Lifecycle Disconnected with reason code:{}".format(
+        lifecycle_disconnect_data.disconnect_packet.reason_code if lifecycle_disconnect_data.disconnect_packet else "None"))
+
+
 if __name__ == '__main__':
 
     # Create MQTT5 Client with a custom authorizer
@@ -114,6 +120,7 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         on_lifecycle_attempting_connect=on_lifecycle_attempting_connect,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
         on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         client_id=args.input_clientId)
 
 

samples/mqtt/mqtt5_custom_auth_unsigned.py

@@ -79,7 +79,7 @@ def on_lifecycle_attempting_connect(lifecycle_attempting_connect_data: mqtt5.Lif
 # Callback for the lifecycle event Connection Success
 def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.LifecycleConnectSuccessData):
     connack_packet = lifecycle_connect_success_data.connack_packet
-    print("Lifecycle Connection Success with reason_code:{}\n".format(
+    print("Lifecycle Connection Success with reason code:{}\n".format(
         repr(connack_packet.reason_code)))
     connection_success_event.set()
 
@@ -90,6 +90,12 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         lifecycle_connection_failure.exception))
 
 
+# Callback for the lifecycle event Disconnection
+def on_lifecycle_disconnection(lifecycle_disconnect_data: mqtt5.LifecycleDisconnectData):
+    print("Lifecycle Disconnected with reason code:{}".format(
+        lifecycle_disconnect_data.disconnect_packet.reason_code if lifecycle_disconnect_data.disconnect_packet else "None"))
+
+
 if __name__ == '__main__':
 
     # Create MQTT5 Client with a custom authorizer
@@ -104,6 +110,7 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         on_lifecycle_attempting_connect=on_lifecycle_attempting_connect,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
         on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         client_id=args.input_clientId)
 
 

samples/mqtt/mqtt5_pkcs11_connect.py

@@ -72,6 +72,16 @@ def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.Lifecy
     global future_connection_success
     future_connection_success.set_result(lifecycle_connect_success_data)
 
+# Callback for the lifecycle event Connection Failure
+def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.LifecycleConnectFailureData):
+    print("Lifecycle Connection Failure with exception:{}".format(
+        lifecycle_connection_failure.exception))
+
+# Callback for the lifecycle event Disconnection
+def on_lifecycle_disconnection(lifecycle_disconnect_data: mqtt5.LifecycleDisconnectData):
+    print("Lifecycle Disconnected with reason code:{}".format(
+        lifecycle_disconnect_data.disconnect_packet.reason_code if lifecycle_disconnect_data.disconnect_packet else "None"))
+
 
 if __name__ == '__main__':
     print("\nStarting MQTT5 pkcs11 connect Sample\n")
@@ -99,6 +109,8 @@ def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.Lifecy
         ca_filepath=args.input_ca,
         on_lifecycle_stopped=on_lifecycle_stopped,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
+        on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         client_id=args.input_clientId)
 
     print("MQTT5 Client Created")

samples/mqtt/mqtt5_x509.py

@@ -78,7 +78,7 @@ def on_lifecycle_attempting_connect(lifecycle_attempting_connect_data: mqtt5.Lif
 # Callback for the lifecycle event Connection Success
 def on_lifecycle_connection_success(lifecycle_connect_success_data: mqtt5.LifecycleConnectSuccessData):
     connack_packet = lifecycle_connect_success_data.connack_packet
-    print("Lifecycle Connection Success with reason_code:{}\n".format(
+    print("Lifecycle Connection Success with reason code:{}\n".format(
         repr(connack_packet.reason_code)))
     connection_success_event.set()
 
@@ -89,6 +89,12 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         lifecycle_connection_failure.exception))
 
 
+# Callback for the lifecycle event Disconnection
+def on_lifecycle_disconnection(lifecycle_disconnect_data: mqtt5.LifecycleDisconnectData):
+    print("Lifecycle Disconnected with reason code:{}".format(
+        lifecycle_disconnect_data.disconnect_packet.reason_code if lifecycle_disconnect_data.disconnect_packet else "None"))
+
+
 if __name__ == '__main__':
     print("\nStarting MQTT5 X509 PubSub Sample\n")
     message_count = args.input_count
@@ -107,6 +113,7 @@ def on_lifecycle_connection_failure(lifecycle_connection_failure: mqtt5.Lifecycl
         on_lifecycle_attempting_connect=on_lifecycle_attempting_connect,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
         on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         client_id=args.input_clientId)
 
 

samples/service_clients/fleet_provisioning_basic.py

@@ -40,6 +40,10 @@ def on_lifecycle_connection_success(event: mqtt5.LifecycleConnectSuccessData):
     def on_lifecycle_connection_failure(event: mqtt5.LifecycleConnectFailureData):
         initial_connection_success.set_exception(Exception("Failed to connect"))
 
+    def on_lifecycle_disconnection(event: mqtt5.LifecycleDisconnectData):
+        print("Lifecycle Disconnected with reason code:{}".format(
+            event.disconnect_packet.reason_code if event.disconnect_packet else "None"))
+
     stopped = Future()
     def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         stopped.set_result(True)
@@ -58,6 +62,8 @@ def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         clean_session=True,
         keep_alive_secs=1200,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
+        on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         on_lifecycle_stopped=on_lifecycle_stopped)
 
     mqtt5_client.start()

samples/service_clients/fleet_provisioning_csr.py

@@ -45,6 +45,10 @@ def on_lifecycle_connection_success(event: mqtt5.LifecycleConnectSuccessData):
     def on_lifecycle_connection_failure(event: mqtt5.LifecycleConnectFailureData):
         initial_connection_success.set_exception(Exception("Failed to connect"))
 
+    def on_lifecycle_disconnection(event: mqtt5.LifecycleDisconnectData):
+        print("Lifecycle Disconnected with reason code:{}".format(
+            event.disconnect_packet.reason_code if event.disconnect_packet else "None"))
+
     stopped = Future()
     def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         stopped.set_result(True)
@@ -63,6 +67,8 @@ def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         clean_session=True,
         keep_alive_secs=1200,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
+        on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         on_lifecycle_stopped=on_lifecycle_stopped)
 
     mqtt5_client.start()

samples/service_clients/jobs.py

@@ -137,6 +137,10 @@ def on_lifecycle_connection_success(event: mqtt5.LifecycleConnectSuccessData):
     def on_lifecycle_connection_failure(event: mqtt5.LifecycleConnectFailureData):
         initial_connection_success.set_exception(Exception("Failed to connect"))
 
+    def on_lifecycle_disconnection(event: mqtt5.LifecycleDisconnectData):
+        print("Lifecycle Disconnected with reason code:{}".format(
+            event.disconnect_packet.reason_code if event.disconnect_packet else "None"))
+
     stopped = Future()
     def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         stopped.set_result(True)
@@ -150,6 +154,8 @@ def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         clean_session=True,
         keep_alive_secs=1200,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
+        on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         on_lifecycle_stopped=on_lifecycle_stopped)
 
     mqtt5_client.start()

samples/service_clients/shadow.py

@@ -98,6 +98,10 @@ def on_lifecycle_connection_success(event: mqtt5.LifecycleConnectSuccessData):
     def on_lifecycle_connection_failure(event: mqtt5.LifecycleConnectFailureData):
         initial_connection_success.set_exception(Exception("Failed to connect"))
 
+    def on_lifecycle_disconnection(event: mqtt5.LifecycleDisconnectData):
+        print("Lifecycle Disconnected with reason code:{}".format(
+            event.disconnect_packet.reason_code if event.disconnect_packet else "None"))
+
     stopped = Future()
     def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         stopped.set_result(True)
@@ -111,6 +115,8 @@ def on_lifecycle_stopped(event: mqtt5.LifecycleStoppedData):
         clean_session=True,
         keep_alive_secs=1200,
         on_lifecycle_connection_success=on_lifecycle_connection_success,
+        on_lifecycle_connection_failure=on_lifecycle_connection_failure,
+        on_lifecycle_disconnection=on_lifecycle_disconnection,
         on_lifecycle_stopped=on_lifecycle_stopped)
 
     mqtt5_client.start()