hoss-agent 1.0.9

data/lib/hoss/error_builder.rb

@@ -0,0 +1,90 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  class ErrorBuilder
+    def initialize(agent)
+      @agent = agent
+    end
+
+    def build_exception(exception, context: nil, handled: true)
+      error = Error.new context: context || Context.new
+      error.exception =
+        Error::Exception.from_exception(exception, handled: handled)
+
+      Util.reverse_merge!(error.context.labels, @agent.config.default_labels)
+
+      if exception.backtrace
+        add_stacktrace error, :exception, exception.backtrace
+      end
+
+      add_current_transaction_fields error, Hoss.current_transaction
+
+      error
+    end
+
+    def build_log(message, context: nil, backtrace: nil, **attrs)
+      error = Error.new context: context || Context.new
+      error.log = Error::Log.new(message, **attrs)
+
+      if backtrace
+        add_stacktrace error, :log, backtrace
+      end
+
+      add_current_transaction_fields error, Hoss.current_transaction
+
+      error
+    end
+
+    private
+
+    def add_stacktrace(error, kind, backtrace)
+      stacktrace =
+        @agent.stacktrace_builder.build(backtrace, type: :error)
+      return unless stacktrace
+
+      case kind
+      when :exception
+        error.exception.stacktrace = stacktrace
+      when :log
+        error.log.stacktrace = stacktrace
+      end
+
+      error.culprit = stacktrace.frames.first&.function
+    end
+
+    def add_current_transaction_fields(error, transaction)
+      return unless transaction
+
+      error.transaction_id = transaction.id
+      error.transaction = {
+        sampled: transaction.sampled?,
+        type: transaction.type
+      }
+      error.trace_id = transaction.trace_id
+      error.parent_id = Hoss.current_span&.id || transaction.id
+
+      return unless transaction.context
+
+      Util.reverse_merge!(error.context.labels, transaction.context.labels)
+      Util.reverse_merge!(error.context.custom, transaction.context.custom)
+    end
+  end
+end

data/lib/hoss/event.rb

@@ -0,0 +1,131 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+# require 'hoss/span/context'
+
+module Hoss
+  # @api private
+  class Event
+    extend Forwardable
+    include ChildDurations::Methods
+
+    DEFAULT_TYPE = 'custom'
+
+    class Request
+      extend Forwardable
+      def initialize
+        @headers = {}
+      end
+      attr_accessor(
+        :method,
+        :url,
+        :headers,
+        :body,
+        :received_at
+      )
+    end
+
+    class Response
+      extend Forwardable
+      def initialize
+        @headers = {}
+      end
+      attr_accessor(
+        :status_code,
+        :headers,
+        :body,
+        :received_at
+      )
+    end
+
+    class Error
+      extend Forwardable
+      ConnectionTimeout = 'CONNECTION_TIMEOUT'
+      RequestTimeout = 'REQUEST_TIMEOUT'
+      ConnectionError = 'CONNECTION_ERROR'
+      def initialize(type)
+        @type = type
+      end
+      attr_accessor(
+        :type,
+        :received_at
+      )
+    end
+
+    # rubocop:disable Metrics/ParameterLists
+    def initialize
+      @id = SecureRandom.uuid
+      @request = Request.new
+      # @response = Response.new
+      @filtered = false
+      @retries = 0
+    end
+    # rubocop:enable Metrics/ParameterLists
+
+    attr_accessor(
+      :request,
+      :response,
+      :error,
+      :filtered,
+      :retries,
+    )
+
+    attr_reader(
+      :id
+    )
+
+    # life cycle
+
+    def start(clock_start = Util.monotonic_micros)
+      @timestamp = Util.micros
+      @clock_start = clock_start
+      self
+    end
+
+    def stop(clock_end = Util.monotonic_micros)
+      @duration ||= (clock_end - @clock_start)
+      @self_time = @duration - child_durations.duration
+      self
+    end
+
+    def done(clock_end: Util.monotonic_micros)
+      stop clock_end
+      self
+    end
+
+    def stopped?
+      !!duration
+    end
+
+    def started?
+      !!timestamp
+    end
+
+    def running?
+      started? && !stopped?
+    end
+
+    # relations
+
+    def inspect
+      "<Hoss::Event id:#{id.inspect}" \
+        '>'
+    end
+  end
+end

data/lib/hoss/instrumenter.rb

@@ -0,0 +1,107 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+require 'hoss/child_durations'
+require 'hoss/event'
+
+module Hoss
+  # @api private
+  class Instrumenter
+    EVENT_KEY = :__elastic_instrumenter_events_key
+
+    include Logging
+
+    # @api private
+    class Current
+      def initialize
+        self.events = []
+      end
+
+      def events
+        Thread.current[EVENT_KEY] ||= []
+      end
+
+      def events=(events)
+        Thread.current[EVENT_KEY] ||= []
+        Thread.current[EVENT_KEY] = events
+      end
+    end
+
+    def initialize(config, metrics:, stacktrace_builder:, &enqueue)
+      @config = config
+      @stacktrace_builder = stacktrace_builder
+      @enqueue = enqueue
+      @metrics = metrics
+
+      @current = Current.new
+    end
+
+    attr_reader :stacktrace_builder, :enqueue
+
+    def start
+      debug 'Starting instrumenter'
+      # We call register! on @subscriber in case the
+      # instrumenter was stopped and started again
+      @subscriber&.register!
+    end
+
+    def stop
+      debug 'Stopping instrumenter'
+
+      current_events.pop until current_events.empty?
+
+      @subscriber&.unregister!
+    end
+
+    def handle_forking!
+      stop
+      start
+    end
+
+    def subscriber=(subscriber)
+      debug 'Registering subscriber'
+      @subscriber = subscriber
+      @subscriber.register!
+    end
+
+
+    # events
+
+    def current_events
+      @current.events
+    end
+
+    def current_event
+      current_events.last
+    end
+
+    def start_event
+      event = Event.new
+      current_events.push event
+      event.start
+    end
+
+    def end_event
+      return unless (event = current_events.pop)
+      event.done
+      enqueue.call event
+      event
+    end
+  end
+end

data/lib/hoss/internal_error.rb

@@ -0,0 +1,23 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  class InternalError < StandardError; end
+  class ExistingTransactionError < InternalError; end
+end

data/lib/hoss/logging.rb

@@ -0,0 +1,70 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  module Logging
+    PREFIX = '[Hoss] '
+
+    LEVELS = {
+      debug: Logger::DEBUG,
+      info: Logger::INFO,
+      warn: Logger::WARN,
+      error: Logger::ERROR,
+      fatal: Logger::FATAL
+    }.freeze
+
+    def debug(msg, *args, &block)
+      log(:debug, msg, *args, &block)
+    end
+
+    def info(msg, *args, &block)
+      log(:info, msg, *args, &block)
+    end
+
+    def warn(msg, *args, &block)
+      log(:warn, msg, *args, &block)
+    end
+
+    def error(msg, *args, &block)
+      log(:error, msg, *args, &block)
+    end
+
+    def fatal(msg, *args, &block)
+      log(:fatal, msg, *args, &block)
+    end
+
+    private
+
+    def log(lvl, msg, *args)
+      return unless (logger = @config&.logger)
+      return unless LEVELS[lvl] >= (@config&.log_level || 0)
+
+      formatted_msg = prepend_prefix(format(msg.to_s, *args))
+
+      return logger.send(lvl, formatted_msg) unless block_given?
+
+      logger.send(lvl, "#{formatted_msg}\n#{yield}")
+    end
+
+    def prepend_prefix(str)
+      "#{PREFIX}#{str}"
+    end
+  end
+end

data/lib/hoss/metadata.rb

@@ -0,0 +1,36 @@
+# Licensed to Elasticsearch B.V. under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch B.V. licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# frozen_string_literal: true
+
+module Hoss
+  # @api private
+  class Metadata
+    def initialize(config)
+      @service = ServiceInfo.new(config)
+      @process = ProcessInfo.new(config)
+      @system = SystemInfo.new(config)
+      @labels = config.global_labels
+    end
+
+    attr_reader :service, :process, :system, :labels
+  end
+end
+
+require 'hoss/metadata/service_info'
+require 'hoss/metadata/system_info'
+require 'hoss/metadata/process_info'