skylight 4.3.1 → 5.0.0.beta

data/lib/skylight/util/clock.rb

@@ -0,0 +1,56 @@
+module Skylight
+  module Util
+    # A more precise clock
+    class Clock
+      def self.use_native!
+        class_eval do
+          def tick
+            native_hrtime
+          end
+        end
+      end
+
+      # rubocop:disable Lint/DuplicateMethods
+      def tick
+        now = Time.now
+        now.to_i * 1_000_000_000 + now.usec * 1_000
+      end
+      # rubocop:enable Lint/DuplicateMethods
+
+      # TODO: rename to secs
+      def absolute_secs
+        Time.now.to_i
+      end
+
+      # TODO: remove
+      def nanos
+        tick
+      end
+
+      # TODO: remove
+      def secs
+        nanos / 1_000_000_000
+      end
+
+      class << self
+        def absolute_secs
+          default.absolute_secs
+        end
+
+        def nanos
+          default.nanos
+        end
+
+        def secs
+          default.secs
+        end
+
+        def default
+          @default ||= Clock.new
+        end
+
+        attr_writer :default
+      end
+    end
+  end
+end

data/lib/skylight/util/component.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+
 require "uri"
 
 module Skylight
@@ -6,7 +7,7 @@ module Skylight
     class Component
       attr_accessor :environment, :name
 
-      NAME_FORMAT = /\A[a-zA-Z0-9_-]+\z/
+      NAME_FORMAT = /\A[a-zA-Z0-9_-]+\z/.freeze
       DEFAULT_NAME = "web"
       WORKER_NAME = "worker"
       DEFAULT_ENVIRONMENT = "production"
@@ -16,6 +17,7 @@ module Skylight
         @name        = resolve_name(name, force_worker)
 
         raise ArgumentError, "environment can't be blank" if @environment.empty?
+
         validate_string!(@environment, "environment")
         validate_string!(@name, "name")
       end
@@ -40,7 +42,7 @@ module Skylight
       def as_json(*)
         {
           component: name,
-          env: environment
+          env:       environment
         }
       end
 
@@ -64,6 +66,7 @@ module Skylight
 
         def validate_string!(string, kind)
           return true if string =~ NAME_FORMAT
+
           raise ArgumentError, "#{kind} can only contain lowercase letters, numbers, and dashes"
         end
     end

data/lib/skylight/util/deploy.rb

@@ -1,5 +1,5 @@
 require "json"
-require "skylight/core/util/logging"
+require "skylight/util/logging"
 
 module Skylight
   module Util
@@ -35,8 +35,8 @@ module Skylight
 
         def to_query_hash
           hash = {
-            timestamp:   timestamp,
-            deploy_id:   id.to_s[0..100] # Keep this sane
+            timestamp: timestamp,
+            deploy_id: id.to_s[0..100] # Keep this sane
           }
           hash[:git_sha] = git_sha.to_s[0..40] if git_sha # A valid SHA will never exceed 40
           hash[:description] = description[0..255] if description # Avoid massive descriptions
@@ -45,7 +45,7 @@ module Skylight
       end
 
       class DefaultDeploy < EmptyDeploy
-        include Core::Util::Logging
+        include Util::Logging
 
         def initialize(*)
           super

data/lib/skylight/util/gzip.rb

@@ -0,0 +1,20 @@
+require "zlib"
+
+module Skylight
+  module Util
+    # Provides Gzip compressing support
+    module Gzip
+      # Compress a string with Gzip
+      #
+      # @param str [String] uncompressed string
+      # @return [String] compressed string
+      def self.compress(str)
+        output = StringIO.new
+        gz = Zlib::GzipWriter.new(output)
+        gz.write(str)
+        gz.close
+        output.string
+      end
+    end
+  end
+end

data/lib/skylight/util/http.rb

@@ -3,7 +3,7 @@ require "json"
 require "openssl"
 require "net/http"
 require "net/https"
-require "skylight/core/util/gzip"
+require "skylight/util/gzip"
 require "skylight/util/ssl"
 
 module Skylight
@@ -19,7 +19,7 @@ module Skylight
       DEFLATE          = "deflate".freeze
       GZIP             = "gzip".freeze
 
-      include Core::Util::Logging
+      include Logging
 
       attr_accessor :authentication
       attr_reader :host, :port, :config
@@ -118,7 +118,7 @@ module Skylight
 
           yield res
         ensure
-          client.finish if client
+          client&.finish
         end
 
         def execute(req, body = nil)
@@ -189,13 +189,7 @@ module Skylight
           end
 
           def get(key)
-            return nil unless body.is_a?(Hash)
-
-            res = body
-            key.split(".").each do |part|
-              return unless (res = res[part])
-            end
-            res
+            body.dig(*key.split(".")) if body.is_a?(Hash)
           end
 
           def respond_to_missing?(name, include_all = false)

data/lib/skylight/util/instrumenter_method.rb

@@ -0,0 +1,26 @@
+module Skylight
+  module Util
+    module InstrumenterMethod
+      def instrumenter_method(name, block: false)
+        if block
+          module_eval <<-RUBY, __FILE__, __LINE__ + 1
+            def #{name}(*args)
+              unless instrumenter
+                return yield if block_given?
+                return
+              end
+
+              instrumenter.#{name}(*args) { yield }
+            end
+          RUBY
+        else
+          module_eval <<-RUBY, __FILE__, __LINE__ + 1
+            def #{name}(*args)
+              instrumenter&.#{name}(*args)
+            end
+          RUBY
+        end
+      end
+    end
+  end
+end

data/lib/skylight/util/logging.rb

@@ -0,0 +1,138 @@
+require "logger"
+
+module Skylight
+  module Util
+    # Log both to the specified logger and STDOUT
+    class AlertLogger
+      def initialize(logger)
+        @logger = logger
+      end
+
+      def write(*args)
+        STDERR.write(*args)
+
+        # Try to avoid writing to STDOUT/STDERR twice
+        logger_logdev = @logger.instance_variable_get(:@logdev)
+        logger_out = logger_logdev&.respond_to?(:dev) ? logger_logdev.dev : nil
+        if logger_out != STDOUT && logger_out != STDERR
+          @logger.<<(*args)
+        end
+      end
+
+      def close; end
+    end
+
+    module Logging
+      def log_context
+        {}
+      end
+
+      def trace?
+        !!ENV[-"SKYLIGHT_ENABLE_TRACE_LOGS"]
+      end
+
+      def raise_on_error?
+        !!ENV[-"SKYLIGHT_RAISE_ON_ERROR"]
+      end
+
+      # Logs if tracing
+      #
+      # @param (see #debug)
+      #
+      # See {trace?}.
+      def trace(msg, *args)
+        return unless trace?
+
+        log :debug, msg, *args
+      end
+
+      # Evaluates and logs the result of the block if tracing
+      #
+      # @yield block to be evaluted
+      # @yieldreturn arguments for {#debug}
+      #
+      # See {trace?}.
+      def t
+        return unless trace?
+
+        log :debug, yield
+      end
+
+      # @param msg (see #log)
+      # @param args (see #log)
+      def debug(msg, *args)
+        log :debug, msg, *args
+      end
+
+      # @param msg (see #log)
+      # @param args (see #log)
+      def info(msg, *args)
+        log :info, msg, *args
+      end
+
+      # @param msg (see #log)
+      # @param args (see #log)
+      def warn(msg, *args)
+        log :warn, msg, *args
+      end
+
+      # @param msg (see #log)
+      # @param args (see #log)
+      def error(msg, *args)
+        log :error, msg, *args
+        raise format(msg, *args) if raise_on_error?
+      end
+
+      alias log_trace trace
+      alias log_debug debug
+      alias log_info  info
+      alias log_warn  warn
+      alias log_error error
+
+      # Alias for `Kernel#sprintf`
+      # @return [String]
+      def fmt(*args)
+        sprintf(*args)
+      end
+
+      def config_for_logging
+        if respond_to?(:config)
+          config
+        elsif is_a?(Skylight::Config)
+          self
+        end
+      end
+
+      # @param level [String,Symbol] the method on `logger` to use for logging
+      # @param msg [String] the message to log
+      # @param args [Array] values for `Kernel#sprintf` on `msg`
+      def log(level, msg, *args)
+        c = config_for_logging
+        logger = c ? c.logger : nil
+
+        msg = log_context.map { |(k, v)| "#{k}=#{v}; " }.join << msg
+
+        if logger
+          if logger.respond_to?(level)
+            if !args.empty?
+              logger.send level, format(msg, *args)
+            else
+              logger.send level, msg
+            end
+            return # rubocop:disable Style/RedundantReturn
+          else
+            Kernel.warn "Invalid logger"
+          end
+        # Fallback to stderr for warn and error levels
+        elsif %i[warn error].include?(level)
+          $stderr.puts format("[SKYLIGHT] #{msg}", *args)
+        end
+      rescue Exception => e
+        if trace?
+          puts "[ERROR] #{e.message}"
+          puts e.backtrace
+        end
+      end
+    end
+  end
+end

data/lib/skylight/util/lru_cache.rb

@@ -0,0 +1,42 @@
+# Based on code by Sam Saffron: https://stackoverflow.com/a/16161783/181916
+module Skylight
+  module Util
+    class LruCache
+      def initialize(max_size)
+        @max_size = max_size
+        @data = {}
+      end
+
+      def max_size=(size)
+        raise ArgumentError, :max_size if @max_size < 1
+
+        @max_size = size
+        while @data.size > @max_size
+          @data.shift
+        end
+      end
+
+      # Individual hash operations here are atomic in MRI.
+      def fetch(key)
+        found = true
+        value = @data.delete(key) { found = false }
+
+        if !found && block_given?
+          value = yield
+        end
+
+        @data[key] = value if value
+
+        if !found && value && @data.length > @max_size
+          @data.shift
+        end
+
+        value
+      end
+
+      def clear
+        @data.clear
+      end
+    end
+  end
+end

data/lib/skylight/vendor/cli/thor/rake_compat.rb

@@ -11,7 +11,7 @@ class Thor
   #   class Default < Thor
   #     include Thor::RakeCompat
   #
-  #     RSpec::Core::RakeTask.new(:spec) do |t|
+  #     RSpec::RakeTask.new(:spec) do |t|
   #       t.spec_opts = ['--options', "./.rspec"]
   #       t.spec_files = FileList['spec/**/*_spec.rb']
   #     end

data/lib/skylight/version.rb

@@ -1,3 +1,7 @@
 module Skylight
-  VERSION = "4.3.1".freeze
+  # pre-release versions should be given here as "5.0.0-alpha"
+  # for compatibility with semver when it is parsed by the rust agent.
+  # This string will be transformed in the gemspec to "5.0.0.alpha"
+  # to conform with rubygems.
+  VERSION = "5.0.0-beta".freeze
 end

data/lib/skylight/vm/gc.rb

@@ -0,0 +1,68 @@
+module Skylight
+  # @api private
+  module VM
+    if defined?(JRUBY_VERSION)
+
+      # This doesn't quite work as we would like it. I believe that the GC
+      # statistics includes time that is not stop-the-world, this does not
+      # necessarily take time away from the application.
+      #
+      # require 'java'
+      # class GC
+      #   def initialize
+      #     @factory = Java::JavaLangManagement::ManagementFactory
+      #   end
+      #
+      #   def enable
+      #   end
+      #
+      #   def total_time
+      #     res = 0.0
+      #
+      #     @factory.garbage_collector_mx_beans.each do |mx|
+      #       res += (mx.collection_time.to_f / 1_000.0)
+      #     end
+      #
+      #     res
+      #   end
+      # end
+
+    elsif defined?(::GC::Profiler)
+
+      class GC
+        def initialize
+          @total = 0
+        end
+
+        def enable
+          ::GC::Profiler.enable
+        end
+
+        def total_time
+          # Reported in seconds
+          run = (::GC::Profiler.total_time * 1_000_000).to_i
+
+          if run > 0
+            ::GC::Profiler.clear
+          end
+
+          @total += run
+        end
+      end
+
+    end
+
+    # Fallback
+    unless defined?(VM::GC)
+
+      class GC
+        def enable; end
+
+        def total_time
+          0
+        end
+      end
+
+    end
+  end
+end