FarmBot/Farmbot-Web-App

# A singleton that runs on a separate process than the web server.
# Listens to *ALL* incoming logs and stores them to the DB.
# Also handles throttling.
class LogService < AbstractServiceRunner
  THROTTLE_POLICY = ThrottlePolicy.new(name, min: 250, hour: 5_000, day: 25_000)

  LOG_TPL = Rails.env.test? ?
    "\e[32m.\e[0m" : "device_%s FBOS_LOG v%s %s\n"
  ERR_TPL = "MALFORMED LOG CAPTURE: %s"
  # Clean up excess logs in a non-deterministic manner.
  # Performs the slow DB query every nth request.
  TIDY_RATE = Rails.env.test? ? 0 : 10

  def process(delivery_info, payload)
    params = { routing_key: delivery_info.routing_key, payload: payload }
    m = AmqpLogParser.run!(params)
    THROTTLE_POLICY.track(m.device_id)
    m.device.trim_excess_logs if rand(0..TIDY_RATE) == TIDY_RATE
    maybe_deliver(m)
  rescue Mutations::ValidationException => e
    msg = ERR_TPL % [params.merge({ e: e }).to_json]
    puts msg unless Rails.env.test?
    raise e
  end

  def maybe_deliver(data)
    return unless data.valid?

    violation = THROTTLE_POLICY.violation_for(data.device_id)

    if violation
      return warn_user(data, violation)
    end

    data.device.auto_sync_transaction do
      deliver(data)
    end
  end

  def deliver(data)
    dev, log = [data.device, data.payload]
    dev.maybe_unthrottle
    Log.deliver(Logs::Create.run!(log, device: dev).id)
    print LOG_TPL % [data.device_id,
                     dev.fbos_version || "?",
                     data.payload["message"] || "??"]
  rescue => x
    Rollbar.error(x)
  end

  def warn_user(data, violation)
    data.device.maybe_throttle(violation)
  end
end