event_tracer 0.4.6 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 64dde68b086b57c4b676d62613a5a36be4fa5c754ccb4acd167aa5e69b4247d8
-  data.tar.gz: '083bca3d0a990b4d0ac49a20935c6d160b4880604fdeb48bb050720f1713820f'
+  metadata.gz: e27db9c289f5e94f2f3370fccb3c219222ccb538c21ffc022d67d83a3a42b1c6
+  data.tar.gz: efae6fe651ef5833b44a8ec420d72df76f2ca234e639348b3d756c92343640b5
 SHA512:
-  metadata.gz: 68f56c5bab8b095372ee8dbbe993eb08d9c1692acf7774e3c003abcebb8ac07093eddabb79012c9a6abbb42662da626d4b69ad11221616a6dcebd012c56fb868
-  data.tar.gz: ec619f4d64f70a81371063a0ac3ec2fa6666db0030b3af45b364b01de6664757c9824ac010179eeb3af147eb9b2fbd48acc53a3094926891199613ba31d2826a
+  metadata.gz: 1071f39a1ee582dad14d580bf3086330d83fefaf3942d48abb8d01402b13888a9f83867cb0795c6ed57a150c793d0acf76c0245cfef7cdfda2e330e7afb8674e
+  data.tar.gz: 6abbecd3c7e32610c66b336f1d2e304376a42e73789538f4ea93b18bad96f67f6b4949363fb0831d19ad6bfa6dfa69de2269fca6ce27cfe7652808792b6fcef3

data/lib/event_tracer/buffered_logger.rb

@@ -46,6 +46,8 @@ module EventTracer
       )
 
       worker.perform_async(filtered_payloads) if filtered_payloads.any?
+    rescue StandardError => error
+      raise EventTracer::ErrorWithPayload.new(error, payloads)
     end
 
     def filter_invalid_data(payloads)

data/lib/event_tracer/config.rb

@@ -12,6 +12,7 @@ module EventTracer
       setting :dynamo_db_table_name, default: 'logs'
       setting :dynamo_db_client
       setting :dynamo_db_queue_name, default: 'low'
+      setting :error_handler, default: ->(error, _payload) { raise error }
     else
       setting :app_name, 'app_name'
 
@@ -19,6 +20,7 @@ module EventTracer
       setting :dynamo_db_table_name, 'logs'
       setting :dynamo_db_client
       setting :dynamo_db_queue_name, 'low'
+      setting :error_handler, ->(error, _payload) { raise error }
     end
   end
 end

data/lib/event_tracer/datadog_logger.rb

@@ -18,15 +18,16 @@ module EventTracer
       gauge: :gauge,
       set: :set,
       histogram: :histogram
-    }
+    }.freeze
     DEFAULT_METRIC_TYPE = :count
     DEFAULT_COUNTER = 1
 
     attr_reader :allowed_tags
 
-    def initialize(decoratee, allowed_tags: [])
+    def initialize(decoratee, allowed_tags: [], default_tags: {})
       super(decoratee)
       @allowed_tags = allowed_tags.freeze
+      @default_tags = default_tags.freeze
     end
 
     LOG_TYPES.each do |log_type|
@@ -59,12 +60,14 @@ module EventTracer
 
     alias_method :datadog, :decoratee
 
+    attr_reader :default_tags
+
     def valid_args?(metrics)
       metrics && (metrics.is_a?(Hash) || metrics.is_a?(Array))
     end
 
     def build_tags(args)
-      args.slice(*allowed_tags).map do |tag, value|
+      default_tags.merge(args.slice(*allowed_tags)).map do |tag, value|
         "#{tag}:#{value}"
       end
     end

data/lib/event_tracer/error_with_payload.rb

@@ -0,0 +1,10 @@
+module EventTracer
+  class ErrorWithPayload < StandardError
+    attr_reader :payload
+
+    def initialize(error, payload)
+      super(error)
+      @payload = payload
+    end
+  end
+end

data/lib/event_tracer/prometheus_logger.rb

@@ -0,0 +1,97 @@
+module EventTracer
+  class PrometheusLogger < BasicDecorator
+
+    SUPPORTED_METRIC_TYPES = {
+      counter: :increment_count,
+      gauge: :set_gauge
+    }.freeze
+    DEFAULT_INCREMENT = 1
+
+    attr_reader :allowed_tags
+
+    def initialize(prometheus, allowed_tags: [], default_tags: {}, raise_if_missing: true)
+      super(prometheus)
+      @allowed_tags = allowed_tags.freeze
+      @default_tags = default_tags.freeze
+      @raise_if_missing = raise_if_missing
+    end
+
+    LOG_TYPES.each do |log_type|
+      define_method log_type do |**args|
+        metrics = args[:metrics]
+
+        return fail_result('Invalid metrics for Prometheus') unless valid_args?(metrics)
+        return success_result if metrics.empty?
+
+        labels = build_metric_labels(args)
+
+        case metrics
+        when Array
+          metrics.each do |metric_name|
+            increment_count(metric_name, DEFAULT_INCREMENT, labels: labels)
+          end
+        when Hash
+          metrics.each do |metric_name, metric_payload|
+            payload = metric_payload.transform_keys(&:to_sym)
+            metric_type = SUPPORTED_METRIC_TYPES[payload.fetch(:type).to_sym]
+
+            if metric_type
+              send(
+                metric_type,
+                metric_name,
+                payload.fetch(:value),
+                labels: labels
+              )
+            end
+          end
+        end
+
+        success_result
+      end
+    end
+
+    private
+
+      alias_method :prometheus, :decoratee
+
+      attr_reader :default_tags, :raise_if_missing
+
+      def valid_args?(metrics)
+        metrics && (metrics.is_a?(Hash) || metrics.is_a?(Array))
+      end
+
+      def increment_count(metric_name, value, labels:)
+        metric = get_metric(metric_name.to_sym, :counter)
+        metric.increment(by: value, labels: labels)
+      end
+
+      def set_gauge(metric_name, value, labels:)
+        metric = get_metric(metric_name.to_sym, :gauge)
+        metric.set(value, labels: labels)
+      end
+
+      def get_metric(metric_name, metric_type)
+        metric = prometheus.get(metric_name)
+
+        return metric if metric
+        raise "Metric #{metric_name} not registered" if raise_if_missing
+
+        prometheus.public_send(
+          metric_type,
+          metric_name,
+          docstring: "A #{metric_type} for #{metric_name}",
+          labels: labels_for_registration
+        )
+      end
+
+      def build_metric_labels(args)
+        allowed_tags.inject(default_tags) do |metric_labels, tag|
+          metric_labels.merge(tag => args[tag])
+        end
+      end
+
+      def labels_for_registration
+        (allowed_tags + default_tags.keys).uniq
+      end
+  end
+end

data/lib/event_tracer/version.rb

@@ -1,3 +1,3 @@
 module EventTracer
-  VERSION = '0.4.6'.freeze
+  VERSION = '0.6.0'.freeze
 end

data/lib/event_tracer.rb

@@ -32,7 +32,15 @@ module EventTracer
 
       loggers.each do |code, logger|
         if args[:action] && args[:message]
-          result.record code, logger.send(log_type, **args)
+          begin
+            result.record code, logger.send(log_type, **args)
+          rescue EventTracer::ErrorWithPayload => error
+            EventTracer::Config.config.error_handler.call(error.cause || error, error.payload)
+            result.record code, LogResult.new(false, error.message)
+          rescue StandardError => error
+            EventTracer::Config.config.error_handler.call(error, args)
+            result.record code, LogResult.new(false, error.message)
+          end
         else
           result.record code, LogResult.new(false, 'Fields action & message need to be populated')
         end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: event_tracer
 version: !ruby/object:Gem::Version
-  version: 0.4.6
+  version: 0.6.0
 platform: ruby
 authors:
 - melvrickgoh
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-02-23 00:00:00.000000000 Z
+date: 2022-05-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -100,7 +100,9 @@ files:
 - lib/event_tracer/dynamo_db/default_processor.rb
 - lib/event_tracer/dynamo_db/logger.rb
 - lib/event_tracer/dynamo_db/worker.rb
+- lib/event_tracer/error_with_payload.rb
 - lib/event_tracer/log_result.rb
+- lib/event_tracer/prometheus_logger.rb
 - lib/event_tracer/version.rb
 homepage: https://github.com/melvrickgoh/event_tracer
 licenses: