cloudwatch-metrics 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7e76f596fe51773a068e074190d2b73a050772e565f57a9728d7e05e8d0dc426
-  data.tar.gz: 1be98427ad0a9f15b9090bf0275d53d929d47d461c873fe574960895a3d47fb1
+  metadata.gz: fb532d3ef368646603acb0246884a36c398dd42b8bcd17cfe9aad3c69dabe8b4
+  data.tar.gz: b3f06ecce07db061df5d992eb847c82ae28ff704f4649d5f6bdd8bc720519c3f
 SHA512:
-  metadata.gz: 33bcd4130b0fc125d24fae7827ffa16005d2b43217e457a817ac419f4560885d9a7410249e22a8aca85ab8358e3e012a2a9d86d38f58c8ace23265449db76fa5
-  data.tar.gz: 1bd146a97750aa9be1b00f10491087a5b69933eda91f97821a467bcd2caf5415971641707bab97a9874843219ed9ddc51a6fefc0a3edce75f6754fab02667d7e
+  metadata.gz: 39104d7d1796b1fd74a65ad98df38f81f84d635370d71aefb27bc851de97543b07061efa4a51ddb7468bc3190305517dde60f251d383e24a03e9abfadfc59931
+  data.tar.gz: ba0a2202fde50efd5fe08e3cd641653901955281b60b30d4a9dbca488268183c3f2520b3dcf265cb7afb69213292f36b72c113b9680afa6a745c84d007505ac0

data/lib/cloudwatch_metrics/configuration.rb

@@ -2,8 +2,8 @@
 
 module CloudwatchMetrics
   class Configuration
-    attr_accessor :namespace, :region, :environment
-    attr_writer :no_cw_alias, :publish_environments, :logger
+    attr_accessor :namespace, :region, :environment, :error_handler, :default_dimensions
+    attr_writer :no_cw_alias, :publish_environments, :logger, :non_production_prefix
 
     def no_cw_alias
       @no_cw_alias ||= false
@@ -16,5 +16,9 @@ module CloudwatchMetrics
     def logger
       @logger ||= Logger.new($stdout)
     end
+
+    def non_production_prefix
+      @non_production_prefix ||= 'test-'
+    end
   end
 end

data/lib/cloudwatch_metrics/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module CloudwatchMetrics
-  VERSION = '1.0.0'
+  VERSION = '1.1.0'
 end

data/lib/cloudwatch_metrics.rb

@@ -35,7 +35,7 @@ module CloudwatchMetrics
     end
 
     def record(
-      name:, value:, unit: nil, namespace: configuration.namespace, dimensions: nil, timestamp: nil
+      name:, value:, unit: nil, namespace: nil, dimensions: nil, timestamp: nil
     )
       metric_data = [{
         metric_name: name,
@@ -44,11 +44,11 @@ module CloudwatchMetrics
         dimensions: map_dimensions(dimensions),
         timestamp: timestamp
       }]
-      put_data(namespace: namespace, metric_data: metric_data)
+      put_data(namespace: full_namespace(namespace), metric_data: metric_data)
     end
 
     def record_all(
-      name:, values:, counts:, unit: nil, namespace: configuration.namespace, dimensions: nil, timestamp: nil
+      name:, values:, counts:, unit: nil, namespace: nil, dimensions: nil, timestamp: nil
     )
       metric_data = [{
         metric_name: name,
@@ -58,7 +58,7 @@ module CloudwatchMetrics
         dimensions: map_dimensions(dimensions),
         timestamp: timestamp
       }]
-      put_data(namespace: namespace, metric_data: metric_data)
+      put_data(namespace: full_namespace(namespace), metric_data: metric_data)
     end
 
     private
@@ -72,8 +72,7 @@ module CloudwatchMetrics
 
       nil
     rescue StandardError => e
-      configuration.logger.error("CloudwatchMetrics error: #{e.message}")
-      raise CloudwatchMetrics::Error, e.message
+      handle_error(e)
     end
 
     def cloudwatch_client
@@ -92,17 +91,38 @@ module CloudwatchMetrics
 
     # convert key/value hash to format expected by AWS SDK
     def map_dimensions(dimensions)
-      return [] if dimensions.nil?
+      combined_dimensions = (configuration.default_dimensions || {}).merge(dimensions || {})
 
       mapped_values = []
-      dimensions.keys.each do |key|
+      combined_dimensions.keys.each do |key|
         mapped_values << {
           name: key.to_s,
-          value: dimensions[key]
+          value: combined_dimensions[key]
         }
       end
 
       mapped_values
     end
+
+    def handle_error(error)
+      if !configuration.error_handler.nil?
+        configuration.error_handler.call(error)
+        return
+      end
+
+      configuration.logger.error("CloudwatchMetrics error: #{error.message}")
+      raise CloudwatchMetrics::Error, error.message
+    end
+
+    def full_namespace(provided_namespace)
+      constructed_namespace = provided_namespace || configuration.namespace
+      raise CloudwatchMetrics::Error, 'Namespace must be provided' if constructed_namespace.nil?
+
+      if environment != 'production' && configuration.non_production_prefix.present?
+        constructed_namespace = configuration.non_production_prefix + constructed_namespace
+      end
+
+      constructed_namespace
+    end
   end
 end

data/lib/generators/cloudwatch_metrics/templates/cloudwatch_metrics.rb

@@ -32,4 +32,21 @@ CloudwatchMetrics.configure do |config|
   # This is useful if the gem is being used in a non-rails application.
   # Defaults to ENV['RAILS_ENV'] if present or 'development'` otherwise.
   # config.environment = 'production'
+
+  # This is an optional configuration to set the namespace prefix for non-production
+  # environments. This is useful if you want to avoid polluting your production
+  # metrics with test data. Defaults to 'test-'.
+  # config.non_production_prefix = 'test-'
+
+  # This is an optional configuration to set an error handler proc.
+  # If not set the default error handler will log the error
+  # and raise a CloudwatchMetrics::Error
+  # config.error_handler = proc do |exception|
+  #   Rails.logger.error(exception)
+  #   Sentry.capture_exception(exception)
+  # end
+
+  # This is an optional configuration to set default dimensions for all metrics.
+  # Dimensions passed to the `record` or `record_all` methods will be merged with these.
+  # config.default_dimensions = {name: 'value'}
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cloudwatch-metrics
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Seth Puckett
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-16 00:00:00.000000000 Z
+date: 2023-06-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport