[Project] Adopt C modules and stuff

1 files changed, 15 insertions, 2 deletions

diff --git a/src/libstat/learn_cache/redis_cache.c b/src/libstat/learn_cache/redis_cache.c
index 11bc13aae..789991186 100644
--- a/src/libstat/learn_cache/redis_cache.c
+++ b/src/libstat/learn_cache/redis_cache.c
@@ -41,6 +41,7 @@ struct rspamd_redis_cache_ctx {
 struct rspamd_redis_cache_runtime {
 	struct rspamd_redis_cache_ctx *ctx;
 	struct rspamd_task *task;
+	struct rspamd_symcache_item *item;
 	struct upstream *selected;
 	struct event timeout_event;
 	redisAsyncContext *redis;
@@ -151,6 +152,9 @@ rspamd_stat_cache_redis_get (redisAsyncContext *c, gpointer r, gpointer priv)
 	}
 
 	if (rt->has_event) {
+		if (rt->item) {
+			rspamd_symcache_item_async_dec_check (task, rt->item);
+		}
 		rspamd_session_remove_event (task->s, rspamd_redis_cache_fin, rt);
 	}
 }
@@ -173,6 +177,9 @@ rspamd_stat_cache_redis_set (redisAsyncContext *c, gpointer r, gpointer priv)
 	}
 
 	if (rt->has_event) {
+		if (rt->item) {
+			rspamd_symcache_item_async_dec_check (task, rt->item);
+		}
 		rspamd_session_remove_event (task->s, rspamd_redis_cache_fin, rt);
 	}
 }
@@ -453,7 +460,11 @@ rspamd_stat_cache_redis_check (struct rspamd_task *task,
 	if (redisAsyncCommand (rt->redis, rspamd_stat_cache_redis_get, rt,
 			"HGET %s %s",
 			rt->ctx->redis_object, h) == REDIS_OK) {
-		rspamd_session_add_event (task->s, NULL, rspamd_redis_cache_fin, rt, rspamd_stat_cache_redis_quark ());
+		rspamd_session_add_event (task->s,
+				rspamd_redis_cache_fin,
+				rt,
+				rspamd_stat_cache_redis_quark ());
+		rt->item = rspamd_symbols_cache_get_cur_item (task);
 		event_add (&rt->timeout_event, &tv);
 		rt->has_event = TRUE;
 	}
@@ -485,7 +496,9 @@ rspamd_stat_cache_redis_learn (struct rspamd_task *task,
 	if (redisAsyncCommand (rt->redis, rspamd_stat_cache_redis_set, rt,
 			"HSET %s %s %d",
 			rt->ctx->redis_object, h, flag) == REDIS_OK) {
-		rspamd_session_add_event (task->s, NULL, rspamd_redis_cache_fin, rt, rspamd_stat_cache_redis_quark ());
+		rspamd_session_add_event (task->s,
+				rspamd_redis_cache_fin, rt, rspamd_stat_cache_redis_quark ());
+		rt->item = rspamd_symbols_cache_get_cur_item (task);
 		event_add (&rt->timeout_event, &tv);
 		rt->has_event = TRUE;
 	}