[Project] Implement expiration

1 files changed, 25 insertions, 0 deletions

diff --git a/src/fuzzy_storage.c b/src/fuzzy_storage.c
index ad1697dc1..770f428ad 100644
--- a/src/fuzzy_storage.c
+++ b/src/fuzzy_storage.c
@@ -138,6 +138,7 @@ struct fuzzy_key {
 	double burst;
 	double rate;
 	ev_tstamp expire;
+	bool expired;
 	ref_entry_t ref;
 };
 
@@ -1599,6 +1600,30 @@ rspamd_fuzzy_process_command(struct fuzzy_session *session)
 			}
 		}
 
+		if (session->key && !isnan(session->key->expire)) {
+			/* Check expire */
+			static ev_tstamp today = NAN;
+
+			if (isnan(today)) {
+				today = ev_time();
+			}
+			else if (rspamd_random_uint64_fast() > 0xFFFF000000000000ULL) {
+				today = ev_time();
+			}
+
+			if (today > session->key->expire) {
+				if (!session->key->expired) {
+					msg_info("key %s is expired", session->key->name);
+					session->key->expired = true;
+				}
+
+				result.v1.value = 503;
+				result.v1.prob = 0.0f;
+				rspamd_fuzzy_make_reply(cmd, &result, session, send_flags);
+				return;
+			}
+		}
+
 		if (is_rate_allowed) {
 			REF_RETAIN(session);
 			rspamd_fuzzy_backend_check(session->ctx->backend, cmd,