semlogr/semlogr

View on GitHub
lib/semlogr/config/sink_config.rb

Summary

Maintainability
A
0 mins
Test Coverage
# frozen_string_literal: true

require 'semlogr/component_registry'
require 'semlogr/sinks/enriching'
require 'semlogr/sinks/filtering'
require 'semlogr/log_severity'

module Semlogr
  module Config
    class SinkConfig
      def initialize
        @min_severity = LogSeverity::DEBUG
        @enrichers = []
        @filters = []
      end

      def emit_at(severity)
        @min_severity = LogSeverity.create(severity)
      end

      def filter(filter)
        @filters << filter
      end

      def enrich_with(enricher, *params)
        enricher = resolve_enricher(enricher, params)
        @enrichers << enricher
      end

      def create_sink(sink, *params)
        if sink.is_a?(Symbol)
          sink = ComponentRegistry.resolve(:sink, sink, *params)
        end

        if @min_severity > LogSeverity::DEBUG
          severity_filter = ->(log_event) { log_event.severity < @min_severity }
          @filters.unshift(severity_filter)
        end

        sink = Sinks::Filtering.new(@filters, sink) if @filters.any?
        sink = Sinks::Enriching.new(@enrichers, sink) if @enrichers.any?
        sink
      end

      private

      def resolve_enricher(enricher, params)
        return enricher unless enricher.is_a?(Symbol)
        ComponentRegistry.resolve(:enricher, enricher, *params)
      end
    end
  end
end