From 17b0c14341dbaaa9712cf12e8f516fe513282f82 Mon Sep 17 00:00:00 2001
From: scooletz <scooletz@gmail.com>
Date: Wed, 16 Jul 2025 09:35:01 +0200
Subject: [PATCH] UnreferenceDisposable made slimmer

---
 src/ModelContextProtocol.AspNetCore/HttpMcpSession.cs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/ModelContextProtocol.AspNetCore/HttpMcpSession.cs b/src/ModelContextProtocol.AspNetCore/HttpMcpSession.cs
index c34aba6c..1456ce56 100644
--- a/src/ModelContextProtocol.AspNetCore/HttpMcpSession.cs
+++ b/src/ModelContextProtocol.AspNetCore/HttpMcpSession.cs
@@ -25,13 +25,15 @@ internal sealed class HttpMcpSession<TTransport>(
     public bool IsActive => !SessionClosed.IsCancellationRequested && _referenceCount > 0;
     public long LastActivityTicks { get; private set; } = timeProvider.GetTimestamp();
 
+    private TimeProvider TimeProvider => timeProvider;
+
     public IMcpServer? Server { get; set; }
     public Task? ServerRunTask { get; set; }
 
     public IDisposable AcquireReference()
     {
         Interlocked.Increment(ref _referenceCount);
-        return new UnreferenceDisposable(this, timeProvider);
+        return new UnreferenceDisposable(this);
     }
 
     public bool TryStartGetRequest() => Interlocked.Exchange(ref _getRequestStarted, 1) == 0;
@@ -70,13 +72,13 @@ public async ValueTask DisposeAsync()
     public bool HasSameUserId(ClaimsPrincipal user)
         => UserIdClaim == StreamableHttpHandler.GetUserIdClaim(user);
 
-    private sealed class UnreferenceDisposable(HttpMcpSession<TTransport> session, TimeProvider timeProvider) : IDisposable
+    private sealed class UnreferenceDisposable(HttpMcpSession<TTransport> session) : IDisposable
     {
         public void Dispose()
         {
             if (Interlocked.Decrement(ref session._referenceCount) == 0)
             {
-                session.LastActivityTicks = timeProvider.GetTimestamp();
+                session.LastActivityTicks = session.TimeProvider.GetTimestamp();
             }
         }
     }