tasks/gems/aws-sdk-cloudwatch.rb

# frozen_string_literal: true

module Benchmark
  module Gems
    class CloudWatch < Benchmark::Gem
      def gem_name
        'aws-sdk-cloudwatch'
      end

      def gem_dir
        "gems/#{gem_name}"
      end

      def client_module_name
        'Aws::CloudWatch'
      end

      def operation_benchmarks
        {
          put_metric_data_small: {
            setup: proc do |_client|
              {
                namespace: 'namespace',
                metric_data: [
                  {
                    metric_name: 'metric',
                    timestamp: Time.now,
                    value: 1.0,
                    unit: 'Seconds'
                  }
                ]
              }
            end,
            test: proc do |client, req|
              client.put_metric_data(req)
            end
          },
          put_metric_data_large: {
            setup: proc do |_client|
              {
                namespace: 'namespace',
                metric_data: (0...10).map do
                  {
                    metric_name: 'metric',
                    timestamp: Time.now,
                    values: (0...150).to_a,
                    unit: 'Seconds'
                  }
                end
              }
            end,
            test: proc do |client, req|
              client.put_metric_data(req)
            end
          }
        }
      end
    end
  end
end