Skip to content

Commit 8d69c1c

Browse files
committed
Update KafkaHelper to use the new TaskStatus
1 parent 0ad39cb commit 8d69c1c

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

  • tracer/src/Datadog.Trace/ClrProfiler/AutoInstrumentation/Kafka

tracer/src/Datadog.Trace/ClrProfiler/AutoInstrumentation/Kafka/KafkaHelper.cs

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
using System.Reflection;
1111
using System.Text;
1212
using System.Threading;
13+
using System.Threading.Tasks;
1314
using Datadog.Trace.Configuration.Schema;
1415
using Datadog.Trace.DataStreamsMonitoring;
1516
using Datadog.Trace.DataStreamsMonitoring.Utils;
@@ -463,6 +464,11 @@ internal static void TryInjectHeaders<TTopicPartitionMarker, TMessage>(
463464

464465
var duckTask = adminClient.DescribeClusterAsync(options);
465466

467+
if (duckTask.Status == TaskStatus.RanToCompletion)
468+
{
469+
return duckTask.Result?.ClusterId;
470+
}
471+
466472
var originalContext = SynchronizationContext.Current;
467473
try
468474
{

0 commit comments

Comments
 (0)