Add support for tracking manual exceptions within transactions

CHANGELOG.md

@@ -5,7 +5,8 @@
 ### New features
 
 - [#113](https://github.com/rootstrap/exception_hunter/pull/113) Add configuration to turn on/off async logging using ActionJob. (
-  [@matteo95g][])
+  [@matteo95g][]) 
+- [#119](https://github.com/rootstrap/exception_hunter/pull/119) Add support for tracking manual exceptions within transactions. ([@lalopsb][])
 
 ### Bug fixes
 

lib/exception_hunter/tracking.rb

@@ -19,6 +19,22 @@ module Tracking
     # @param [User] user in the current session. (optional)
     # @return [void]
     def track(exception, custom_data: {}, user: nil)
+      open_transactions? ? create_error_within_new_thread(exception, custom_data, user) : create_error(exception, custom_data, user)
+
+      nil
+    end
+
+    private
+
+    def create_error_within_new_thread(exception, custom_data, user)
+      Thread.new do
+        ActiveRecord::Base.connection_pool.with_connection do
+          create_error(exception, custom_data, user)
+        end
+      end
+    end
+
+    def create_error(exception, custom_data, user)
       ErrorCreator.call(
         tag: ErrorCreator::MANUAL_TAG,
         class_name: exception.class.to_s,
@@ -28,8 +44,10 @@ def track(exception, custom_data: {}, user: nil)
         user: user,
         environment_data: {}
       )
+    end
 
-      nil
+    def open_transactions?
+      ActiveRecord::Base.connection.open_transactions.positive?
     end
   end
 end

spec/exception_hunter_spec.rb

@@ -17,6 +17,8 @@ module ExceptionHunter
       let(:user) { OpenStruct.new(id: 3, email: 'example@example.com', name: 'John') }
 
       context 'when tracking is enabled' do
+        before { allow(ActiveRecord::Base.connection).to receive(:open_transactions).and_return(0) }
+
         let(:error) { Error.last }
 
         it 'creates a new error' do
@@ -55,6 +57,24 @@ module ExceptionHunter
 
           expect(error.error_group.tags).to eq(['Manual'])
         end
+
+        it ' does not create a new thread' do
+          expect(Thread).to_not receive(:new)
+
+          subject
+        end
+
+        context 'when the error is tracked within a transaction' do
+          before do
+            allow(ActiveRecord::Base.connection).to receive(:open_transactions).and_return(1)
+            allow(Thread).to receive(:new).and_yield
+          end
+
+          it 'creates a new error within a new thread' do
+            expect { subject }.to change(Error, :count).by(1)
+            expect(Thread).to have_received(:new)
+          end
+        end
       end
 
       context 'when tracking is disabled' do