Use RegisterGrainTimer instead of obsolete RegisterTimer

dotnet · Jan 18, 2025 · adc1432 · adc1432
1 parent e8ec057
commit adc1432
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/orleans/Adventure/AdventureGrains/MonsterGrain.cs b/orleans/Adventure/AdventureGrains/MonsterGrain.cs
@@ -1,25 +1,27 @@
-using AdventureGrainInterfaces;
+using AdventureGrainInterfaces;
 
 namespace AdventureGrains;
 
-public class MonsterGrain : Grain, IMonsterGrain
+public class MonsterGrain : Grain, IMonsterGrain, IDisposable
 {
     private MonsterInfo _monsterInfo = new();
+    private IGrainTimer? _timer;
     private IRoomGrain? _roomGrain; // Current room
 
     public override Task OnActivateAsync(CancellationToken cancellationToken)
     {
         _monsterInfo = _monsterInfo with { Id = this.GetPrimaryKeyLong() };
 
-        RegisterTimer(
+        _timer = RegisterGrainTimer(
             _ => Move(),
-            null!,
             TimeSpan.FromSeconds(150),
             TimeSpan.FromMinutes(150));
 
         return base.OnActivateAsync(cancellationToken);
     }
 
+    public void Dispose() => _timer?.Dispose();
+
     Task IMonsterGrain.SetInfo(MonsterInfo info)
     {
         _monsterInfo = info;