add support for notify_keyspace_event (#133)

* add support for notify_keyspace_event
* add events example

Author: gkorland

examples/block.rs

@@ -1,7 +1,7 @@
 #[macro_use]
 extern crate redis_module;
 
-use redis_module::{Context, RedisError, RedisResult, RedisValue, ThreadSafeContext};
+use redis_module::{Context, RedisResult, RedisValue, ThreadSafeContext};
 use std::thread;
 use std::time::Duration;
 

examples/events.rs

@@ -1,7 +1,7 @@
 #[macro_use]
 extern crate redis_module;
 
-use redis_module::{Context, NotifyEvent};
+use redis_module::{Context, NotifyEvent, RedisError, RedisResult, Status};
 
 fn on_event(ctx: &Context, event_type: NotifyEvent, event: &str, key: &str) {
     let msg = format!(
@@ -15,13 +15,27 @@ fn on_stream(ctx: &Context, _event_type: NotifyEvent, _event: &str, _key: &str)
     ctx.log_debug("Stream event received!");
 }
 
+fn event_send(ctx: &Context, args: Vec<String>) -> RedisResult {
+    if args.len() > 1 {
+        return Err(RedisError::WrongArity);
+    }
+
+    let status = ctx.notify_keyspace_event(NotifyEvent::GENERIC, "events.send", "mykey");
+    match status {
+        Status::Ok => Ok("Event sent".into()),
+        Status::Err => Err(RedisError::Str("Generic error")),
+    }
+}
+
 //////////////////////////////////////////////////////
 
 redis_module! {
     name: "events",
     version: 1,
     data_types: [],
-    commands: [],
+    commands: [
+        ["events.send", event_send, "", 0, 0, 0],
+    ],
     event_handlers: [
         [@EXPIRED @EVICTED: on_event],
         [@STREAM: on_stream],

examples/timer.rs

@@ -1,7 +1,7 @@
 #[macro_use]
 extern crate redis_module;
 
-use redis_module::{Context, NextArg, RedisError, RedisResult};
+use redis_module::{Context, NextArg, RedisResult};
 use std::time::Duration;
 
 fn callback(ctx: &Context, data: String) {

src/context/mod.rs

@@ -213,4 +213,14 @@ impl Context {
     pub fn get_raw(&self) -> *mut raw::RedisModuleCtx {
         return self.ctx;
     }
+
+    #[cfg(feature = "experimental-api")]
+    pub fn notify_keyspace_event(
+        &self,
+        event_type: raw::NotifyEvent,
+        event: &str,
+        keyname: &str,
+    ) -> raw::Status {
+        raw::notify_keyspace_event(self.ctx, event_type, event, keyname)
+    }
 }

src/lib.rs

@@ -29,8 +29,8 @@ pub use crate::context::thread_safe::{DetachedFromClient, ThreadSafeContext};
 #[cfg(feature = "experimental-api")]
 pub use crate::raw::NotifyEvent;
 
-pub use crate::raw::Status;
 pub use crate::context::Context;
+pub use crate::raw::Status;
 pub use crate::redismodule::*;
 
 /// Ideally this would be `#[cfg(not(test))]`, but that doesn't work:

src/logging.rs

@@ -1,5 +1,5 @@
-use crate::LogLevel;
 use crate::raw;
+use crate::LogLevel;
 use std::ffi::CString;
 use std::ptr;
 

src/raw.rs

@@ -474,3 +474,23 @@ pub fn subscribe_to_server_event(
 ) -> Status {
     unsafe { RedisModule_SubscribeToServerEvent.unwrap()(ctx, event, callback).into() }
 }
+
+#[cfg(feature = "experimental-api")]
+pub fn notify_keyspace_event(
+    ctx: *mut RedisModuleCtx,
+    event_type: NotifyEvent,
+    event: &str,
+    keyname: &str,
+) -> Status {
+    let event = CString::new(event).unwrap();
+    let keyname = RedisString::create(ctx, keyname);
+    unsafe {
+        RedisModule_NotifyKeyspaceEvent.unwrap()(
+            ctx,
+            event_type.bits,
+            event.as_ptr(),
+            keyname.inner,
+        )
+        .into()
+    }
+}