test

Author: LiamHamsters

src/Ydb.Sdk/src/Ado/PoolManager.cs

@@ -35,11 +35,18 @@ CancellationToken cancellationToken
                 : Drivers[settings.GrpcConnectionString] = await settings.BuildDriver();
             driver.RegisterOwner();
 
-            var factory = new PoolingSessionFactory(driver, settings);
-            var newSessionPool = new PoolingSessionSource<PoolingSession>(factory, settings);
+            ISessionSource newSessionPool;
+            if (settings.EnableImplicitSession)
+            {
+                newSessionPool = new ImplicitSessionSource(driver);
+            }
+            else
+            {
+                var factory = new PoolingSessionFactory(driver, settings);
+                newSessionPool = new PoolingSessionSource<PoolingSession>(factory, settings);
+            }
 
             Pools[settings.ConnectionString] = newSessionPool;
-
             return newSessionPool;
         }
         finally

src/Ydb.Sdk/src/Ado/Session/ImplicitSession.cs

@@ -9,11 +9,11 @@ internal sealed class ImplicitSession : ISession
     private int _disposed;
 
     public ImplicitSession(IDriver driver)
+        : this(driver, owner: null)
     {
-        Driver = driver;
     }
 
-    public ImplicitSession(IDriver driver, ImplicitSessionSource owner)
+    public ImplicitSession(IDriver driver, ImplicitSessionSource? owner)
     {
         Driver = driver;
         _owner = owner;
@@ -34,10 +34,10 @@ public ValueTask<IServerStream<ExecuteQueryResponsePart>> ExecuteQuery(
 
         var request = new ExecuteQueryRequest
         {
-            ExecMode     = ExecMode.Execute,
+            ExecMode = ExecMode.Execute,
             QueryContent = new QueryContent { Text = query, Syntax = Syntax.YqlV1 },
-            StatsMode    = StatsMode.None,
-            TxControl    = txControl
+            StatsMode = StatsMode.None,
+            TxControl = txControl
         };
         request.Parameters.Add(parameters);
 

src/Ydb.Sdk/src/Ado/YdbConnection.cs

@@ -41,19 +41,11 @@ internal ISession Session
 
     internal bool EnableImplicitSession => ConnectionStringBuilder.EnableImplicitSession;
 
-    internal ISession GetExecutionSession(bool useImplicit)
-    {
-        if (!useImplicit)
-            return Session;
-
-        if (_implicitSource == null)
-            _implicitSource = new ImplicitSessionSource(Session.Driver);
-
-        return new ImplicitSession(Session.Driver, _implicitSource);
-    }
-
     private ImplicitSessionSource? _implicitSource;
 
+    internal ISession GetExecutionSession(bool useImplicit) =>
+        useImplicit ? new ImplicitSession(Session.Driver) : Session;
+
     internal void AdoptSession(ISession session) => _session = session;
 
     public YdbConnection()