spartan_apm 0.0.0.rc1

data/app/views/styles.css

@@ -0,0 +1,332 @@
+* {
+  box-sizing: border-box;
+}
+
+body {
+  font-family: sans-serif;
+  font-size: 14px;
+  line-height: 36px ;
+  text-align: left;
+  color: #212529;
+  background-color: #ffffff;
+  margin: 0 0 1rem 0;
+  padding: 0;
+}
+
+header {
+  background-color: #666;
+  color: white;
+  padding: 1rem;
+  margin-bottom: 1rem;
+}
+
+header h1 {
+  font-weight: bold;
+  font-size: 1.2rem;
+  margin: 0;
+}
+
+header .dates {
+  float: right;
+  max-width: 50%;
+  margin-left: 1rem;
+}
+
+header .help-btn {
+  float: right;
+  margin-right: 1rem;
+}
+
+header .help-btn a {
+  color: white;
+  text-decoration: none;
+}
+header .help-btn a:visited {
+  color: white;
+}
+header .help-btn a:hover {
+  color: #CCC;
+}
+header .help-btn a:active {
+  color: #CCC;
+}
+
+h2 {
+  font-weight: normal;
+  font-size: 1.2rem;
+  margin: 1rem 0;
+}
+
+.text-center {
+  text-align: center;
+}
+
+.text-right {
+  text-align: right;
+}
+
+.text-normal {
+  font-weight: normal;
+}
+
+.container {
+  padding-left: 1rem;
+  padding-right: 1rem;
+  margin-left: auto;
+  margin-right: auto;
+}
+
+.push-up {
+  margin-top: 1rem;
+}
+
+.nowrap {
+  white-space: nowrap;
+}
+
+.flex-container {
+  display: flex;
+  gap: 2rem;
+  align-items: flex-start;
+  align-content: space-between;
+}
+
+.flex-item {
+  width: 50%;
+  flex-grow: 1;
+}
+
+.show-xs {
+  display: none;
+}
+
+@media only screen and (max-width: 800px) {
+  .flex-container {
+    display: block;
+  }
+
+  .flex-item {
+    width: 100%;
+  }
+
+  .hide-xs {
+    display: none;
+  }
+
+  .show-xs {
+    display: inherit;
+  }
+
+  #start-time {
+    display: block;
+    margin-top: -1rem;
+  }
+
+  #end-time {
+    display: block;
+    margin-top: -0.5rem;
+    margin-bottom: 1rem;
+  }
+}
+
+.btn {
+  height: 2.4rem;
+	box-shadow: inset 0px 1px 0px 0px #ffffff;
+	background: linear-gradient(to bottom, #f9f9f9 5%, #e9e9e9 100%);
+	background-color: #f9f9f9;
+	border-radius: 6px;
+	border: 1px solid #dcdcdc;
+	display: inline-block;
+	color: #666666;
+	font-family: Arial;
+	font-size: 1rem;
+	font-weight: 200;
+	padding: 0.47rem 1rem;
+	text-decoration: none;
+	text-shadow: 0px 1px 0px #ffffff;
+  vertical-align: middle;
+}
+.btn:hover {
+	background:linear-gradient(to bottom, #e9e9e9 5%, #f9f9f9 100%);
+	background-color:#e9e9e9;
+}
+.btn:active {
+	position:relative;
+	top:1px;
+}
+.btn:disabled {
+	background:linear-gradient(to bottom, #f9f9f9 5%, #e9e9e9 100%);
+	background-color:#f9f9f9;
+  opacity: 0.65;
+  box-shadow: none;
+}
+
+.btn svg {
+  height: 1.2em;
+}
+
+.table {
+  width: 100%;
+  max-width: 100%;
+  margin-bottom: 1rem;
+  border-collapse: collapse;
+}
+
+.table thead th {
+  vertical-align: bottom;
+  border-bottom: 2px solid #dee2e6;
+}
+
+.table td, .table th {
+  padding: 0.75rem;
+  vertical-align: top;
+  border-top: 1px solid #dee2e6;
+}
+
+.table-striped tbody tr:nth-of-type(odd) {
+  background-color: rgba(0, 0, 0, .05);
+}
+
+.nice-select {
+  float: none;
+  display: inline-block;
+  border: 1px solid #ccc;
+  margin: 0 0 0.5rem 0;
+}
+
+.nice-select .nice-select-search {
+  border: 1px solid #ccc;
+}
+
+.form-control {
+  margin: 0 0 0.5rem 0;
+  padding: .375rem .5rem;
+  color: #495057;
+  background-color: #fff;
+  background-clip: padding-box;
+  border: 1px solid #ccc;
+  border-radius: 5px;
+  transition: border-color .15s ease-in-out,box-shadow .15s ease-in-out;
+  max-width: 100%;
+  line-height: 36px;
+  height: 38px;
+  font-size: 14px;
+  display: inline-block;
+  outline: none;
+}
+
+.form-control:disabled {
+  color: #999;
+}
+
+.form-control:focus, .nice-select:focus, .nice-select:active, .nice-select.open {
+  border-color: #333;
+}
+
+select {
+  display: none;
+}
+
+.hidden {
+  display: none;
+}
+
+.spinner {
+  width: 100%;
+  height: 64px;
+  background-repeat: no-repeat;
+  background-position: center;
+  background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' width='64px' height='64px' viewBox='0 0 128 128'%3E%3Cg%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%23000000'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(45, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(90, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(135, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(180, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(225, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(270, 64, 64)'/%3E%3Cpath d='M116.3 60.18a52.28 52.28 0 0 0-12.64-30.45l8.06-8.06a63.6 63.6 0 0 1 15.97 38.5h-11.4z' fill='%239c9c9c' transform='rotate(315, 64, 64)'/%3E%3CanimateTransform attributeName='transform' type='rotate' values='0 64 64;45 64 64;90 64 64;135 64 64;180 64 64;225 64 64;270 64 64;315 64 64' calcMode='discrete' dur='720ms' repeatCount='indefinite'%3E%3C/animateTransform%3E%3C/g%3E%3C/svg%3E");
+}
+
+.download-btn {
+  cursor: pointer;
+  width: 1.3rem;
+}
+
+.error-class-name {
+  font-size: 0.9rem;
+  overflow-wrap: break-word;
+}
+
+.error-count {
+  font-size: 0.9rem;
+}
+
+.error-backtrace {
+  font-family: monospace;
+  font-size: 0.9rem;
+}
+
+.backtrace-line {
+  text-indent: -2rem;
+  margin-left: 2rem;
+  line-height: 1.1rem;
+  margin-bottom: 0.2rem;
+  overflow-wrap: break-word;
+}
+
+#alert {
+  display: none;
+  text-align: center;
+  margin: 0 auto;
+  color: red;
+}
+
+#modal {
+  z-index: 1000000000000;
+  display: none;
+  padding-top: 5rem;
+  position: fixed;
+  left: 0;
+  top: 0;
+  width: 100%;
+  height: 100%;
+  overflow: auto;
+  background-color: rgba(0,0,0,0.4);
+}
+
+.modal-dialog {
+  margin: auto;
+  background-color: #fff;
+  position: relative;
+  outline: 0;
+  padding: 2em;
+  box-shadow: 3px 3px 7px 3px rgba(0, 0, 0, 0.6);
+  border-radius: 4px;
+  width: 80%;
+  height: 80%;
+}
+
+.modal-content {
+  height: 100%;
+  overflow: auto;
+  line-height: 1rem;
+}
+
+.modal-content p {
+  margin-top: 0;
+  margin-bottom: 0.75rem;
+}
+
+.modal-close {
+  display: block;
+  position: absolute;
+  top: 0;
+  right: 0;
+  border: 0;
+  padding: 1ex;
+  background-color: inherit;
+  font-size: 1.5rem;
+  font-weight: 500;
+}
+
+dd {
+  margin-inline-start: 0;
+  margin: 0 0 0.5rem 0;
+  font-weight: bold;
+}
+
+dt {
+  margin: 0 0 1rem 0;
+}

data/config.ru

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require "rack"
+
+require_relative "lib/spartan_apm"
+
+if SpartanAPM.apps.empty?
+  raise ArgumentError.new("Apps must be specified in comma delimited list in the SPARTAN_APM environment variable")
+else
+  puts "Using apps #{SpartanAPM.apps.join(', ')} in environments #{SpartanAPM.environments.join(', ')}"
+end
+
+# Make sure we don't log access tokens.
+class SecureLogger
+  ACCESS_TOKEN_PATTERN = /([?&]access_token=)[^&\s]+/
+
+  def write(message)
+    $stdout.write(message.gsub(ACCESS_TOKEN_PATTERN, '\1******'))
+  end
+end
+
+use Rack::CommonLogger, SecureLogger.new
+
+basic_auth_user = ENV.fetch("BASIC_AUTH_USER", "")
+basic_auth_password = ENV.fetch("BASIC_AUTH_PASSWORD", "")
+unless basic_auth_user.empty?
+  use Rack::Auth::Basic do |user, password|
+    user == basic_auth_user && password.to_s == basic_auth_password
+  end
+end
+
+if File.exist?(File.join(__dir__, "public", "plotly.js"))
+  use Rack::Static, urls: ["/plotly.js"], root: "public"
+end
+
+run SpartanAPM::Web::Router.new

data/lib/spartan_apm/engine.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  # Rails engine that will automatically instrument a Rails application
+  # for web requests. If Sidekiq is installed, it will be instrumented
+  # as well.
+  class Engine < Rails::Engine
+    initializer "spartan_apm" do |app|
+      ActiveSupport.on_load(:action_controller) do
+        ActionController::Base.prepend_around_action do |controller, action|
+          # rubocop:disable Lint/RescueException
+          begin
+            action.call
+          rescue Exception => e
+            SpartanAPM.capture_error(e)
+            raise
+          end
+          # rubocop:enable Lint/RescueException
+        end
+
+        if defined?(Sidekiq) && Sidekiq.server?
+          Sidekiq.configure_server do |config|
+            config.server_middleware do |chain|
+              chain.prepend SpartanAPM::Middleware::Sidekiq::StartMiddleware
+              chain.add SpartanAPM::Middleware::Sidekiq::EndMiddleware
+            end
+          end
+        end
+
+        app.config.middleware.insert(0, SpartanAPM::Middleware::Rack::StartMiddleware)
+        app.config.middleware.use(SpartanAPM::Middleware::Rack::EndMiddleware)
+      end
+    end
+
+    config.after_initialize do
+      SpartanAPM.configure do |config|
+        config.env = Rails.env.to_s unless config.env_set?
+        config.logger ||= Rails.logger
+        apps = ["web"]
+        apps << "sidekiq" if defined?(Sidekiq)
+        config.apps = apps if config.apps.empty?
+      end
+    end
+  end
+end

data/lib/spartan_apm/error_info.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  # Data structure for capturing information about errors.
+  class ErrorInfo
+    attr_reader :time, :class_name, :message, :backtrace
+    attr_accessor :count
+
+    def initialize(time, class_name, message, backtrace, count)
+      @time = time
+      @class_name = class_name
+      @message = message
+      @backtrace = backtrace
+      @count = count.to_i
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/active_record.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class ActiveRecord < Base
+      def initialize
+        @klass = ::ActiveRecord::Base.connection.class if defined?(::ActiveRecord::Base)
+        @name = :database
+        @methods = [:exec_query]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/base.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    # Base class for describing how to inject instrumentation code into another class.
+    # This class should be extended and the subclass should set the klass, name, and methods
+    # attributes in the constructor.
+    class Base
+      # The class that should be instrumented.
+      attr_accessor :klass
+
+      # The component name that metrics should be recorded as.
+      attr_accessor :name
+
+      # Flag indicating if metrics should be captured exclusively.
+      attr_accessor :exclusive
+
+      # List of instance methods to instrument.
+      attr_accessor :methods
+
+      # Inject instrumentation code into the specified class' methods.
+      def instrument!
+        raise ArgumentError.new("klass not specified") unless klass
+        raise ArgumentError.new("name not specified") unless name
+        Instrumentation.instrument!(klass, name, methods, exclusive: exclusive, module_name: "#{self.class.name.split("::").last}Instrumentation")
+      end
+
+      # Determine if the instrumentation definition is valid.
+      def valid?
+        return false if klass.nil? || name.nil?
+        all_methods = klass.public_instance_methods + klass.protected_instance_methods + klass.private_instance_methods
+        Array(methods).all? { |m| all_methods.include?(m) }
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/bunny.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Bunny < Base
+      def initialize
+        @klass = ::Bunny::Channel if defined?(::Bunny::Channel)
+        @name = :rabbitmq
+        @methods = [
+          :basic_get,
+          :basic_publish,
+          :basic_ack,
+          :basic_nack,
+          :basic_consume,
+          :basic_consume_with,
+          :basic_recover,
+          :basic_cancel,
+          :basic_qos,
+          :basic_reject
+        ]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/cassandra.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Cassandra < Base
+      def initialize
+        @klass = ::Cassandra::Session if defined?(::Cassandra::Session)
+        @name = :cassandra
+        @methods = [:execute, :prepare]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/curb.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Curb < Base
+      def initialize
+        @klass = ::Curl::Multi if defined?(::Curl::Multi)
+        @name = :http
+        @methods = [:perform]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/dalli.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Dalli < Base
+      def initialize
+        @klass = ::Dalli::Client if defined?(::Dalli::Client)
+        @name = :memcache
+        @methods = [:perform]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/elasticsearch.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Elasticsearch < Base
+      def initialize
+        if defined?(::Elastic::Transport::Client)
+          @klass = ::Elastic::Transport::Client
+        elsif defined?(::Elasticsearch::Transport::Client)
+          @klass = ::Elasticsearch::Transport::Client
+        end
+        @name = :elasticsearch
+        @methods = [:perform_request]
+        @exclusive = true
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/excon.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Excon < Base
+      def initialize
+        @klass = ::Excon::Connection if defined?(::Excon::Connection)
+        @name = :http
+        @methods = [:request]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/http.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class HTTP < Base
+      def initialize
+        @klass = ::HTTP::Client if defined?(::HTTP::Client)
+        @name = :http
+        @methods = [:perform]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/httpclient.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class HTTPClient < Base
+      def initialize
+        @klass = ::HTTPClient if defined?(::HTTPClient)
+        @name = :http
+        @methods = [:do_get_block]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/net_http.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class NetHTTP < Base
+      def initialize
+        @klass = ::Net::HTTP if defined?(::Net::HTTP)
+        @name = :http
+        @methods = [:request]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/redis.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Redis < Base
+      def initialize
+        @klass = ::Redis::Client if defined?(::Redis::Client)
+        @name = :redis
+        @methods = [:process]
+      end
+    end
+  end
+end

data/lib/spartan_apm/instrumentation/typhoeus.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module SpartanAPM
+  module Instrumentation
+    class Typhoeus < Base
+      def initialize
+        @klass = ::Typhoeus::Hydra if defined?(::Typhoeus::Hydra)
+        @name = :http
+        @methods = [:run]
+      end
+    end
+  end
+end