File: logstash_formatter.rb

package info (click to toggle)
ruby-httparty 0.24.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 964 kB
  • sloc: ruby: 7,521; xml: 425; sh: 35; makefile: 14
file content (62 lines) | stat: -rw-r--r-- 1,449 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# frozen_string_literal: true

module HTTParty
  module Logger
    class LogstashFormatter #:nodoc:
      TAG_NAME = HTTParty.name

      attr_accessor :level, :logger

      def initialize(logger, level)
        @logger = logger
        @level  = level.to_sym
      end

      def format(request, response)
        @request = request
        @response = response

        logger.public_send level, logstash_message
      end

      private

      attr_reader :request, :response

      def logstash_message
        require 'json'
        {
          '@timestamp' => current_time,
          '@version' => 1,
          'content_length' => content_length || '-',
          'http_method' => http_method,
          'message' => message,
          'path' => path,
          'response_code' => response.code,
          'severity' => level,
          'tags' => [TAG_NAME],
        }.to_json
      end

      def message
        "[#{TAG_NAME}] #{response.code} \"#{http_method} #{path}\" #{content_length || '-'} "
      end

      def current_time
        Time.now.strftime('%Y-%m-%d %H:%M:%S %z')
      end

      def http_method
        @http_method ||= request.http_method.name.split('::').last.upcase
      end

      def path
        @path ||= request.path.to_s
      end

      def content_length
        @content_length ||= response.respond_to?(:headers) ? response.headers['Content-Length'] : response['Content-Length']
      end
    end
  end
end