PID header added (#464)

Author: Travonka

src/Ydb.Sdk/CHANGELOG.md

@@ -1,3 +1,4 @@
+- Added 'x-ydb-client-pid' header to any RPC calls
 - Added DisableServerBalancer option to ADO.NET session creation; default false.
 
 ## v0.20.1

src/Ydb.Sdk/src/DriverConfig.cs

@@ -6,6 +6,8 @@ namespace Ydb.Sdk;
 
 public class DriverConfig
 {
+    private readonly string _pid = Environment.ProcessId.ToString();
+
     public string Endpoint { get; }
     public string Database { get; }
     public ICredentialsProvider? Credentials { get; }
@@ -61,7 +63,8 @@ public DriverConfig(
     internal Grpc.Core.Metadata GetCallMetadata => new()
     {
         { Metadata.RpcDatabaseHeader, Database },
-        { Metadata.RpcSdkInfoHeader, SdkVersion }
+        { Metadata.RpcSdkInfoHeader, SdkVersion },
+        { Metadata.RpcClientPid, _pid }
     };
 
     private static string FormatEndpoint(string endpoint)

src/Ydb.Sdk/src/Metadata.cs

@@ -2,13 +2,18 @@
 
 internal static class Metadata
 {
+    // Outgoing headers
     public const string RpcDatabaseHeader = "x-ydb-database";
     public const string RpcAuthHeader = "x-ydb-auth-ticket";
     public const string RpcRequestTypeHeader = "x-ydb-request-type";
     public const string RpcTraceIdHeader = "x-ydb-trace-id";
     public const string RpcSdkInfoHeader = "x-ydb-sdk-build-info";
+    public const string RpcClientPid = "x-ydb-client-pid";
+
+    // Incoming headers
     public const string RpcServerHintsHeader = "x-ydb-server-hints";
     public const string RpcClientCapabilitiesHeader = "x-ydb-client-capabilities";
 
+    //Incoming hints
     public const string GracefulShutdownHint = "session-close";
 }

src/Ydb.Sdk/tests/Sys/QuerySessionTests.cs

@@ -0,0 +1,29 @@
+using Xunit;
+using Ydb.Sdk.Tests.Ado.Specification;
+using Ydb.Sdk.Tests.Fixture;
+
+namespace Ydb.Sdk.Tests.Sys;
+
+public class QuerySessionTests : YdbAdoNetFixture
+{
+    public QuerySessionTests(YdbFactoryFixture fixture) : base(fixture)
+    {
+    }
+
+    [Fact]
+    public async Task QuerySessionPidTest()
+    {
+        await using var connection = await CreateOpenConnectionAsync();
+        var dbCommand = connection.CreateCommand();
+        dbCommand.CommandText = "SELECT ClientPID FROM `.sys/query_sessions` LIMIT 1;";
+
+        var expectedPid = Environment.ProcessId.ToString();
+
+        await dbCommand.ExecuteNonQueryAsync();
+        await using var reader = await dbCommand.ExecuteReaderAsync();
+
+        Assert.True(reader.HasRows);
+        Assert.True(await reader.ReadAsync());
+        Assert.Equal(reader.GetString(0), expectedPid);
+    }
+}