Merge pull request #207 from debasishg/client-on-subscribe

Unset pubsub flag on unsubscribe

Author: debasishg

src/main/scala/com/redis/PubSub.scala

@@ -98,11 +98,15 @@ trait PubSub extends PubOperations { self: Redis =>
   }
 
   def unsubscribe(): Unit = {
-    send("UNSUBSCRIBE")(())
+    val r = send("UNSUBSCRIBE")(())
+    pubSub = false
+    r
   }
 
   def unsubscribe(channel: String, channels: String*): Unit = {
-    send("UNSUBSCRIBE", channel :: channels.toList)(())
+    val r = send("UNSUBSCRIBE", channel :: channels.toList)(())
+    pubSub = false
+    r
   }
 }
 

src/test/scala/com/redis/RedisClientSpec.scala

@@ -28,4 +28,34 @@ class RedisClientSpec extends FunSpec
       new RedisClient("localhost", 6379, 1).toString shouldBe "localhost:6379/1"
     }
   }
+
+  describe("test subscribe") {
+    val r = new RedisClient("localhost", 6379)
+  
+    println(r.get("vvl:qm"))
+  
+    r.subscribe("vvl.qm") { m =>
+      println(m)
+    }
+  
+    Thread.sleep(3000)
+  
+    r.unsubscribe("vvl.qm")
+  
+    Thread.sleep(3000)
+  
+    println(r.get("vvl:qm"))
+  
+    r.subscribe("vvl.qm") { m =>
+      println(m)
+    }
+  
+    Thread.sleep(3000)
+  
+    r.unsubscribe("vvl.qm")
+  
+    Thread.sleep(3000)
+  
+    r.get("vvl:qm")
+  }
 }