44 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			44 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
module Gitlab
 | 
						|
  module Lograge
 | 
						|
    module CustomOptions
 | 
						|
      include ::Gitlab::Logging::CloudflareHelper
 | 
						|
 | 
						|
      LIMITED_ARRAY_SENTINEL = { key: 'truncated', value: '...' }.freeze
 | 
						|
      IGNORE_PARAMS = Set.new(%w(controller action format)).freeze
 | 
						|
      KNOWN_PAYLOAD_PARAMS = [:remote_ip, :user_id, :username, :ua, :queue_duration_s,
 | 
						|
                              :etag_route, :request_urgency, :target_duration_s] + CLOUDFLARE_CUSTOM_HEADERS.values
 | 
						|
 | 
						|
      def self.call(event)
 | 
						|
        params = event
 | 
						|
          .payload[:params]
 | 
						|
          .each_with_object([]) { |(k, v), array| array << { key: k, value: v } unless IGNORE_PARAMS.include?(k) }
 | 
						|
        payload = {
 | 
						|
          time: Time.now.utc.iso8601(3),
 | 
						|
          params: Gitlab::Utils::LogLimitedArray.log_limited_array(params, sentinel: LIMITED_ARRAY_SENTINEL)
 | 
						|
        }
 | 
						|
 | 
						|
        payload.merge!(event.payload[:metadata]) if event.payload[:metadata]
 | 
						|
        optional_payload_params = event.payload.slice(*KNOWN_PAYLOAD_PARAMS).compact
 | 
						|
        payload.merge!(optional_payload_params)
 | 
						|
 | 
						|
        ::Gitlab::InstrumentationHelper.add_instrumentation_data(payload)
 | 
						|
 | 
						|
        payload[Labkit::Correlation::CorrelationId::LOG_KEY] = event.payload[Labkit::Correlation::CorrelationId::LOG_KEY] || Labkit::Correlation::CorrelationId.current_id
 | 
						|
 | 
						|
        # https://github.com/roidrage/lograge#logging-errors--exceptions
 | 
						|
        exception = event.payload[:exception_object]
 | 
						|
 | 
						|
        ::Gitlab::ExceptionLogFormatter.format!(exception, payload)
 | 
						|
 | 
						|
        if Feature.enabled?(:feature_flag_state_logs, type: :ops)
 | 
						|
          payload[:feature_flag_states] = Feature.logged_states.map { |key, state| "#{key}:#{state ? 1 : 0}" }
 | 
						|
        end
 | 
						|
 | 
						|
        payload
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |