Contents

# Custom array class we use in order to track extra attributes
# on a collection of results.
module Azure
  module Armrest
    class ArmrestCollection < Array
      attr_accessor :continuation_token
      attr_accessor :response_headers
      attr_accessor :response_code

      alias skip_token continuation_token
      alias skip_token= continuation_token=

      class << self
        # Creates and returns a ArmrestCollection object based on JSON input,
        # using +klass+ to generate the list elements. In addition, both the
        # response headers and continuation token are set.
        #
        def create_from_response(response, klass = nil)
          json_response = JSON.parse(response)
          array = new(json_response['value'].map { |hash| klass.new(hash) })

          array.response_code = response.code
          array.response_headers = response.headers
          array.continuation_token = parse_skip_token(json_response)

          array
        end

        private

        # Parse the skip token value out of the nextLink attribute from a response.
        def parse_skip_token(json)
          return nil unless json['nextLink']
          json['nextLink'][/.*?skipToken=(.*?)$/i, 1]
        end
      end
    end
  end
end

Version data entries

23 entries across 23 versions & 1 rubygems

Version	Path
azure-armrest-0.3.10	lib/azure/armrest/armrest_collection.rb
azure-armrest-0.3.9	lib/azure/armrest/armrest_collection.rb
azure-armrest-0.3.8	lib/azure/armrest/armrest_collection.rb