rails-autoscale-core 1.5.1 → 1.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aea9f97efeaa39ee58923722220094d45ee721e925064f9f344e442cf839e10b
-  data.tar.gz: 2eb632696c60647be6af0023e3202007d1ef045f26a07ecc7d84b33db108efc6
+  metadata.gz: ecb527ab4d6f7a97247bd1aa47424414ea57ef62eaea688e24674820d9c20e8b
+  data.tar.gz: d1fa82cb48feb2fe4dd8136647b51d83a439632a19cc14f2cc6114d2e3677a54
 SHA512:
-  metadata.gz: 80b2d814748d4ec90fce53a11f98a22e6ba4c17135798188742ce0b1b8a5a8ed5f4231ced13df03ce19fde892423e965ada294e287696b9d903860c178f562f4
-  data.tar.gz: d12cde36b76a75f737b1a1aab3e2959ab318e93c461680532905868de7d4a31503e401ae8544d989a2f2a5e3bdef3bd33854711220405ddd5815244ecc51bc52
+  metadata.gz: 4f4403334de767802741291af1799bfa1f0f911e002d89a6f8d9e82465da3e1bef724aa75196fd09e25342fa43e66e516852463e8c8e4063eab526d597605f61
+  data.tar.gz: d3f82c8c7387ff7627c30b70571c72bca3f4c8d000933e5917a595b6cb6a7a20827f52c1fe4ad9b55d343907085710e59ac21426bfea9a241ecd63b2899e5df1

data/lib/judoscale/config.rb

@@ -65,7 +65,7 @@ module Judoscale
     end
 
     attr_accessor :api_base_url, :report_interval_seconds,
-      :max_request_size_bytes, :logger, :log_tag, :current_runtime_container, :allow_rake_tasks
+      :max_request_size_bytes, :logger, :log_tag, :current_runtime_container
     attr_reader :log_level
 
     def initialize
@@ -77,7 +77,6 @@ module Judoscale
       @log_tag = "Judoscale"
       @max_request_size_bytes = 100_000 # ignore request payloads over 100k since they skew the queue times
       @report_interval_seconds = 10
-      @allow_rake_tasks = []
 
       self.log_level = ENV["JUDOSCALE_LOG_LEVEL"] || ENV["RAILS_AUTOSCALE_LOG_LEVEL"]
       @logger = ::Logger.new($stdout)

data/lib/judoscale/job_metrics_collector/active_record_helper.rb

@@ -12,6 +12,13 @@ module Judoscale
         sql
       end
 
+      # This will respect a multiple-database setup, unlike the `table_exists?` method.
+      def self.table_exists_for_model?(model)
+        model.connection.schema_cache.data_source_exists?(model.table_name)
+      rescue ActiveRecord::NoDatabaseError
+        false
+      end
+
       def self.table_exists?(table_name)
         ::ActiveRecord::Base.connection.table_exists?(table_name)
       rescue ActiveRecord::NoDatabaseError

data/lib/judoscale/logger.rb

@@ -6,7 +6,11 @@ require "logger"
 module Judoscale
   module Logger
     def logger
-      @logger ||= LoggerProxy.new(Config.instance.logger, Config.instance.log_level)
+      if @logger && @logger.log_level == Config.instance.log_level
+        @logger
+      else
+        @logger = LoggerProxy.new(Config.instance.logger, Config.instance.log_level)
+      end
     end
   end
 

data/lib/judoscale/metric.rb

@@ -5,7 +5,7 @@ module Judoscale
     # No queue_name is assumed to be a web request metric
     # Metrics: qt = queue time (default), qd = queue depth, busy
     def initialize(identifier, value, time, queue_name = nil)
-      super identifier, value.to_i, time.utc, queue_name
+      super(identifier, value.to_i, time.utc, queue_name)
     end
   end
 end

data/lib/judoscale/metrics_collector.rb

@@ -3,22 +3,11 @@
 module Judoscale
   class MetricsCollector
     def self.collect?(config)
-      in_rake_task = defined?(::Rake) && Rake.respond_to?(:application) && Rake.application.top_level_tasks.any?
-
-      !in_rake_task || in_whitelisted_rake_tasks?(config.allow_rake_tasks)
+      true
     end
 
     def collect
       []
     end
-
-    def self.in_whitelisted_rake_tasks?(allowed_rake_tasks)
-      # Get the tasks that were invoked from the command line.
-      tasks = Rake.application.top_level_tasks
-
-      allowed_rake_tasks.any? do |task_regex|
-        tasks.any? { |task| task =~ task_regex }
-      end
-    end
   end
 end

data/lib/judoscale/reporter.rb

@@ -19,25 +19,24 @@ module Judoscale
     def start!(config, adapters)
       @pid = Process.pid
 
-      if !config.api_base_url
+      if config.api_base_url.nil? || config.api_base_url.strip.empty?
         logger.debug "Set api_base_url to enable metrics reporting"
         return
       end
 
       enabled_adapters, skipped_adapters = adapters.partition { |adapter|
-        # judoscale-ruby adapter does not have a metrics collector
-        adapter.metrics_collector.nil? || adapter.metrics_collector.collect?(config)
+        adapter.metrics_collector&.collect?(config)
       }
       metrics_collectors_classes = enabled_adapters.map(&:metrics_collector)
-      metrics_collectors_classes.compact!
+      adapters_msg = enabled_adapters.map(&:identifier).concat(
+        skipped_adapters.map { |adapter| "#{adapter.identifier}[skipped]" }
+      ).join(", ")
 
       if metrics_collectors_classes.empty?
-        adapters_msg = skipped_adapters.map(&:identifier).join(", ")
         logger.debug "No metrics need to be collected (adapters: #{adapters_msg})"
         return
       end
 
-      adapters_msg = enabled_adapters.map(&:identifier).join(", ")
       logger.info "Reporter starting, will report every ~#{config.report_interval_seconds} seconds (adapters: #{adapters_msg})"
 
       metrics_collectors = metrics_collectors_classes.map(&:new)
@@ -73,17 +72,26 @@ module Judoscale
       @_thread&.terminate
       @_thread = nil
       @pid = nil
+      @reported = false
     end
 
     private
 
     def report(config, metrics)
+      # Make sure we report at least once, even if there are no metrics,
+      # so Judoscale knows the adapter is installed and running.
+      if @reported && metrics.empty?
+        logger.debug "No metrics to report - skipping"
+        return
+      end
+
       report = Report.new(Judoscale.adapters, config, metrics)
       logger.info "Reporting #{report.metrics.size} metrics"
       result = AdapterApi.new(config).report_metrics(report.as_json)
 
       case result
       when AdapterApi::SuccessResponse
+        @reported = true
         logger.debug "Reported successfully"
       when AdapterApi::FailureResponse
         logger.error "Reporter failed: #{result.failure_message}"

data/lib/judoscale/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Judoscale
-  VERSION = "1.5.1"
+  VERSION = "1.5.3"
 end

data/lib/judoscale-ruby.rb

@@ -20,7 +20,7 @@ module Judoscale
     attr_reader :adapters
   end
 
-  Adapter = Struct.new(:identifier, :adapter_info, :metrics_collector) do
+  class Adapter < Struct.new(:identifier, :adapter_info, :metrics_collector)
     def as_json
       {identifier => adapter_info}
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails-autoscale-core
 version: !ruby/object:Gem::Version
-  version: 1.5.1
+  version: 1.5.3
 platform: ruby
 authors:
 - Adam McCrea
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-08-09 00:00:00.000000000 Z
+date: 2024-01-25 00:00:00.000000000 Z
 dependencies: []
 description: 
 email:
@@ -20,7 +20,6 @@ extensions: []
 extra_rdoc_files: []
 files:
 - Gemfile
-- Gemfile.lock
 - Rakefile
 - judoscale-ruby.gemspec
 - lib/judoscale-ruby.rb

data/Gemfile.lock

@@ -1,48 +0,0 @@
-PATH
-  remote: .
-  specs:
-    judoscale-ruby (1.5.1)
-
-GEM
-  remote: https://rubygems.org/
-  specs:
-    addressable (2.8.1)
-      public_suffix (>= 2.0.2, < 6.0)
-    crack (0.4.5)
-      rexml
-    debug (1.7.1)
-      irb (>= 1.5.0)
-      reline (>= 0.3.1)
-    hashdiff (1.0.1)
-    io-console (0.6.0)
-    irb (1.6.2)
-      reline (>= 0.3.0)
-    minitest (5.17.0)
-    minitest-stub-const (0.6)
-    public_suffix (5.0.1)
-    rake (13.0.6)
-    reline (0.3.2)
-      io-console (~> 0.5)
-    rexml (3.2.5)
-    webmock (3.18.1)
-      addressable (>= 2.8.0)
-      crack (>= 0.3.2)
-      hashdiff (>= 0.4.0, < 2.0.0)
-
-PLATFORMS
-  arm64-darwin-20
-  arm64-darwin-21
-  arm64-darwin-22
-  x86_64-darwin-21
-  x86_64-linux
-
-DEPENDENCIES
-  debug
-  judoscale-ruby!
-  minitest
-  minitest-stub-const
-  rake (>= 12.3.3)
-  webmock
-
-BUNDLED WITH
-   2.3.9