karait 0.0.8 → 1.0.0

data/VERSION

@@ -1 +1 @@
-0.0.8
+1.0.0

data/karait.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = "karait"
-  s.version = "0.0.8"
+  s.version = "1.0.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["bcoe"]
-  s.date = "2011-11-12"
+  s.date = "2011-12-04"
   s.description = "A ridiculously simple cross-language queuing system, built on top of MongoDB."
   s.email = "bencoe@gmail.com"
   s.extra_rdoc_files = [

data/lib/queue.rb

@@ -27,9 +27,14 @@ module Karait
         :visible_after => -1.0
       }
       
-      message_dict[:_meta][:routing_key] = opts.fetch(:routing_key) if opts[:routing_key]
+      message_dict[:_meta][:routing_key] = opts[:routing_key] if opts[:routing_key]
+      
+      if opts[:unique_key]
+        unique_insert(message_dict, opts[:unique_key])
+      else
+        @queue_collection.insert(message_dict, :safe => true)
+      end
       
-      @queue_collection.insert(message_dict, :safe => true)
     end
     
     def read(opts={})
@@ -130,6 +135,13 @@ module Karait
     end
     
     private
+
+    def unique_insert(message_dict, unique_key)
+      return @database.eval(
+        "function(obj) { if ( db.#{@queue}.count({#{unique_key}: obj.#{unique_key}}) ) { return false; } db.#{@queue}.insert(obj); return true;}",
+        message_dict
+      )
+    end
     
     def block_until_message_available(query, polling_interval=1.0, polling_timeout=None)
         current_time = Time.new.to_f

data/test/test_queue.rb

@@ -251,4 +251,19 @@ class TestQueue < Test::Unit::TestCase
     stop_time = Time.new.to_f
     assert_equal true, stop_time - start_time > 0.1
   end
+
+  should "should not insert two documents with the same value for a unique key" do
+    queue = Karait::Queue.new(
+      :database => 'karait_test',
+      :queue => 'queue_test'
+    )
+
+    queue.write Karait::Message.new({'foo' => 'value1'}), :unique_key => 'foo'
+    queue.write Karait::Message.new({'foo' => 'value1'}), :unique_key => 'foo'
+    queue.write Karait::Message.new({'foo' => 'value2'}), :unique_key => 'foo'
+
+    messages = queue.read()
+
+    assert_equal 2, messages.count
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: karait
 version: !ruby/object:Gem::Version 
-  hash: 15
+  hash: 23
   prerelease: 
   segments: 
+  - 1
   - 0
   - 0
-  - 8
-  version: 0.0.8
+  version: 1.0.0
 platform: ruby
 authors: 
 - bcoe
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-11-12 00:00:00 Z
+date: 2011-12-04 00:00:00 Z
 dependencies: []
 
 description: A ridiculously simple cross-language queuing system, built on top of MongoDB.