Initial metrics PORO.

Author: metaskills

lib/lambdakiq.rb

@@ -13,6 +13,7 @@
 require 'lambdakiq/job'
 require 'lambdakiq/record'
 require 'lambdakiq/backoff'
+require 'lambdakiq/metrics'
 require 'rails/railtie'
 require 'lambdakiq/railtie'
 

lib/lambdakiq/metrics.rb

@@ -0,0 +1,73 @@
+module Lambdakiq
+  class Metrics
+
+    def initialize
+      @logger = ActiveJob::Base.logger
+      @namespace = Rails.application.class.name.split('::').first
+      @dimensions = Concurrent::Array.new
+      @metrics = Concurrent::Array.new
+      @properties = Concurrent::Hash.new
+    end
+
+    def metrics
+      yield(self)
+    ensure
+      flush
+    end
+
+    def flush
+      @logger.info(message) unless empty?
+    end
+
+    def benchmark
+      value = nil
+      seconds = Benchmark.realtime { value = yield }
+      milliseconds = (seconds * 1000).to_i
+      [value, milliseconds]
+    end
+
+    def put_dimension(name, value)
+      @dimensions << { name => value }
+      self
+    end
+
+    def put_metric(name, value, unit = nil)
+      @metrics << { 'Name' => name }.tap do |m|
+        m['Unit'] = unit if unit
+      end
+      set_property name, value
+    end
+
+    def set_property(name, value)
+      @properties[name] = value
+      self
+    end
+
+    def empty?
+      [@dimensions, @metrics, @properties].all?(&:empty?)
+    end
+
+    def message
+      {
+        '_aws' => {
+          'Timestamp' => timestamp,
+          'CloudWatchMetrics' => [
+            {
+              'Namespace' => @namespace,
+              'Dimensions' => [@dimensions.map(&:keys).flatten],
+              'Metrics' => @metrics
+            }
+          ]
+        }
+      }.tap do |m|
+        @dimensions.each { |dim| m.merge!(dim) }
+        m.merge!(@properties)
+      end
+    end
+
+    def timestamp
+      Time.now.strftime('%s%3N').to_i
+    end
+
+  end
+end

test/test_helper/client_helpers.rb

@@ -1,5 +1,10 @@
 module TestHelper
   module ClientHelpers
+    extend ActiveSupport::Concern
+
+    included do
+      let(:max_receive_count) { 8 }
+    end
 
     private
 
@@ -25,9 +30,5 @@ def api_requests
       client.api_requests
     end
 
-    def max_receive_count
-      8
-    end
-
   end
 end