module ThreeScaleToolbox module Commands module PlansCommand module Import class ImportLimitsStep include Step ## # Writes Plan limits def call # SET semantics # First, delete existing limits # Second, add new limits plan.limits.each do |limit| limit.delete() puts "Deleted existing plan limit: [metric: #{limit.metric_id}, #{limit.attrs}]" end resource_limits_processed.each do |limit_attrs| metric_id = limit_attrs.delete('metric_id') plan.create_limit(metric_id, limit_attrs) puts "Created plan limit: [metric: #{metric_id}, #{limit_attrs}]" end end private def resource_limits_processed resource_limits.map do |limit| metric_system_name = limit.delete('metric_system_name') backend_system_name = limit.delete('metric_backend_system_name') metric_owner = if backend_system_name.nil? service else find_backend(backend_system_name) end metric = metric_owner.find_metric_or_method(metric_system_name) # this ImportMetricLimitsStep step is assuming all metrics/methods have been created # in previous step, so finding metric should always succeed. raise ThreeScaleToolbox::Error, "metric [#{metric_system_name}, #{backend_system_name}] not found" if metric.nil? limit.merge('metric_id' => metric.id) end end end end end end end