Merge pull request #44 from the9ball/chore/execute_count

chore: add ExecuteCount to WorkloadContext
Cysharp · Mar 21, 2024 · 609be9f · 609be9f
2 parents 0c5d71a + a11a506
commit 609be9f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 1 deletion.
diff --git a/sandbox/ConsoleWorker/Workloads.cs b/sandbox/ConsoleWorker/Workloads.cs
@@ -26,7 +26,7 @@ public override Task SetupAsync(WorkloadContext context)
 
     public override async Task ExecuteAsync(WorkloadContext context)
     {
-        logger.LogInformation("Execute:" + (++execCount));
+        logger.LogInformation($"Execute:{context.ExecuteCount}");
         await Task.Delay(TimeSpan.FromSeconds(1));
     }
 

diff --git a/src/DFrame.Unity/Assets/Plugins/DFrame/Runtime/DFrameWorkerApp.cs b/src/DFrame.Unity/Assets/Plugins/DFrame/Runtime/DFrameWorkerApp.cs
@@ -341,6 +341,7 @@ await Task.WhenAll(workloads.Select((x, workloadIndex) => Task.Run(async () =>
                             var sw = ValueStopwatch.StartNew();
                             try
                             {
+                                x.context.ExecuteCount = i;
                                 await x.workload.ExecuteAsync(x.context);
                             }
                             catch (OperationCanceledException e) when (e.CancellationToken == token.Value)

diff --git a/src/DFrame.Unity/Assets/Plugins/DFrame/Runtime/WorkloadContext.cs b/src/DFrame.Unity/Assets/Plugins/DFrame/Runtime/WorkloadContext.cs
@@ -8,6 +8,7 @@ public class WorkloadContext
         public WorkloadId WorkloadId { get; }
         public int WorkloadCount { get; }
         public int WorkloadIndex { get; }
+        public long ExecuteCount { get; internal set; } = 0;
         public int Concurrency { get; }
         public long TotalRequestCount { get; }
         public CancellationToken CancellationToken { get; }

diff --git a/src/DFrame.Worker/DFrameWorkerApp.cs b/src/DFrame.Worker/DFrameWorkerApp.cs
@@ -341,6 +341,7 @@ await Task.WhenAll(workloads.Select((x, workloadIndex) => Task.Run(async () =>
                             var sw = ValueStopwatch.StartNew();
                             try
                             {
+                                x.context.ExecuteCount = i;
                                 await x.workload.ExecuteAsync(x.context);
                             }
                             catch (OperationCanceledException e) when (e.CancellationToken == token.Value)

diff --git a/src/DFrame.Worker/WorkloadContext.cs b/src/DFrame.Worker/WorkloadContext.cs
@@ -8,6 +8,7 @@ public class WorkloadContext
         public WorkloadId WorkloadId { get; }
         public int WorkloadCount { get; }
         public int WorkloadIndex { get; }
+        public long ExecuteCount { get; internal set; } = 0;
         public int Concurrency { get; }
         public long TotalRequestCount { get; }
         public CancellationToken CancellationToken { get; }