judoscale-shoryuken 1.7.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: dafaf3727137764032e23a8b29915b7cb259bc03b95569393cb6c2f6d70cd09a
+  data.tar.gz: 3069eb9d26c0323867ff5e8e0db22a63ec80179854896696e61573232fae3c6f
+SHA512:
+  metadata.gz: 239d278c606e72725f523e54fe1a79a48932524644a847e3ccf7d4f412a3ab24c06736ec4d780f87a034640ca1533d1a833854a8e67f86b876b9223a1ef49c66
+  data.tar.gz: 69ab83bfaabfae905f5fde3c1b1e960875b3c842746ee2f2dcce7bdd9f222b2f362a6ad8c2db9388c5bc45966181a6920337296e64482ecc23ef666837c9e3a5

data/Gemfile

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+gemspec name: "judoscale-shoryuken"
+
+gem "judoscale-ruby", path: "../judoscale-ruby"
+gem "shoryuken", "~> 6.2"
+gem "aws-sdk-sqs" # shoryuken integration with AWS SDK v3+
+gem "rexml" # aws-sdk-core requires an XML parser.
+gem "minitest"
+gem "rake"

data/Rakefile

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+require "rake/testtask"
+
+Rake::TestTask.new(:test) do |t|
+  t.libs = %w[lib test]
+  t.pattern = "test/**/*_test.rb"
+end
+
+task default: :test

data/judoscale-shoryuken.gemspec

@@ -0,0 +1,30 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "judoscale/shoryuken/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "judoscale-shoryuken"
+  spec.version = Judoscale::Shoryuken::VERSION
+  spec.authors = ["Adam McCrea", "Carlos Antonio da Silva", "Jon Sullivan"]
+  spec.email = ["hello@judoscale.com"]
+
+  spec.summary = "Autoscaling for Shoryuken workers."
+  spec.homepage = "https://judoscale.com"
+  spec.license = "MIT"
+
+  spec.metadata = {
+    "homepage_uri" => "https://judoscale.com",
+    "bug_tracker_uri" => "https://github.com/judoscale/judoscale-ruby/issues",
+    "documentation_uri" => "https://judoscale.com/docs",
+    "changelog_uri" => "https://github.com/judoscale/judoscale-ruby/blob/main/CHANGELOG.md",
+    "source_code_uri" => "https://github.com/judoscale/judoscale-ruby"
+  }
+
+  spec.files = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.require_paths = ["lib"]
+
+  spec.required_ruby_version = ">= 2.6.0"
+
+  spec.add_dependency "judoscale-ruby", Judoscale::Shoryuken::VERSION
+  spec.add_dependency "shoryuken", ">= 6.0"
+end

data/lib/judoscale/shoryuken/metrics_collector.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require "judoscale/job_metrics_collector"
+require "judoscale/metric"
+
+module Judoscale
+  module Shoryuken
+    class MetricsCollector < Judoscale::JobMetricsCollector
+      SQS_QUEUE_DEPTH_ATTRIBUTE = "ApproximateNumberOfMessages"
+
+      def self.adapter_config
+        Judoscale::Config.instance.shoryuken
+      end
+
+      def collect
+        metrics = []
+        self.queues |= ::Shoryuken.ungrouped_queues
+
+        queues.each do |queue_name|
+          queue = ::Shoryuken::Client.queues(queue_name)
+
+          # Reach out to SQS client directly to fetch the queue attribute we need to report.
+          # Shoryuken has a private `queue_attributes` API to fetch all attributes, this call mimics that.
+          sqs_queue_attributes = ::Shoryuken.sqs_client
+            .get_queue_attributes(queue_url: queue.url, attribute_names: [SQS_QUEUE_DEPTH_ATTRIBUTE])
+          depth = sqs_queue_attributes.attributes[SQS_QUEUE_DEPTH_ATTRIBUTE]
+
+          metrics.push Metric.new(:qd, depth, Time.now, queue_name)
+        end
+
+        log_collection(metrics)
+        metrics
+      end
+    end
+  end
+end

data/lib/judoscale/shoryuken/version.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+module Judoscale
+  module Shoryuken
+    VERSION = "1.7.0"
+  end
+end

data/lib/judoscale/shoryuken.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+require "judoscale-ruby"
+require "judoscale/config"
+require "judoscale/shoryuken/version"
+require "judoscale/shoryuken/metrics_collector"
+require "shoryuken"
+
+Judoscale.add_adapter :"judoscale-shoryuken",
+  {
+    adapter_version: Judoscale::Shoryuken::VERSION,
+    framework_version: ::Shoryuken::VERSION
+  },
+  metrics_collector: Judoscale::Shoryuken::MetricsCollector,
+  expose_config: Judoscale::Config::JobAdapterConfig.new(:shoryuken, support_busy_jobs: false)

data/lib/judoscale-shoryuken.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require "judoscale/shoryuken"

data/lib/rails-autoscale-shoryuken.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require "judoscale/shoryuken"

data/rails-autoscale-shoryuken.gemspec

@@ -0,0 +1,30 @@
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "judoscale/shoryuken/version"
+
+Gem::Specification.new do |spec|
+  spec.name = "rails-autoscale-shoryuken"
+  spec.version = Judoscale::Shoryuken::VERSION
+  spec.authors = ["Adam McCrea", "Carlos Antonio da Silva", "Jon Sullivan"]
+  spec.email = ["hello@judoscale.com"]
+
+  spec.summary = "Autoscaling for Shoryuken workers."
+  spec.homepage = "https://judoscale.com"
+  spec.license = "MIT"
+
+  spec.metadata = {
+    "homepage_uri" => "https://judoscale.com",
+    "bug_tracker_uri" => "https://github.com/judoscale/judoscale-ruby/issues",
+    "documentation_uri" => "https://judoscale.com/docs",
+    "changelog_uri" => "https://github.com/judoscale/judoscale-ruby/blob/main/CHANGELOG.md",
+    "source_code_uri" => "https://github.com/judoscale/judoscale-ruby"
+  }
+
+  spec.files = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.require_paths = ["lib"]
+
+  spec.required_ruby_version = ">= 2.6.0"
+
+  spec.add_dependency "rails-autoscale-core", Judoscale::Shoryuken::VERSION
+  spec.add_dependency "shoryuken", ">= 6.0"
+end

metadata

@@ -0,0 +1,87 @@
+--- !ruby/object:Gem::Specification
+name: judoscale-shoryuken
+version: !ruby/object:Gem::Version
+  version: 1.7.0
+platform: ruby
+authors:
+- Adam McCrea
+- Carlos Antonio da Silva
+- Jon Sullivan
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2024-05-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: judoscale-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.7.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.7.0
+- !ruby/object:Gem::Dependency
+  name: shoryuken
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '6.0'
+description: 
+email:
+- hello@judoscale.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Rakefile
+- judoscale-shoryuken.gemspec
+- lib/judoscale-shoryuken.rb
+- lib/judoscale/shoryuken.rb
+- lib/judoscale/shoryuken/metrics_collector.rb
+- lib/judoscale/shoryuken/version.rb
+- lib/rails-autoscale-shoryuken.rb
+- rails-autoscale-shoryuken.gemspec
+homepage: https://judoscale.com
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://judoscale.com
+  bug_tracker_uri: https://github.com/judoscale/judoscale-ruby/issues
+  documentation_uri: https://judoscale.com/docs
+  changelog_uri: https://github.com/judoscale/judoscale-ruby/blob/main/CHANGELOG.md
+  source_code_uri: https://github.com/judoscale/judoscale-ruby
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.6.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.4.10
+signing_key: 
+specification_version: 4
+summary: Autoscaling for Shoryuken workers.
+test_files: []