require "influxdb/rails/middleware/subscriber"

module InfluxDB
  module Rails
    module Middleware
      # Subscriber acts as base class for different *Subscriber classes,
      # which are intended as ActiveSupport::Notifications.subscribe
      # consumers.
      class SimpleSubscriber < Subscriber
        def call(_name, started, finished, _id, payload)
          return unless enabled?

          InfluxDB::Rails.client.write_point \
            configuration.measurement_name,
            values:    values(started, finished, payload),
            tags:      tags(payload),
            timestamp: timestamp(finished)
        rescue StandardError => e
          ::Rails.logger.error("[InfluxDB::Rails] Unable to write points: #{e.message}")
        end

        private

        def values(started, finished, _payload)
          result = { value: ((finished - started) * 1000).ceil }
          result.merge(InfluxDB::Rails.current.values).reject do |_, value|
            value.nil? || value == ""
          end
        end
      end
    end
  end
end