increment and decrement users count

granstel · Mar 14, 2024 · 208c181 · 208c181
1 parent 50d1277
commit 208c181
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/Dodo1000Bot.Services/Metrics/MetricsCollector.cs b/Dodo1000Bot.Services/Metrics/MetricsCollector.cs
@@ -21,4 +21,9 @@ public static void Set(string key, double value)
     {
         Metrics.WithLabels(key).Set(value);
     }
+
+    public static void Decrement(string key)
+    {
+        Metrics.WithLabels(key).Dec();
+    }
 }
diff --git a/Dodo1000Bot.Services/UsersService.cs b/Dodo1000Bot.Services/UsersService.cs
@@ -32,6 +32,8 @@ public async Task SaveAndNotify(User user, CancellationToken cancellationToken)
     {
         await _usersRepository.SaveUser(user, cancellationToken);
 
+        MetricsCollector.Increment("users");
+
         await CheckAndNotifyAboutSubscribers(cancellationToken);
     }
 
@@ -62,8 +64,11 @@ internal async Task CheckAndNotifyAboutSubscribers(CancellationToken cancellatio
         }
     }
 
-    public Task Delete(User user, CancellationToken cancellationToken) => 
-        _usersRepository.Delete(user, cancellationToken);
+    public async Task Delete(User user, CancellationToken cancellationToken)
+    {
+        await _usersRepository.Delete(user, cancellationToken);
+        MetricsCollector.Decrement("users");
+    }
 
     public Task<IList<User>> GetUsers(Source messengerType, CancellationToken cancellationToken) =>
         _usersRepository.GetUsers(messengerType, cancellationToken);