Sha256: e6e15214e809848b25607200f416bb9ea3d2f1628dc21809054873c3c5417aff
Contents?: true
Size: 1.46 KB
Versions: 64
Compression:
Stored size: 1.46 KB
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
if use_transaction? env
call_with_transaction(add_database_connector(env))
else
call_without_transaction(add_database_connector(env))
end
end
def add_database_connector(env)
# maintain any existing one set by previous middleware
{ "pactbroker.database_connector" => @default_database_connector }.merge(env)
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
end
end
end
Version data entries
64 entries across 64 versions & 1 rubygems