module Apartment::Delayed::Job::Hooks

def after
  Apartment::Database.switch(@_current_database)
end

def before(job)
  @_current_database = Apartment::Database.current_database
  Apartment::Database.switch(job.payload_object.database) if job.payload_object.database
end