commit 9ffe99e: [Minor] Try to save ratelimits from the proper worker

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2023-07-01 19:56:23 +0100
URL: https://github.com/rspamd/rspamd/commit/9ffe99ee139743c3ea562d3b7cd999bdab584aa6 (HEAD -> master)

[Minor] Try to save ratelimits from the proper worker

---
 src/fuzzy_storage.c | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/fuzzy_storage.c b/src/fuzzy_storage.c
index 42e0417ac..4bdfabee6 100644
--- a/src/fuzzy_storage.c
+++ b/src/fuzzy_storage.c
@@ -3244,9 +3244,6 @@ start_fuzzy (struct rspamd_worker *worker)
 	rspamd_fuzzy_backend_close (ctx->backend);
 
 	if (worker->index == 0) {
-		if (ctx->ratelimit_buckets) {
-			rspamd_fuzzy_maybe_save_ratelimits (ctx);
-		}
 		g_array_free (ctx->updates_pending, TRUE);
 		ctx->updates_pending = NULL;
 	}
@@ -3256,6 +3253,12 @@ start_fuzzy (struct rspamd_worker *worker)
 	}
 
 	if (ctx->ratelimit_buckets) {
+		/* Try the best to save ratelimits from the proper worker */
+		if ((!ctx->dedicated_update_worker && worker->index == 0) ||
+			(ctx->dedicated_update_worker && worker->index == 1)) {
+			rspamd_fuzzy_maybe_save_ratelimits (ctx);
+		}
+
 		rspamd_lru_hash_destroy (ctx->ratelimit_buckets);
 	}