module Apartment::Delayed::Job::Hooks

def after
  Apartment::Database.reset
end

def before(job)
  Apartment::Database.switch(job.payload_object.database) if job.payload_object.database
end