forked from npgsql/npgsql
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathNpgsqlActivitySource.cs
More file actions
88 lines (75 loc) · 2.99 KB
/
NpgsqlActivitySource.cs
File metadata and controls
88 lines (75 loc) · 2.99 KB
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
using Npgsql.Internal;
using System;
using System.Diagnostics;
using System.Net;
using System.Net.Sockets;
using System.Reflection;
namespace Npgsql;
static class NpgsqlActivitySource
{
static readonly ActivitySource Source;
static NpgsqlActivitySource()
{
var assembly = typeof(NpgsqlActivitySource).Assembly;
var version = assembly.GetCustomAttribute<AssemblyFileVersionAttribute>()?.Version ?? "0.0.0";
Source = new("Npgsql", version);
}
internal static bool IsEnabled => Source.HasListeners();
internal static Activity? CommandStart(NpgsqlConnector connector, string sql)
{
var settings = connector.Settings;
var activity = Source.StartActivity(settings.Database!, ActivityKind.Client);
if (activity is not { IsAllDataRequested: true })
return activity;
activity.SetTag("db.system", "postgresql");
activity.SetTag("db.connection_string", connector.UserFacingConnectionString);
activity.SetTag("db.user", settings.Username);
activity.SetTag("db.name", settings.Database);
activity.SetTag("db.statement", sql);
activity.SetTag("db.connection_id", connector.Id);
var endPoint = connector.ConnectedEndPoint;
Debug.Assert(endPoint is not null);
switch (endPoint)
{
case IPEndPoint ipEndPoint:
activity.SetTag("net.transport", "ip_tcp");
activity.SetTag("net.peer.ip", ipEndPoint.Address.ToString());
if (ipEndPoint.Port != 5432)
activity.SetTag("net.peer.port", ipEndPoint.Port);
activity.SetTag("net.peer.name", settings.Host);
break;
case UnixDomainSocketEndPoint:
activity.SetTag("net.transport", "unix");
activity.SetTag("net.peer.name", settings.Host);
break;
default:
throw new ArgumentOutOfRangeException("Invalid endpoint type: " + endPoint.GetType());
}
return activity;
}
internal static void ReceivedFirstResponse(Activity activity)
{
var activityEvent = new ActivityEvent("received-first-response");
activity.AddEvent(activityEvent);
}
internal static void CommandStop(Activity activity)
{
activity.SetTag("otel.status_code", "OK");
activity.Dispose();
}
internal static void SetException(Activity activity, Exception ex, bool escaped = true)
{
var tags = new ActivityTagsCollection
{
{ "exception.type", ex.GetType().FullName },
{ "exception.message", ex.Message },
{ "exception.stacktrace", ex.ToString() },
{ "exception.escaped", escaped }
};
var activityEvent = new ActivityEvent("exception", tags: tags);
activity.AddEvent(activityEvent);
activity.SetTag("otel.status_code", "ERROR");
activity.SetTag("otel.status_description", ex is PostgresException pgEx ? pgEx.SqlState : ex.Message);
activity.Dispose();
}
}