Added keymiss event support

examples/events.rs

@@ -1,7 +1,12 @@
 #[macro_use]
 extern crate redis_module;
 
-use redis_module::{Context, NotifyEvent, RedisError, RedisResult, RedisString, Status};
+use redis_module::{
+    Context, NotifyEvent, RedisError, RedisResult, RedisString, RedisValue, Status,
+};
+use std::sync::atomic::{AtomicI64, Ordering};
+
+static NUM_KEY_MISSES: AtomicI64 = AtomicI64::new(0);
 
 fn on_event(ctx: &Context, event_type: NotifyEvent, event: &str, key: &str) {
     let msg = format!(
@@ -28,6 +33,14 @@ fn event_send(ctx: &Context, args: Vec<RedisString>) -> RedisResult {
     }
 }
 
+fn on_key_miss(_ctx: &Context, _event_type: NotifyEvent, _event: &str, _key: &str) {
+    NUM_KEY_MISSES.fetch_add(1, Ordering::SeqCst);
+}
+
+fn num_key_miss(_ctx: &Context, _args: Vec<RedisString>) -> RedisResult {
+    Ok(RedisValue::Integer(NUM_KEY_MISSES.load(Ordering::SeqCst)))
+}
+
 //////////////////////////////////////////////////////
 
 redis_module! {
@@ -36,10 +49,12 @@ redis_module! {
     data_types: [],
     commands: [
         ["events.send", event_send, "", 0, 0, 0],
+        ["events.num_key_miss", num_key_miss, "", 0, 0, 0],
     ],
     event_handlers: [
         [@EXPIRED @EVICTED: on_event],
         [@STREAM: on_stream],
+        [@MISSED: on_key_miss],
     ]
 }
 

src/raw.rs

@@ -117,6 +117,7 @@ bitflags! {
         const STREAM = REDISMODULE_NOTIFY_STREAM;
         const MODULE = REDISMODULE_NOTIFY_MODULE;
         const LOADED = REDISMODULE_NOTIFY_LOADED;
+        const MISSED = REDISMODULE_NOTIFY_KEY_MISS;
         const ALL = REDISMODULE_NOTIFY_ALL;
     }
 }

tests/integration.rs

@@ -1,8 +1,7 @@
 use crate::utils::{get_redis_connection, start_redis_server_with_module};
 use anyhow::Context;
 use anyhow::Result;
-use redis::RedisError;
-use redis::RedisResult;
+use redis::{RedisError, RedisResult};
 
 mod utils;
 
@@ -323,3 +322,22 @@ fn test_verify_acl_on_user() -> Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn test_key_miss_event() -> Result<()> {
+    let port: u16 = 6492;
+    let _guards = vec![start_redis_server_with_module("events", port)
+        .with_context(|| "failed to start redis server")?];
+    let mut con =
+        get_redis_connection(port).with_context(|| "failed to connect to redis server")?;
+
+    let res: usize = redis::cmd("events.num_key_miss").query(&mut con)?;
+    assert_eq!(res, 0);
+
+    let _: RedisResult<String> = redis::cmd("GET").arg(&["x"]).query(&mut con);
+
+    let res: usize = redis::cmd("events.num_key_miss").query(&mut con)?;
+    assert_eq!(res, 1);
+
+    Ok(())
+}