librato/librato-metrics

module Librato
  module Metrics

    # An internal class used for extracting pagination logic
    #
    # @api private
    class Collection

      MAX_RESULTS = 100

      # Aggregates all results of paginated elements, requesting more collections as needed
      #
      # @param [Excon] connection Connection to Metrics service
      # @param [String] path API uri
      # @param [Hash] query Query options
      def self.paginated_metrics(connection, path, query)
        paginated_collection("metrics", connection, path, query)
      end

      def self.paginated_collection(name, connection, path, query)
        results = []
        # expects 200
        url = connection.build_url(path, query)
        response = connection.get(url)
        parsed = SmartJSON.read(response.body)
        results = parsed[name]
        return results if parsed["query"]["found"] <= MAX_RESULTS
        query[:offset] = MAX_RESULTS
        begin
          # expects 200
          url = connection.build_url(path, query)
          response = connection.get(url)
          parsed = SmartJSON.read(response.body)
          results.push(*parsed[name])
          query[:offset] += MAX_RESULTS
        end while query[:offset] < parsed["query"]["found"]
        results

      end

    end
  end
end