event_tracer 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c5207c9165b5c41811f3128fdf9d22c1dd21f805131aab1d2dfa2861bc55a43b
-  data.tar.gz: 3cda4e4b5dcb3c994797e9cae5ef3d31253e49d09a5d7527b49f107fc430e1df
+  metadata.gz: '002520584662d6b31ce2f05310a70c4fb8af37c3c865f6bdf2b8ebc44bf16654'
+  data.tar.gz: daa3fc169b3daa2771adb7d832872c055b723dbe25d28fff4ba24c618f2b54a2
 SHA512:
-  metadata.gz: 6333e12e55d047a9d4b6f5a6cb8af0781218a5be884b06a1b96fa3e994a73901f72d14e47269929df96284c0a8bb5ffc8f7aa6677413d2952a87b1b54b9f81c4
-  data.tar.gz: 8a27bbcf11291c1ca89c39944b784fb44661b00496f10412da4efbe933c7a014bac7a7fa186d55580e0fd2e766b1fcbe18565c13475091d1a01e6c1d0369c653
+  metadata.gz: '038e91c1d03b17070b97c3c170d73992d4050b9cee39f3529fc712a2cfda324d8afdfb10013f6a6838176966f94d6dd17b5f4851a62af6bdc0e6c8ea2b2e95b1'
+  data.tar.gz: fdeb28420163369840b82b0a56e6bd83a638fe4c5e7069f280c6807d8f9ad00b9044c642263fc89391628c110af8e527596c2e86c2b759a50691c3d13a7390c5

data/lib/event_tracer/buffer.rb

@@ -0,0 +1,65 @@
+require 'concurrent'
+
+module EventTracer
+  # This is an implementation of buffer storage. We use Concurrent::Array underneath
+  # to ensure thread-safe behavior. Data is stored until certain size / interval
+  # before flushing.
+  #
+  # Caveats: We should only store non-important data like logs in this buffer
+  # because if a process is killed, the data in this buffer is lost.
+  class Buffer
+    # Buffer can store maximum 10 items.
+    # Bigger size requires more memory to store, so choose a reasonable number
+    DEFAULT_BUFFER_SIZE = 10
+    # An item can live in buffer for at least 10s between each `Buffer#add` if the buffer is not full
+    # If there are larger interval between the calls, it can live longer.
+    DEFAULT_FLUSH_INTERVAL = 10
+
+    def initialize(
+      buffer_size: DEFAULT_BUFFER_SIZE,
+      flush_interval: DEFAULT_FLUSH_INTERVAL
+    )
+      @buffer_size = buffer_size
+      @flush_interval = flush_interval
+      @buffer = Concurrent::Array.new
+    end
+
+    # Add an item to buffer
+    #
+    # @param item: data to be added to buffer
+    # @return true if the item can be added, otherwise false
+    def add(item)
+      if add_item?
+        buffer.push({ item: item, created_at: Time.now })
+        true
+      else
+        false
+      end
+    end
+
+    # Remove all existing items from buffer
+    #
+    # @return all items in buffer
+    def flush
+      data = []
+
+      data << buffer.shift[:item] until buffer.empty?
+
+      data
+    end
+
+    # This method is only used to facilitate testing
+    def size
+      buffer.size
+    end
+
+    private
+
+      attr_reader :buffer_size, :flush_interval, :buffer
+
+      def add_item?
+        buffer.size < buffer_size &&
+          (buffer.empty? || buffer.first[:created_at] > Time.now - flush_interval)
+      end
+  end
+end

data/lib/event_tracer/config.rb

@@ -0,0 +1,10 @@
+require 'dry-configurable'
+
+module EventTracer
+  class Config
+    extend Dry::Configurable
+
+    setting :app_name, default: 'app_name'
+    setting :dynamo_db_table_name, default: 'logs'
+  end
+end

data/lib/event_tracer/dynamo_db_client.rb

@@ -0,0 +1,7 @@
+module EventTracer
+  class DynamoDBClient
+    def self.call
+      Aws::DynamoDB::Client.new
+    end
+  end
+end

data/lib/event_tracer/dynamo_db_default_processor.rb

@@ -0,0 +1,13 @@
+module EventTracer
+  class DynamoDBDefaultProcessor
+    def call(log_type, action:, message:, args:)
+      args.merge(
+        timestamp: Time.now.utc.iso8601(6),
+        action: action,
+        message: message,
+        log_type: log_type,
+        app: EventTracer::Config.config.app_name
+      )
+    end
+  end
+end

data/lib/event_tracer/dynamo_db_log_worker.rb

@@ -0,0 +1,68 @@
+# frozen_string_literal: true
+
+require_relative 'dynamo_db_client'
+
+begin
+  require 'sidekiq'
+  require 'aws-sdk-dynamodb'
+rescue LoadError => e
+  puts "Please add the missing gem into your app Gemfile: #{e.message}"
+  raise
+end
+
+module EventTracer
+  class DynamoDBLogWorker
+    include ::Sidekiq::Worker
+
+    sidekiq_options retry: 1, queue: 'low'
+
+    # See https://docs.aws.amazon.com/sdk-for-ruby/v3/api/Aws/DynamoDB/Client.html#batch_write_item-instance_method
+    MAX_DYNAMO_DB_ITEM_PER_REQUEST = 25
+
+    def perform(items)
+      wrap(items).each_slice(MAX_DYNAMO_DB_ITEM_PER_REQUEST) do |batch|
+        data = batch.map do |item|
+          { put_request: { item: clean_empty_values(item) } }
+        end
+
+        EventTracer::DynamoDBClient.call.batch_write_item(
+          request_items: { EventTracer::Config.config.dynamo_db_table_name => data }
+        )
+
+      rescue Aws::DynamoDB::Errors::ServiceError => e
+        EventTracer.error(
+          loggers: %i(base),
+          action: 'DynamoDBLogWorker',
+          app: EventTracer::Config.config.app_name,
+          error: e.class.name,
+          message: e.message
+        )
+      end
+    end
+
+    private
+
+      def wrap(items)
+        # NOTE: This allows us to handle either buffered or unbuffered payloads
+        if items.is_a?(Hash)
+          [items]
+        else
+          Array(items)
+        end
+      end
+
+      # dynamo can't serialise empty strings/ non-zero numerics
+      def clean_empty_values(data)
+        data.delete_if do |_key, value|
+          case value
+          when Hash
+            clean_empty_values(value)
+            false
+          when String then value.empty?
+          else false
+          end
+        end
+      end
+
+  end
+end

data/lib/event_tracer/dynamo_db_logger.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require 'time'
+require_relative 'dynamo_db_client'
+
+module EventTracer
+  class DynamoDBLogger
+    def initialize(buffer: Buffer.new(buffer_size: 0), log_processor: EventTracer::DynamoDBDefaultProcessor.new)
+      @buffer = buffer
+      @log_processor = log_processor
+    end
+
+    EventTracer::LOG_TYPES.each do |log_type|
+      define_method log_type do |**args|
+        save_message log_type, **args
+      end
+    end
+
+    private
+
+      attr_reader :buffer, :log_processor
+
+      def save_message(log_type, action:, message:, **args)
+        payload = log_processor.call(log_type, action: action, message: message, args: args)
+
+        unless buffer.add(payload)
+          all_payloads = buffer.flush + [payload]
+          DynamoDBLogWorker.perform_async(all_payloads)
+        end
+
+        LogResult.new(true)
+      end
+
+  end
+end

data/lib/event_tracer/version.rb

@@ -1,3 +1,3 @@
 module EventTracer
-  VERSION = '0.3.1'.freeze
+  VERSION = '0.3.2'.freeze
 end

metadata

@@ -1,15 +1,43 @@
 --- !ruby/object:Gem::Specification
 name: event_tracer
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - melvrickgoh
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-08-19 00:00:00.000000000 Z
+date: 2021-10-11 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dry-configurable
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -53,7 +81,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '3.0'
 description: 'Thin wrapper for formatted logging/ metric services to be used as a
-  single service. External service(s) supported: Appsignal'
+  single service. External service(s) supported: Appsignal, Datadog, DynamoDB'
 email:
 - melvrickgoh@hotmail.com
 executables: []
@@ -64,7 +92,13 @@ files:
 - lib/event_tracer/appsignal_logger.rb
 - lib/event_tracer/base_logger.rb
 - lib/event_tracer/basic_decorator.rb
+- lib/event_tracer/buffer.rb
+- lib/event_tracer/config.rb
 - lib/event_tracer/datadog_logger.rb
+- lib/event_tracer/dynamo_db_client.rb
+- lib/event_tracer/dynamo_db_default_processor.rb
+- lib/event_tracer/dynamo_db_log_worker.rb
+- lib/event_tracer/dynamo_db_logger.rb
 - lib/event_tracer/log_result.rb
 - lib/event_tracer/version.rb
 homepage: https://github.com/melvrickgoh/event_tracer
@@ -87,7 +121,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.2.28
 signing_key:
 specification_version: 4
 summary: Thin wrapper for formatted logging/ metric services to be used as a single