-
Notifications
You must be signed in to change notification settings - Fork 729
/
HttpDiagnosticsHttpRequestMessageExtensions.cs
51 lines (45 loc) · 1.8 KB
/
HttpDiagnosticsHttpRequestMessageExtensions.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
#if NET5_0_OR_GREATER
using System.Collections.Generic;
#endif
using Microsoft.Extensions.Http.Diagnostics;
using Microsoft.Shared.Diagnostics;
namespace System.Net.Http;
/// <summary>
/// Extensions for telemetry utilities.
/// </summary>
public static class HttpDiagnosticsHttpRequestMessageExtensions
{
/// <summary>
/// Sets metadata for outgoing requests to be used for telemetry purposes.
/// </summary>
/// <param name="request"><see cref="HttpRequestMessage"/> object.</param>
/// <param name="metadata">Metadata for the request.</param>
public static void SetRequestMetadata(this HttpRequestMessage request, RequestMetadata metadata)
{
_ = Throw.IfNull(request);
_ = Throw.IfNull(metadata);
#if NET5_0_OR_GREATER
_ = request.Options.TryAdd(TelemetryConstants.RequestMetadataKey, metadata);
#else
request.Properties.Add(TelemetryConstants.RequestMetadataKey, metadata);
#endif
}
/// <summary>
/// Gets metadata for outgoing requests to be used for telemetry purposes.
/// </summary>
/// <param name="request"><see cref="HttpRequestMessage"/> object.</param>
/// <returns>Request metadata or <see langword="null"/>.</returns>
public static RequestMetadata? GetRequestMetadata(this HttpRequestMessage request)
{
_ = Throw.IfNull(request);
#if NET5_0_OR_GREATER
_ = request.Options.TryGetValue(new HttpRequestOptionsKey<RequestMetadata>(TelemetryConstants.RequestMetadataKey), out var metadata);
return metadata;
#else
_ = request.Properties.TryGetValue(TelemetryConstants.RequestMetadataKey, out var metadata);
return (RequestMetadata?)metadata;
#endif
}
}