datadog_api_client 2.18.0 → 2.20.0

data/lib/datadog_api_client/v2/api/downtimes_api.rb

@@ -325,6 +325,8 @@ module DatadogAPIClient::V2
     #
     # @param monitor_id [Integer] The id of the monitor.
     # @param opts [Hash] the optional parameters
+    # @option opts [Integer] :page_offset Specific offset to use as the beginning of the returned page.
+    # @option opts [Integer] :page_limit Maximum number of downtimes in the response.
     # @return [Array<(MonitorDowntimeMatchResponse, Integer, Hash)>] MonitorDowntimeMatchResponse data, response status code and response headers
     def list_monitor_downtimes_with_http_info(monitor_id, opts = {})
 
@@ -340,6 +342,8 @@ module DatadogAPIClient::V2
 
       # query parameters
       query_params = opts[:query_params] || {}
+      query_params[:'page[offset]'] = opts[:'page_offset'] if !opts[:'page_offset'].nil?
+      query_params[:'page[limit]'] = opts[:'page_limit'] if !opts[:'page_limit'].nil?
 
       # header parameters
       header_params = opts[:header_params] || {}
@@ -376,6 +380,27 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
+    # Get active downtimes for a monitor.
+    #
+    # Provide a paginated version of {#list_monitor_downtimes}, returning all items.
+    #
+    # To use it you need to use a block: list_monitor_downtimes_with_pagination { |item| p item }
+    #
+    # @yield [MonitorDowntimeMatchResponseData] Paginated items
+    def list_monitor_downtimes_with_pagination(monitor_id, opts = {})
+        api_version = "V2"
+        page_size = @api_client.get_attribute_from_path(opts, "page_limit", 30)
+        @api_client.set_attribute_from_path(api_version, opts, "page_limit", Integer, page_size)
+        while true do
+            response = list_monitor_downtimes(monitor_id, opts)
+            @api_client.get_attribute_from_path(response, "data").each { |item| yield(item) }
+            if @api_client.get_attribute_from_path(response, "data").length < page_size
+              break
+            end
+            @api_client.set_attribute_from_path(api_version, opts, "page_offset", Integer, @api_client.get_attribute_from_path(opts, "page_offset", 0) + page_size)
+        end
+    end
+
     # Update a downtime.
     #
     # @see #update_downtime_with_http_info

data/lib/datadog_api_client/v2/api/metrics_api.rb

@@ -541,7 +541,7 @@ module DatadogAPIClient::V2
     # @option opts [String] :filter_tags_configured Filter tag configurations by configured tags.
     # @option opts [MetricTagConfigurationMetricTypes] :filter_metric_type Filter metrics by metric type.
     # @option opts [Boolean] :filter_include_percentiles Filter distributions with additional percentile aggregations enabled or disabled.
-    # @option opts [Boolean] :filter_queried Filter custom metrics that have or have not been queried in the specified window[seconds]. If no window is provided or the window is less than 2 hours, a default of 2 hours will be applied.
+    # @option opts [Boolean] :filter_queried (Beta) Filter custom metrics that have or have not been queried in the specified window[seconds]. If no window is provided or the window is less than 2 hours, a default of 2 hours will be applied.
     # @option opts [String] :filter_tags Filter metrics that have been submitted with the given tags. Supports boolean and wildcard expressions. Can only be combined with the filter[queried] filter.
     # @option opts [Integer] :window_seconds The number of seconds of look back (from now) to apply to a filter[tag] or filter[queried] query. Default value is 3600 (1 hour), maximum value is 2,592,000 (30 days).
     # @return [Array<(MetricsAndMetricTagConfigurationsResponse, Integer, Hash)>] MetricsAndMetricTagConfigurationsResponse data, response status code and response headers
@@ -980,7 +980,8 @@ module DatadogAPIClient::V2
     # Update the tag configuration of a metric or percentile aggregations of a distribution metric or custom aggregations
     # of a count, rate, or gauge metric. By setting `exclude_tags_mode` to true the behavior is changed
     # from an allow-list to a deny-list, and tags in the defined list will not be queryable.
-    # Can only be used with application keys from users with the `Manage Tags for Metrics` permission.
+    # Can only be used with application keys from users with the `Manage Tags for Metrics` permission. This endpoint requires
+    # a tag configuration to be created first.
     #
     # @param metric_name [String] The name of the metric.
     # @param body [MetricTagConfigurationUpdateRequest] 

data/lib/datadog_api_client/v2/api/organizations_api.rb

@@ -63,7 +63,7 @@ module DatadogAPIClient::V2
       form_params['idp_file'] = opts[:'idp_file'] if !opts[:'idp_file'].nil?
 
       # http body (model)
-      post_body = opts[:debug_body] || @api_client.object_to_http_body(body)
+      post_body = opts[:debug_body]
 
       # return_type
       return_type = opts[:debug_return_type]

data/lib/datadog_api_client/v2/api/security_monitoring_api.rb

@@ -160,6 +160,73 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
+    # Create a suppression rule.
+    #
+    # @see #create_security_monitoring_suppression_with_http_info
+    def create_security_monitoring_suppression(body, opts = {})
+      data, _status_code, _headers = create_security_monitoring_suppression_with_http_info(body, opts)
+      data
+    end
+
+    # Create a suppression rule.
+    #
+    # Create a new suppression rule.
+    #
+    # @param body [SecurityMonitoringSuppressionCreateRequest] The definition of the new suppression rule.
+    # @param opts [Hash] the optional parameters
+    # @return [Array<(SecurityMonitoringSuppressionResponse, Integer, Hash)>] SecurityMonitoringSuppressionResponse data, response status code and response headers
+    def create_security_monitoring_suppression_with_http_info(body, opts = {})
+
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: SecurityMonitoringAPI.create_security_monitoring_suppression ...'
+      end
+      # verify the required parameter 'body' is set
+      if @api_client.config.client_side_validation && body.nil?
+        fail ArgumentError, "Missing the required parameter 'body' when calling SecurityMonitoringAPI.create_security_monitoring_suppression"
+      end
+      # resource path
+      local_var_path = '/api/v2/security_monitoring/configuration/suppressions'
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json'])
+      # HTTP header 'Content-Type'
+      header_params['Content-Type'] = @api_client.select_header_content_type(['application/json'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(body)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'SecurityMonitoringSuppressionResponse'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || [:apiKeyAuth, :appKeyAuth, :AuthZ]
+
+      new_options = opts.merge(
+        :operation => :create_security_monitoring_suppression,
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type,
+        :api_version => "V2"
+      )
+
+      data, status_code, headers = @api_client.call_api(Net::HTTP::Post, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: SecurityMonitoringAPI#create_security_monitoring_suppression\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Delete a security filter.
     #
     # @see #delete_security_filter_with_http_info
@@ -290,6 +357,71 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
+    # Delete a suppression rule.
+    #
+    # @see #delete_security_monitoring_suppression_with_http_info
+    def delete_security_monitoring_suppression(suppression_id, opts = {})
+      delete_security_monitoring_suppression_with_http_info(suppression_id, opts)
+      nil
+    end
+
+    # Delete a suppression rule.
+    #
+    # Delete a specific suppression rule.
+    #
+    # @param suppression_id [String] The ID of the suppression rule
+    # @param opts [Hash] the optional parameters
+    # @return [Array<(nil, Integer, Hash)>] nil, response status code and response headers
+    def delete_security_monitoring_suppression_with_http_info(suppression_id, opts = {})
+
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: SecurityMonitoringAPI.delete_security_monitoring_suppression ...'
+      end
+      # verify the required parameter 'suppression_id' is set
+      if @api_client.config.client_side_validation && suppression_id.nil?
+        fail ArgumentError, "Missing the required parameter 'suppression_id' when calling SecurityMonitoringAPI.delete_security_monitoring_suppression"
+      end
+      # resource path
+      local_var_path = '/api/v2/security_monitoring/configuration/suppressions/{suppression_id}'.sub('{suppression_id}', CGI.escape(suppression_id.to_s).gsub('%2F', '/'))
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['*/*'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body]
+
+      # return_type
+      return_type = opts[:debug_return_type]
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || [:apiKeyAuth, :appKeyAuth, :AuthZ]
+
+      new_options = opts.merge(
+        :operation => :delete_security_monitoring_suppression,
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type,
+        :api_version => "V2"
+      )
+
+      data, status_code, headers = @api_client.call_api(Net::HTTP::Delete, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: SecurityMonitoringAPI#delete_security_monitoring_suppression\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Modify the triage assignee of a security signal.
     #
     # @see #edit_security_monitoring_signal_assignee_with_http_info
@@ -780,6 +912,71 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
+    # Get a suppression rule.
+    #
+    # @see #get_security_monitoring_suppression_with_http_info
+    def get_security_monitoring_suppression(suppression_id, opts = {})
+      data, _status_code, _headers = get_security_monitoring_suppression_with_http_info(suppression_id, opts)
+      data
+    end
+
+    # Get a suppression rule.
+    #
+    # Get the details of a specific suppression rule.
+    #
+    # @param suppression_id [String] The ID of the suppression rule
+    # @param opts [Hash] the optional parameters
+    # @return [Array<(SecurityMonitoringSuppressionResponse, Integer, Hash)>] SecurityMonitoringSuppressionResponse data, response status code and response headers
+    def get_security_monitoring_suppression_with_http_info(suppression_id, opts = {})
+
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: SecurityMonitoringAPI.get_security_monitoring_suppression ...'
+      end
+      # verify the required parameter 'suppression_id' is set
+      if @api_client.config.client_side_validation && suppression_id.nil?
+        fail ArgumentError, "Missing the required parameter 'suppression_id' when calling SecurityMonitoringAPI.get_security_monitoring_suppression"
+      end
+      # resource path
+      local_var_path = '/api/v2/security_monitoring/configuration/suppressions/{suppression_id}'.sub('{suppression_id}', CGI.escape(suppression_id.to_s).gsub('%2F', '/'))
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body]
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'SecurityMonitoringSuppressionResponse'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || [:apiKeyAuth, :appKeyAuth, :AuthZ]
+
+      new_options = opts.merge(
+        :operation => :get_security_monitoring_suppression,
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type,
+        :api_version => "V2"
+      )
+
+      data, status_code, headers = @api_client.call_api(Net::HTTP::Get, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: SecurityMonitoringAPI#get_security_monitoring_suppression\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # List findings.
     #
     # @see #list_findings_with_http_info
@@ -1164,6 +1361,66 @@ module DatadogAPIClient::V2
         end
     end
 
+    # Get all suppression rules.
+    #
+    # @see #list_security_monitoring_suppressions_with_http_info
+    def list_security_monitoring_suppressions(opts = {})
+      data, _status_code, _headers = list_security_monitoring_suppressions_with_http_info(opts)
+      data
+    end
+
+    # Get all suppression rules.
+    #
+    # Get the list of all suppression rules.
+    #
+    # @param opts [Hash] the optional parameters
+    # @return [Array<(SecurityMonitoringSuppressionsResponse, Integer, Hash)>] SecurityMonitoringSuppressionsResponse data, response status code and response headers
+    def list_security_monitoring_suppressions_with_http_info(opts = {})
+
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: SecurityMonitoringAPI.list_security_monitoring_suppressions ...'
+      end
+      # resource path
+      local_var_path = '/api/v2/security_monitoring/configuration/suppressions'
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body]
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'SecurityMonitoringSuppressionsResponse'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || [:apiKeyAuth, :appKeyAuth, :AuthZ]
+
+      new_options = opts.merge(
+        :operation => :list_security_monitoring_suppressions,
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type,
+        :api_version => "V2"
+      )
+
+      data, status_code, headers = @api_client.call_api(Net::HTTP::Get, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: SecurityMonitoringAPI#list_security_monitoring_suppressions\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
+
     # Mute or unmute a batch of findings.
     #
     # @see #mute_findings_with_http_info
@@ -1469,5 +1726,77 @@ module DatadogAPIClient::V2
       end
       return data, status_code, headers
     end
+
+    # Update a suppression rule.
+    #
+    # @see #update_security_monitoring_suppression_with_http_info
+    def update_security_monitoring_suppression(suppression_id, body, opts = {})
+      data, _status_code, _headers = update_security_monitoring_suppression_with_http_info(suppression_id, body, opts)
+      data
+    end
+
+    # Update a suppression rule.
+    #
+    # Update a specific suppression rule.
+    #
+    # @param suppression_id [String] The ID of the suppression rule
+    # @param body [SecurityMonitoringSuppressionUpdateRequest] New definition of the suppression rule. Supports partial updates.
+    # @param opts [Hash] the optional parameters
+    # @return [Array<(SecurityMonitoringSuppressionResponse, Integer, Hash)>] SecurityMonitoringSuppressionResponse data, response status code and response headers
+    def update_security_monitoring_suppression_with_http_info(suppression_id, body, opts = {})
+
+      if @api_client.config.debugging
+        @api_client.config.logger.debug 'Calling API: SecurityMonitoringAPI.update_security_monitoring_suppression ...'
+      end
+      # verify the required parameter 'suppression_id' is set
+      if @api_client.config.client_side_validation && suppression_id.nil?
+        fail ArgumentError, "Missing the required parameter 'suppression_id' when calling SecurityMonitoringAPI.update_security_monitoring_suppression"
+      end
+      # verify the required parameter 'body' is set
+      if @api_client.config.client_side_validation && body.nil?
+        fail ArgumentError, "Missing the required parameter 'body' when calling SecurityMonitoringAPI.update_security_monitoring_suppression"
+      end
+      # resource path
+      local_var_path = '/api/v2/security_monitoring/configuration/suppressions/{suppression_id}'.sub('{suppression_id}', CGI.escape(suppression_id.to_s).gsub('%2F', '/'))
+
+      # query parameters
+      query_params = opts[:query_params] || {}
+
+      # header parameters
+      header_params = opts[:header_params] || {}
+      # HTTP header 'Accept' (if needed)
+      header_params['Accept'] = @api_client.select_header_accept(['application/json'])
+      # HTTP header 'Content-Type'
+      header_params['Content-Type'] = @api_client.select_header_content_type(['application/json'])
+
+      # form parameters
+      form_params = opts[:form_params] || {}
+
+      # http body (model)
+      post_body = opts[:debug_body] || @api_client.object_to_http_body(body)
+
+      # return_type
+      return_type = opts[:debug_return_type] || 'SecurityMonitoringSuppressionResponse'
+
+      # auth_names
+      auth_names = opts[:debug_auth_names] || [:apiKeyAuth, :appKeyAuth, :AuthZ]
+
+      new_options = opts.merge(
+        :operation => :update_security_monitoring_suppression,
+        :header_params => header_params,
+        :query_params => query_params,
+        :form_params => form_params,
+        :body => post_body,
+        :auth_names => auth_names,
+        :return_type => return_type,
+        :api_version => "V2"
+      )
+
+      data, status_code, headers = @api_client.call_api(Net::HTTP::Patch, local_var_path, new_options)
+      if @api_client.config.debugging
+        @api_client.config.logger.debug "API called: SecurityMonitoringAPI#update_security_monitoring_suppression\nData: #{data.inspect}\nStatus code: #{status_code}\nHeaders: #{headers}"
+      end
+      return data, status_code, headers
+    end
   end
 end

data/lib/datadog_api_client/v2/api/teams_api.rb

@@ -717,6 +717,28 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
+    # Get team memberships.
+    #
+    # Provide a paginated version of {#get_team_memberships}, returning all items.
+    #
+    # To use it you need to use a block: get_team_memberships_with_pagination { |item| p item }
+    #
+    # @yield [UserTeam] Paginated items
+    def get_team_memberships_with_pagination(team_id, opts = {})
+        api_version = "V2"
+        page_size = @api_client.get_attribute_from_path(opts, "page_size", 10)
+        @api_client.set_attribute_from_path(api_version, opts, "page_size", Integer, page_size)
+        @api_client.set_attribute_from_path(api_version, opts, "page_number", Integer, 0)
+        while true do
+            response = get_team_memberships(team_id, opts)
+            @api_client.get_attribute_from_path(response, "data").each { |item| yield(item) }
+            if @api_client.get_attribute_from_path(response, "data").length < page_size
+              break
+            end
+            @api_client.set_attribute_from_path(api_version, opts, "page_number", Integer, @api_client.get_attribute_from_path(opts, "page_number", 0) + 1)
+        end
+    end
+
     # Get permission settings for a team.
     #
     # @see #get_team_permission_settings_with_http_info

data/lib/datadog_api_client/v2/api/usage_metering_api.rb

@@ -642,7 +642,7 @@ module DatadogAPIClient::V2
       return data, status_code, headers
     end
 
-    # Get hourly usage for lambda traced invocations.
+    # Get hourly usage for Lambda traced invocations.
     #
     # @see #get_usage_lambda_traced_invocations_with_http_info
     def get_usage_lambda_traced_invocations(start_hr, opts = {})
@@ -650,9 +650,9 @@ module DatadogAPIClient::V2
       data
     end
 
-    # Get hourly usage for lambda traced invocations.
+    # Get hourly usage for Lambda traced invocations.
     #
-    # Get hourly usage for lambda traced invocations.
+    # Get hourly usage for Lambda traced invocations.
     # **Note:** hourly usage data for all products is now available in the [Get hourly usage by product family API](https://docs.datadoghq.com/api/latest/usage-metering/#get-hourly-usage-by-product-family)
     #
     # @param start_hr [Time] Datetime in ISO-8601 format, UTC, precise to hour: `[YYYY-MM-DDThh]` for usage beginning at this hour.

data/lib/datadog_api_client/v2/models/aws_cur_config.rb

@@ -0,0 +1,129 @@
+=begin
+#Datadog API V2 Collection
+
+#Collection of all Datadog Public endpoints.
+
+The version of the OpenAPI document: 1.0
+Contact: support@datadoghq.com
+Generated by: https://github.com/DataDog/datadog-api-client-ruby/tree/master/.generator
+
+ Unless explicitly stated otherwise all files in this repository are licensed under the Apache-2.0 License.
+ This product includes software developed at Datadog (https://www.datadoghq.com/).
+ Copyright 2020-Present Datadog, Inc.
+
+=end
+
+require 'date'
+require 'time'
+
+module DatadogAPIClient::V2
+  # AWS CUR config.
+  class AwsCURConfig
+    include BaseGenericModel
+
+    # Attributes for An AWS CUR config.
+    attr_reader :attributes
+
+    # The ID of the AWS CUR config.
+    attr_accessor :id
+
+    # Type of AWS CUR config.
+    attr_reader :type
+
+    # Attribute mapping from ruby-style variable name to JSON key.
+    # @!visibility private
+    def self.attribute_map
+      {
+        :'attributes' => :'attributes',
+        :'id' => :'id',
+        :'type' => :'type'
+      }
+    end
+
+    # Attribute type mapping.
+    # @!visibility private
+    def self.openapi_types
+      {
+        :'attributes' => :'AwsCURConfigAttributes',
+        :'id' => :'Integer',
+        :'type' => :'AwsCURConfigType'
+      }
+    end
+
+    # Initializes the object
+    # @param attributes [Hash] Model attributes in the form of hash
+    # @!visibility private
+    def initialize(attributes = {})
+      if (!attributes.is_a?(Hash))
+        fail ArgumentError, "The input argument (attributes) must be a hash in `DatadogAPIClient::V2::AwsCURConfig` initialize method"
+      end
+
+      # check to see if the attribute exists and convert string to symbol for hash key
+      attributes = attributes.each_with_object({}) { |(k, v), h|
+        if (!self.class.attribute_map.key?(k.to_sym))
+          fail ArgumentError, "`#{k}` is not a valid attribute in `DatadogAPIClient::V2::AwsCURConfig`. Please check the name to make sure it's valid. List of attributes: " + self.class.attribute_map.keys.inspect
+        end
+        h[k.to_sym] = v
+      }
+
+      if attributes.key?(:'attributes')
+        self.attributes = attributes[:'attributes']
+      end
+
+      if attributes.key?(:'id')
+        self.id = attributes[:'id']
+      end
+
+      if attributes.key?(:'type')
+        self.type = attributes[:'type']
+      end
+    end
+
+    # Check to see if the all the properties in the model are valid
+    # @return true if the model is valid
+    # @!visibility private
+    def valid?
+      return false if @attributes.nil?
+      return false if @type.nil?
+      true
+    end
+
+    # Custom attribute writer method with validation
+    # @param attributes [Object] Object to be assigned
+    # @!visibility private
+    def attributes=(attributes)
+      if attributes.nil?
+        fail ArgumentError, 'invalid value for "attributes", attributes cannot be nil.'
+      end
+      @attributes = attributes
+    end
+
+    # Custom attribute writer method with validation
+    # @param type [Object] Object to be assigned
+    # @!visibility private
+    def type=(type)
+      if type.nil?
+        fail ArgumentError, 'invalid value for "type", type cannot be nil.'
+      end
+      @type = type
+    end
+
+    # Checks equality by comparing each attribute.
+    # @param o [Object] Object to be compared
+    # @!visibility private
+    def ==(o)
+      return true if self.equal?(o)
+      self.class == o.class &&
+          attributes == o.attributes &&
+          id == o.id &&
+          type == o.type
+    end
+
+    # Calculates hash code according to all attributes.
+    # @return [Integer] Hash code
+    # @!visibility private
+    def hash
+      [attributes, id, type].hash
+    end
+  end
+end