Add log for rate limiter

planetary-social · Sep 27, 2024 · a9ecc47 · a9ecc47
1 parent c722c8a
commit a9ecc47
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/domain/followee_notification_factory.rs b/src/domain/followee_notification_factory.rs
@@ -127,17 +127,15 @@ impl FolloweeNotificationFactory {
 
             let tokens_needed = messages.len() as f64;
 
-            // Just to sample the rate limiter
-            if tokens_needed > 1.0 {
+            self.rate_limiter.overcharge(tokens_needed);
+
+            if let Some(follow_change) = followers.first() {
                 info!(
-                    "Rate limiter for followee {} after flush: {}",
-                    self.followee.unwrap(),
+                    "Followee {}. {}",
+                    follow_change.friendly_followee(),
                     self.rate_limiter
                 );
             }
-
-            self.rate_limiter.overcharge(tokens_needed);
-
             return messages;
         }