updated the add routine to ensure we update the set in order.

1 files changed, 47 insertions, 11 deletions

diff --git a/queue/zset/add.rb b/queue/zset/add.rb
index c75bcd6..2d9f27a 100755
--- a/queue/zset/add.rb
+++ b/queue/zset/add.rb
@@ -7,23 +7,59 @@ require "redis"
 
 @redis = Redis.new(:host => "127.0.0.1")
 
-
 x = []
 
-x.push( "test 1" )
-x.push( "test 2" )
-x.push( "test 3" )
-
+#
+# Queue loads of tests
+#
+(1..10).to_a.each do |i|
+  x.push( "test #{i}" )
+end
 
-for i in 0 .. 10 
+loop do
     x.each do |test|
         @redis.watch('zset')
-        if (!(@redis.zscore("zset", test)))
-            res = @redis.multi do |r|
-                r.zadd('zset', Time.now.to_f * 10000000, test)
-            end
+
+        print "adding #{test}"
+
+        #
+        # This is run in a loop, as we have to wait until both
+        #
+        # (a) the score is missing
+        # (b) the zadd function succeeds
+        #
+        loop do
+          #
+          # Print a dot for each go through the loop
+          #
+          print "."
+
+          #
+          # Only update if no score is set
+          #
+          if !@redis.zscore("zset", test)
+
+            #
+            # If MULTI returns nil, the transaction failed, so we need to try
+            # again.
+            #
+            break unless @redis.multi do |r|
+              @redis.zadd('zset', Time.now.to_f, test)
+            end.nil?
+
+          end
+
+          #
+          # This could be tighter..
+          #
+          sleep 0.1
         end
+
+        print "\n"
+
+        #
+        # Do we need to unwatch here?
+        #
         @redis.unwatch
     end
-    sleep 1
 end