Customize the session for the Sequence

lib/mongoid_auto_inc/incrementor.rb

@@ -3,8 +3,9 @@
 module MongoidAutoInc
   class Incrementor
     class Sequence
-      def initialize(sequence, collection_name, seed)
+      def initialize(sequence, session_name, collection_name, seed)
         @sequence = sequence.to_s
+        @session = session_name.to_s
         @collection = collection_name.to_s
         exists? || create(seed)
       end
@@ -31,7 +32,7 @@ def collection
         if ::Mongoid::VERSION < '3'
           Mongoid.database[@collection]
         else
-          Mongoid.default_session[@collection]
+          Mongoid.session(@session)[@collection]
         end
       end
 
@@ -66,15 +67,16 @@ def update_number_with(mongo_func)
     def initialize(options=nil)
       options ||= {}
       @collection = options[:collection] || "sequences"
+      @session = options[:session] || "default"
       @seed = options[:seed].to_i
     end
 
     def [](sequence)
-      Sequence.new(sequence, @collection, @seed)
+      Sequence.new(sequence, @session, @collection, @seed)
     end
 
     def []=(sequence, number)
-      Sequence.new(sequence, @collection, @seed).set(number)
+      Sequence.new(sequence, @session, @collection, @seed).set(number)
     end
   end
 end

mongoid_auto_inc.gemspec

@@ -21,5 +21,4 @@ Gem::Specification.new do |s|
 
   s.required_ruby_version = '>= 1.8.6'
   s.required_rubygems_version = '>= 1.3.5'
-  s.add_bundler_dependencies
 end