Contents

# frozen_string_literal: true

require 'logger'
require 'time'
require 'json'
require 'twiglet/formatter'
require_relative '../hash_extensions'

module Twiglet
  class Logger < ::Logger
    Hash.include HashExtensions

    def initialize(
      service_name,
      default_properties: {},
      now: -> { Time.now.utc },
      output: $stdout
    )
      @service_name = service_name
      @now = now
      @output = output

      raise 'Service name is mandatory' \
        unless service_name.is_a?(String) && !service_name.strip.empty?

      formatter = Twiglet::Formatter.new(service_name, default_properties: default_properties, now: now)
      super(output, formatter: formatter)
    end

    def error(message = {}, error = nil, &block)
      if error
        error_fields = {
          'error': {
            'message': error.message
          }
        }
        add_stack_trace(error_fields, error)
        message.is_a?(Hash) ? message.merge!(error_fields) : error_fields.merge!(message: message)
      end

      super(message, &block)
    end

    def with(default_properties)
      Logger.new(@service_name,
                 default_properties: default_properties,
                 now: @now,
                 output: @output)
    end

    alias_method :warning, :warn
    alias_method :critical, :fatal

    private

    def add_stack_trace(hash_to_add_to, error)
      hash_to_add_to[:error][:stack_trace] = error.backtrace.join("\n") if error.backtrace
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
twiglet-2.2.0	lib/twiglet/logger.rb