Merge pull request #21 from RedisLabsModules/type

Add support for JSON.TYPE

Author: gkorland

src/lib.rs

@@ -47,6 +47,21 @@ fn json_get(ctx: &Context, args: Vec<String>) -> RedisResult {
     Ok(value)
 }
 
+fn json_type(ctx: &Context, args: Vec<String>) -> RedisResult {
+    let mut args = args.into_iter().skip(1);
+    let key = args.next_string()?;
+    let path = args.next_string()?;
+
+    let key = ctx.open_key_writable(&key);
+
+    let value = match key.get_value::<RedisJSON>(&REDIS_JSON_TYPE)? {
+        Some(doc) => { doc.get_type(&path)?.into() }
+        None => ().into()
+    };
+
+    Ok(value)
+}
+
 //////////////////////////////////////////////////////
 
 redis_module! {
@@ -58,5 +73,6 @@ redis_module! {
     commands: [
         ["json.set", json_set, "write"],
         ["json.get", json_get, ""],
+        ["json.type", json_type, ""],
     ],
 }

src/redisjson.rs

@@ -64,4 +64,30 @@ impl RedisJSON {
 
         return Ok(s)
     }
+
+    pub fn get_type(&self, path: &str) -> Result<String, Error> {
+        let s = match self.get_doc(path)? {
+            Some(doc) => {
+                match doc {
+                    Value::Null => "null",
+                    Value::Bool(_) => "boolean",
+                    Value::Number(_) => "number",
+                    Value::String(_) => "string",
+                    Value::Array(_) => "array",
+                    Value::Object(_) => "object",
+                }
+            }
+            None => ""
+        };
+        Ok(s.to_string())
+    }
+
+    pub fn get_doc(&self, path: &str) -> Result<Option<&Value>, Error> {
+        // Create a JSONPath selector
+        let selector = Selector::new(path).map_err(|e| Error {
+            msg: format!("{}", e),
+        })?;
+
+        Ok(selector.find(&self.data).next())
+    }
 }