stackify-ruby-apm 1.11.1 → 1.12.3

data/lib/stackify_apm/spies/sinatra_activerecord/sqlite_adapter.rb

@@ -2,6 +2,8 @@
 
 # Spies for active record when sinatra framework is used and active_record is being extended. (mysql adapter)
 
+require 'stackify_apm/helper/database_helper'
+
 module StackifyRubyAPM
   # @api private
   module Spies
@@ -21,15 +23,11 @@ module StackifyRubyAPM
               exec_query_without_apm(sql, name, binds)
             end
 
-            ctx = Span::Context.new(
-              CATEGORY: 'Database',
-              SUBCATEGORY: 'Execute',
-              COMPONENT_CATEGORY: 'DB Query',
-              COMPONENT_DETAIL: 'Execute SQL Query',
-              SQL: sql,
-              PROVIDER: 'generic'
-            )
+            payload = {sql: sql, binds: binds}
+            statement = query_variables(payload)
+            check_prepared_stmt(statement, payload)
 
+            ctx = Span::Context.new(statement)
             result = exec_query_without_apm(sql, name, binds)
 
             StackifyRubyAPM.span name, TYPE, context: ctx do
@@ -37,6 +35,18 @@ module StackifyRubyAPM
             end
           end
           # rubocop:enable Lint/UnusedMethodArgument
+
+          def query_variables(payload)
+            props = get_common_db_properties
+            props[:SQL] = payload[:sql]
+            props
+          end
+
+          def check_prepared_stmt(statement, payload)
+            if StackifyRubyAPM.agent.config.prefix_enabled
+              check_prepared_stmt_by_placeholder(payload[:sql].include?('?'), statement, payload)
+            end
+          end
         end
       end
     end

data/lib/stackify_apm/spies/stackify_logger.rb

@@ -14,15 +14,21 @@ module StackifyRubyAPM
           def log_message_task(level, msg, call_trace)
             return log_message_task_without_apm(level, msg, call_trace) unless StackifyRubyAPM.current_transaction
 
-            trans_id = StackifyRubyAPM.current_transaction.id
-            log_uuid = SecureRandom.uuid
+            begin
+              trans_id = StackifyRubyAPM.current_transaction.id
+              log_uuid = SecureRandom.uuid
 
-            # build span context
-            ctx = Span::Context.new(
-              CATEGORY: 'Stackify',
-              SUBCATEGORY: 'Log',
-              ID: log_uuid # matching the MsgObject id
-            )
+              # build span context
+              ctx = Span::Context.new(
+                CATEGORY: 'Stackify',
+                SUBCATEGORY: 'Log',
+                ID: log_uuid # matching the MsgObject id
+              )
+            rescue Exception => e
+              StackifyRubyAPM.agent.error "[StackifyLoggerSpy] Error: creating span context."
+              StackifyRubyAPM.agent.error "[StackifyLoggerSpy] #{e.inspect}"
+              return log_message_task_without_apm(level, msg, call_trace)
+            end
 
             # create log span
             StackifyRubyAPM.span(msg, 'stackify.log', context: ctx) do
@@ -34,15 +40,21 @@ module StackifyRubyAPM
           def log_exception_task(level, ex)
             return log_exception_task_without_apm(level, ex) unless StackifyRubyAPM.current_transaction
 
-            trans_id = StackifyRubyAPM.agent.current_transaction.id
-            log_uuid = SecureRandom.uuid
+            begin
+              trans_id = StackifyRubyAPM.agent.current_transaction.id
+              log_uuid = SecureRandom.uuid
 
-            # build span context
-            ctx = Span::Context.new(
-              CATEGORY: 'Stackify',
-              SUBCATEGORY: 'Log',
-              ID: log_uuid # matching the MsgObject id
-            )
+              # build span context
+              ctx = Span::Context.new(
+                CATEGORY: 'Stackify',
+                SUBCATEGORY: 'Log',
+                ID: log_uuid # matching the MsgObject id
+              )
+            rescue Exception => e
+              StackifyRubyAPM.agent.error "[StackifyLoggerSpy] Error: creating span context."
+              StackifyRubyAPM.agent.error "[StackifyLoggerSpy] #{e.inspect}"
+              return log_exception_task_without_apm(level, ex)
+            end
 
             # create log span
             StackifyRubyAPM.span(ex.message, 'stackify.log', context: ctx) do

data/lib/stackify_apm/spies/yell.rb

@@ -0,0 +1,64 @@
+# frozen_string_literal: true
+
+# Monkey patch for the Yell::Logger class for logging log messages.
+#
+module StackifyRubyAPM
+  # @api private
+  module Spies
+    # @api private
+    class YellSpy
+      SEVERITIES = ['DEBUG', 'INFO', 'WARN', 'ERROR', 'FATAL', 'UNKNOWN'].freeze
+
+      def install
+        Yell::Logger.module_eval do
+          alias_method 'add_without_apm', 'add'
+
+          # Log messages
+          def add(options, *messages, &block)
+            return add_without_apm(options, *messages, &block) unless StackifyRubyAPM.current_transaction
+
+            begin
+              name = 'yell;'
+              type = 'ext.yell'
+              log_message = []
+              exception = []
+
+              messages.each { |msg|
+                case msg
+                when ::String
+                  log_message.push(msg)
+                when ::Exception
+                  log_message.push(msg.message)
+                  exception.push("(#{ msg.class })\n#{ msg.backtrace.join("\n") if msg.backtrace }")
+                else
+                  log_message.push(msg.inspect)
+                end
+              }
+
+              ctx = Span::Context.new(
+                CATEGORY: 'Log',
+                SUBCATEGORY: 'Yell',
+                LEVEL: SEVERITIES[options.severity],
+                MESSAGE: log_message.to_s
+              )
+
+              if exception.length > 0
+                ctx.EXCEPTION = exception.to_s
+              end
+            rescue Exception => e
+              StackifyRubyAPM.agent.error "[YellSpy] Error: creating span context."
+              StackifyRubyAPM.agent.error "[YellSpy] #{e.inspect}"
+              return add_without_apm(options, *messages, &block)
+            end
+
+            StackifyRubyAPM.span name, type, context: ctx do
+              add_without_apm(options, *messages, &block)
+            end
+          end
+        end
+      end
+    end
+
+    register 'Yell', 'yell', YellSpy.new
+  end
+end

data/lib/stackify_apm/util.rb

@@ -25,6 +25,16 @@ module StackifyRubyAPM
         "UNKNOWN #{host_os}"
       end
     end
+
+    # Push the element to existing array of object with incremented index(key/value pair)
+    # We get the prepared statement values ['J.K. Rowling', 'Harry Potter', ...] and restructure it.
+    # Example structured: [{'1': 'J.K. Rowling'},{'2': 'Harry Potter'}, ..., {n: 'other data'}]
+    def self.pushToAryIndex(ary, idx, val)
+      obj = {}
+      i = idx + 1
+      obj[i] = val.to_s
+      ary.push(obj)
+    end
   end
 end
 require 'stackify_apm/util/inspector'

data/lib/stackify_apm/version.rb

@@ -2,5 +2,5 @@
 
 # Sets the version of the APM
 module StackifyRubyAPM
-  VERSION = '1.11.1'.freeze
+  VERSION = '1.12.3'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stackify-ruby-apm
 version: !ruby/object:Gem::Version
-  version: 1.11.1
+  version: 1.12.3
 platform: ruby
 authors:
 - Stackify
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-06-22 00:00:00.000000000 Z
+date: 2020-07-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -433,7 +433,9 @@ files:
 - lib/stackify_apm/agent.rb
 - lib/stackify_apm/config.rb
 - lib/stackify_apm/context.rb
+- lib/stackify_apm/context/prefix.rb
 - lib/stackify_apm/context/request.rb
+- lib/stackify_apm/context/request/headers.rb
 - lib/stackify_apm/context/request/socket.rb
 - lib/stackify_apm/context/request/url.rb
 - lib/stackify_apm/context/response.rb
@@ -475,6 +477,9 @@ files:
 - lib/stackify_apm/spies/httparty.rb
 - lib/stackify_apm/spies/httpclient.rb
 - lib/stackify_apm/spies/httprb.rb
+- lib/stackify_apm/spies/log4r.rb
+- lib/stackify_apm/spies/logger.rb
+- lib/stackify_apm/spies/logging.rb
 - lib/stackify_apm/spies/mongo.rb
 - lib/stackify_apm/spies/net_http.rb
 - lib/stackify_apm/spies/redis.rb
@@ -486,6 +491,7 @@ files:
 - lib/stackify_apm/spies/sinatra_activerecord/sqlite_adapter.rb
 - lib/stackify_apm/spies/stackify_logger.rb
 - lib/stackify_apm/spies/tilt.rb
+- lib/stackify_apm/spies/yell.rb
 - lib/stackify_apm/stacktrace.rb
 - lib/stackify_apm/stacktrace/frame.rb
 - lib/stackify_apm/stacktrace_builder.rb