Contents

require 'pact_broker/constants'
require 'sequel'
require 'ostruct'

module Rack
  module PactBroker
    class DatabaseTransaction

      REQUEST_METHOD = "REQUEST_METHOD".freeze
      TRANS_METHODS = %w{POST PUT PATCH DELETE}.freeze

      def initialize app, database_connection
        @app = app
        @database_connection = database_connection
        @default_database_connector = ->(&block) {
          database_connection.synchronize do
            block.call
          end
        }
      end

      def call env
        set_database_connector
        if use_transaction? env
          call_with_transaction(add_database_connector(env))
        else
          call_without_transaction(add_database_connector(env))
        end
      ensure
        clear_database_connector
      end

      def add_database_connector(env)
        env.merge("pactbroker.database_connector" => @default_database_connector)
      end

      def use_transaction? env
        TRANS_METHODS.include? env[REQUEST_METHOD]
      end

      def call_without_transaction env
        @app.call(env)
      end

      def call_with_transaction env
        response = nil
        @database_connection.transaction do
          response = @app.call(env)
          if response.first == 500
            raise Sequel::Rollback unless do_not_rollback?(response)
          end
        end
        response
      end

      def do_not_rollback? response
        response[1].delete(::PactBroker::DO_NOT_ROLLBACK)
      end

      def set_database_connector
        Thread.current[:pact_broker_thread_data] ||= OpenStruct.new
        Thread.current[:pact_broker_thread_data].database_connector ||= @default_database_connector
      end

      def clear_database_connector
        if thread_data = Thread.current[:pact_broker_thread_data]
          if thread_data.database_connector == @default_database_connector
            thread_data.database_connector = nil
          end
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
pact_broker-2.33.0	lib/rack/pact_broker/database_transaction.rb
pact_broker-2.32.0	lib/rack/pact_broker/database_transaction.rb