module Resque::Plugins::AppInstanceJob

def around_perform(args)
  case args.class.to_s
  when "Array"
    if args.first.class == Hash
      data = args.first.merge({:worker_class => self.to_s})
    else
      data = {:worker_class => self.to_s, :args => args}
    end
  when "Hash"
    data = args.merge({:worker_class => self.to_s})
  end
  if Rails.logger.is_a?(Ougai::Logger) 
    Rails.logger.with_fields = {job: data, trace_id: SecureRandom.uuid, name: "RailsWorker"}
  end
  begin
    connection_count ||= 0 
    @appinstance = ZuoraConnect::AppInstance.find(args['app_instance_id'].to_i)
    Rails.logger.info('Starting job') 
    @appinstance.new_session(holding_pattern: true)
  rescue ActiveRecord::RecordNotFound => exception
    # If we can't find app_instance let make sure we cleanup
    if Redis.current.zscore("AppInstance:Deleted", 1.to_i).present?
      Rails.logger.info("No instance found, purge")
      ZuoraConnect::AppInstance.new(id: args['app_instance_id'].to_i).prune_data
    end
    return
  rescue ActiveRecord::StatementInvalid => exception
    if (connection_count += 1) <= 3 &&
      (
        exception.message.include?("PG::UnableToSend: no connection to the server") ||
        exception.message.include?("PG::ConnectionBad: PQconsumeInput()") ||
        exception.message.include?("PG::ConnectionBad: PQsocket()") ||
        exception.message.include?("PG::UnableToSend: SSL SYSCALL")
      )
      sleep 30
      ActiveRecord::Base.establish_connection
      retry
    else
      raise
    end
  rescue PG::ConnectionBad => exception
    Rails.logger.info("Bad Connection Restart")
    Resque.enqueue_to(self.job.queue, self.job.payload['class'], args)
    return
  rescue ZuoraConnect::Exceptions::ConnectCommunicationError => exception
    Rails.logger.warn("Enqueue Job Again ~ 2 mins", exception)
    @appinstance.queue_pause(time: 2.minutes.to_i) 
    Resque.enqueue_to(self.job.queue, self.job.payload['class'], args)
    return
  rescue Net::ReadTimeout, Net::OpenTimeout, Errno::ECONNRESET, Errno::ECONNREFUSED, SocketError => exception
    Rails.logger.warn("Enqueue Job Again ~ 2 mins", exception)
    @appinstance.queue_pause(time: 2.minutes.to_i) 
    Resque.enqueue_to(self.job.queue, self.job.payload['class'], args)
    return
  end
  yield
rescue
  raise
ensure
  @appinstance.cache_app_instance if defined?(@appinstance)
  Rails.logger.flush if Rails.logger.methods.include?(:flush)
end