Use TimeProvider

src/MagicOnion.Server/Diagnostics/MagicOnionMetrics.cs

@@ -63,7 +63,7 @@ public void StreamingHubMethodCompleted(in MetricsContext context, StreamingHubH
             var tags = InitializeTagListForStreamingHub(handler.HubName);
             tags.Add("rpc.method", handler.MethodInfo.Name);
             tags.Add("magiconion.streaminghub.is_error", isErrorOrInterrupted ? BoxedTrue : BoxedFalse);
-            streamingHubMethodDuration.Record((long)StopwatchHelper.GetElapsedTime(startingTimestamp, endingTimestamp).TotalMilliseconds, tags);
+            streamingHubMethodDuration.Record((long)TimeProvider.System.GetElapsedTime(startingTimestamp, endingTimestamp).TotalMilliseconds, tags);
             streamingHubMethodCompletedCounter.Add(1, tags);
         }
     }

src/MagicOnion.Server/Hubs/StreamingHub.cs

@@ -1,8 +1,6 @@
 using System.Buffers;
-using System.Diagnostics;
 using System.Runtime.CompilerServices;
 using System.Threading.Channels;
-using Cysharp.Runtime.Multicast;
 using Cysharp.Runtime.Multicast.Remoting;
 using Grpc.Core;
 using MagicOnion.Internal;
@@ -23,6 +21,7 @@ public abstract class StreamingHubBase<THubInterface, TReceiver> : ServiceBase<T
 {
     IRemoteClientResultPendingTaskRegistry remoteClientResultPendingTasks = default!;
     StreamingHubHeartbeatHandle heartbeatHandle = default!;
+    TimeProvider timeProvider = default!;
 
     protected static readonly Task<Nil> NilTask = Task.FromResult(Nil.Default);
     protected static readonly ValueTask CompletedTask = new ValueTask();
@@ -88,10 +87,11 @@ internal async Task<DuplexStreamingResult<StreamingHubPayload, StreamingHubPaylo
 
         var features = this.Context.CallContext.GetHttpContext().Features;
         var magicOnionOptions = serviceProvider.GetRequiredService<IOptions<MagicOnionOptions>>().Value;
+        timeProvider = magicOnionOptions.TimeProvider ?? TimeProvider.System;
 
         var remoteProxyFactory = serviceProvider.GetRequiredService<IRemoteProxyFactory>();
         var remoteSerializer = serviceProvider.GetRequiredService<IRemoteSerializer>();
-        this.remoteClientResultPendingTasks = new RemoteClientResultPendingTaskRegistry(magicOnionOptions.ClientResultsDefaultTimeout, magicOnionOptions.TimeProvider ?? TimeProvider.System);
+        this.remoteClientResultPendingTasks = new RemoteClientResultPendingTaskRegistry(magicOnionOptions.ClientResultsDefaultTimeout, timeProvider);
         this.Client = remoteProxyFactory.CreateDirect<TReceiver>(new MagicOnionRemoteReceiverWriter(StreamingServiceContext), remoteSerializer, remoteClientResultPendingTasks);
 
         var handlerRepository = serviceProvider.GetRequiredService<StreamingHubHandlerRepository>();
@@ -270,10 +270,10 @@ async ValueTask ProcessRequestAsync(UniqueHashDictionary<StreamingHubHandler> ha
                 hubInstance: this,
                 request: body,
                 messageId: messageId,
-                timestamp: DateTime.UtcNow
+                timestamp: timeProvider.GetUtcNow().UtcDateTime
             );
 
-            var methodStartingTimestamp = Stopwatch.GetTimestamp();
+            var methodStartingTimestamp = timeProvider.GetTimestamp();
             var isErrorOrInterrupted = false;
             MagicOnionServerLog.BeginInvokeHubMethod(Context.MethodHandler.Logger, context, context.Request, handler.RequestType);
             try
@@ -300,8 +300,8 @@ async ValueTask ProcessRequestAsync(UniqueHashDictionary<StreamingHubHandler> ha
             }
             finally
             {
-                var methodEndingTimestamp = Stopwatch.GetTimestamp();
-                MagicOnionServerLog.EndInvokeHubMethod(Context.MethodHandler.Logger, context, context.ResponseSize, context.ResponseType, StopwatchHelper.GetElapsedTime(methodStartingTimestamp, methodEndingTimestamp).TotalMilliseconds, isErrorOrInterrupted);
+                var methodEndingTimestamp = timeProvider.GetTimestamp();
+                MagicOnionServerLog.EndInvokeHubMethod(Context.MethodHandler.Logger, context, context.ResponseSize, context.ResponseType, timeProvider.GetElapsedTime(methodStartingTimestamp, methodEndingTimestamp).TotalMilliseconds, isErrorOrInterrupted);
                 Metrics.StreamingHubMethodCompleted(Context.Metrics, handler, methodStartingTimestamp, methodEndingTimestamp, isErrorOrInterrupted);
 
                 StreamingHubContextPool.Shared.Return(context);

src/MagicOnion.Server/Hubs/StreamingHubHeartbeatManager.cs

@@ -82,7 +82,7 @@ public void Ack(short sequence)
         lock (gate)
         {
             var receivedAtTimestamp = manager.TimeProvider.GetTimestamp();
-            var elapsed = StopwatchHelper.GetElapsedTime(lastSentAtTimestamp, receivedAtTimestamp);
+            var elapsed = manager.TimeProvider.GetElapsedTime(lastSentAtTimestamp, receivedAtTimestamp);
 
             LastReceivedAt = lastSentAt.Add(elapsed);
             Latency = elapsed;