event_tracer 0.5.1 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3c2e5bbf4d4546e3ece6589508e00fc01eb30ca41b572ff2afbb4185b73c93d5
-  data.tar.gz: e3fe3b13cf8fc1e43732a75386502618427cab7a449fe95b8d2c69789383947b
+  metadata.gz: e27db9c289f5e94f2f3370fccb3c219222ccb538c21ffc022d67d83a3a42b1c6
+  data.tar.gz: efae6fe651ef5833b44a8ec420d72df76f2ca234e639348b3d756c92343640b5
 SHA512:
-  metadata.gz: 17cd02fe83f7725b8e2a9d495979f5b183e2a63fc071b1b58795adfb910ed0c9048f1692b608d78d0348377b4c9216c1601481234ff60791206b2c1797e7d51f
-  data.tar.gz: ba23fdd140689fa467671a1f28024f38ea6b5c9e1ddd911c4980cbb91d57faf7a5ad2ce028aaffcead4f75d06ac4d5e30b0728525a3dac074ac366b8318f2797
+  metadata.gz: 1071f39a1ee582dad14d580bf3086330d83fefaf3942d48abb8d01402b13888a9f83867cb0795c6ed57a150c793d0acf76c0245cfef7cdfda2e330e7afb8674e
+  data.tar.gz: 6abbecd3c7e32610c66b336f1d2e304376a42e73789538f4ea93b18bad96f67f6b4949363fb0831d19ad6bfa6dfa69de2269fca6ce27cfe7652808792b6fcef3

data/lib/event_tracer/datadog_logger.rb

@@ -67,7 +67,7 @@ module EventTracer
     end
 
     def build_tags(args)
-      @default_tags.merge(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/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.5.1'.freeze
+  VERSION = '0.6.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: event_tracer
 version: !ruby/object:Gem::Version
-  version: 0.5.1
+  version: 0.6.0
 platform: ruby
 authors:
 - melvrickgoh
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-05-18 00:00:00.000000000 Z
+date: 2022-05-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: concurrent-ruby
@@ -102,6 +102,7 @@ files:
 - 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: