Out put sequencer class name to log

hirocaster · Nov 6, 2015 · f419db5 · f419db5
1 parent ba39b65
commit f419db5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/lib/active_record/sharding/sequencer.rb b/lib/active_record/sharding/sequencer.rb
@@ -28,9 +28,10 @@ def next_sequence_id
         end
 
         def execute_sql(last_insert_id_args)
-          connection = sequencer_repository.fetch(sequencer_name).connection
-          connection.execute "UPDATE `#{sequencer_config.table_name}` SET id = LAST_INSERT_ID(#{last_insert_id_args})"
-          res = connection.execute "SELECT LAST_INSERT_ID()"
+          sequencer_klass = sequencer_repository.fetch(sequencer_name)
+          connection = sequencer_klass.connection
+          connection.execute "UPDATE `#{sequencer_config.table_name}` SET id = LAST_INSERT_ID(#{last_insert_id_args})", sequencer_klass.name
+          res = connection.execute "SELECT LAST_INSERT_ID()", sequencer_klass.name
           new_id = res.first.first.to_i
           new_id
         end

diff --git a/spec/active_record/sharding/sequencer_spec.rb b/spec/active_record/sharding/sequencer_spec.rb
@@ -16,6 +16,12 @@ def self.name
       expect(current_id).to be_a_kind_of Fixnum
       expect(model.current_sequence_id).to eq current_id
     end
+
+    it "output class name to log" do
+      sequencer_klass = model.sequencer_repository.fetch(:user)
+      expect(sequencer_klass.connection).to receive(:execute).with(anything, "User::SequencerForTestUserSequencer").and_return([[0]]).twice
+      model.current_sequence_id
+    end
   end
 
   describe '#next_sequence_id' do