Streaming operation reference fixup

awslabs · Sep 19, 2024 · 2f9d543 · 2f9d543
1 parent dee89a3
commit 2f9d543
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/source/iot/MqttRequestResponseClient.cpp b/source/iot/MqttRequestResponseClient.cpp
@@ -252,13 +252,17 @@ namespace Aws
                     aws_mqtt_request_response_client_create_streaming_operation(client, &streamingOptions);
                 if (!stream)
                 {
+                    Aws::Crt::Delete(implHandle, allocator);
                     return nullptr;
                 }
 
                 auto impl = Aws::Crt::MakeShared<StreamingOperationImpl>(
                     allocator, stream, options, aws_mqtt_request_response_client_get_event_loop(client));
                 auto streamingOperation = Aws::Crt::MakeShared<StreamingOperation>(allocator, impl);
 
+                implHandle->m_allocator = allocator;
+                implHandle->m_impl = impl;
+
                 return streamingOperation;
             }