datadog

Author: Git-push-01

Gemfile

@@ -1,5 +1,7 @@
 source 'https://rubygems.org'
 git_source(:github) { |repo| "https://github.com/#{repo}.git" }
+gem 'lograge'
+gem 'lograge-sql'
 
 ruby '2.3.3'
 gem 'knock', github: 'psantos10/knock', branch: 'master'

Gemfile.lock

@@ -80,6 +80,14 @@ GEM
       rb-fsevent (~> 0.9, >= 0.9.4)
       rb-inotify (~> 0.9, >= 0.9.7)
       ruby_dep (~> 1.2)
+    lograge (0.11.2)
+      actionpack (>= 4)
+      activesupport (>= 4)
+      railties (>= 4)
+      request_store (~> 1.0)
+    lograge-sql (1.0.0)
+      activerecord (>= 4, < 7.0)
+      lograge (~> 0.4)
     loofah (2.2.3)
       crass (~> 1.0.2)
       nokogiri (>= 1.5.9)
@@ -134,6 +142,8 @@ GEM
     rb-inotify (0.10.0)
       ffi (~> 1.0)
     rb-readline (0.5.5)
+    request_store (1.4.1)
+      rack (>= 1.4)
     ruby_dep (1.5.0)
     spring (2.0.2)
       activesupport (>= 4.2)
@@ -166,6 +176,8 @@ DEPENDENCIES
   jwt
   knock!
   listen (>= 3.0.5, < 3.2)
+  lograge
+  lograge-sql
   pg (>= 0.18, < 2.0)
   pry
   puma (~> 3.11)

app/controllers/application_controller.rb

@@ -8,7 +8,7 @@ def index
     end
 
     def get_current_user
-      jwt_token = request.headers['HTTP_AUTHORIZATION'] 
+      jwt_token = request.headers['HTTP_AUTHORIZATION']
 
       if jwt_token
         user_info = Auth.decode(jwt_token)
@@ -17,4 +17,15 @@ def get_current_user
 
       user
     end
+    def append_info_to_payload(payload)
+      super
+     case
+        when payload[:status] == 200
+          payload[:level] = "INFO"
+        when payload[:status] == 302
+          payload[:level] = "WARN"
+        else
+          payload[:level] = "ERROR"
+        end
+end
   end

config/initializers/lograge.rb

@@ -0,0 +1,12 @@
+require 'lograge/sql/extension'
+
+Rails.application.configure do
+    # Lograge config
+    config.lograge.enabled = true
+    config.lograge.formatter = Lograge::Formatters::Json.new
+    config.colorize_logging = false
+
+    config.lograge.custom_options = lambda do |event|
+      { :params => event.payload[:params] }
+  end
+end