neetodeploy-autoscale 0.5.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c0eba17ecdf99b87711ea1607f3497bba260a773aebf6c064f225916f1bce0ab
-  data.tar.gz: 5d38a45b75d33207f55f539955e9dddc27294cdbc402c2f181086f7f343ae4d5
+  metadata.gz: ae89d71a8dc745af2e4af2c94a94c16497a2067ec5fbc31e1ad351f5f84bd333
+  data.tar.gz: 69e44e72990b27b8ea1e6c9398f74af6b7c73d8dfb5dc0bc3c0995821da8c16a
 SHA512:
-  metadata.gz: fdad9d224da3024944b8f92d5157b9aeee1bbabac81d9d757efb0811cc265720381f4ae88db36fc985381ff5f179d30a4236ca2a5346d0e653041ee5464b2eea
-  data.tar.gz: b98a32aab20efb3228968ed211523d947cce385c275e83159c108aa4252f6817b633077b65f2699df7af77a5a801ed445c2dda7a734f9f93601f8cf0268d94eb
+  metadata.gz: fc14913a507e277fb7e76a5917053e7a7a2cc4039a3180e5cb5625b3807f23c72c244e0a60774a24c26d35ffae3834c4bff4d3a2fc264e19006b49308b741112
+  data.tar.gz: af10749f4ab0e928fe3564c9ee86fb4933bace261e7d3a8e59a4032e596a557f88506dca517c7852790b758ef7b808d8abd6ae2d06083cc7deb72ada008b5885

data/lib/neetodeploy/autoscale/middleware.rb

@@ -12,7 +12,7 @@ module Neetodeploy
     def call(env)
       url = URI.parse("https://nd-queue-time-exporter.neetodeployapp.com/metrics")
 
-      query_params = URI.encode_www_form(app_name: ENV.to_h["NEETODEPLOY_APP_NAME"], queue_time: queue_time(env))
+      query_params = URI.encode_www_form(app_name: ENV.to_h["NEETODEPLOY_APP_NAME"], process_type: "web", queue_time: queue_time(env))
 
       url.query = query_params
 

data/lib/neetodeploy/autoscale/sidekiq_middleware.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "time"
+
+module Neetodeploy
+  class SidekiqMiddleware
+    def call(worker, job, queue)
+      url = URI.parse("https://nd-queue-time-exporter.neetodeployapp.com/metrics")
+
+      query_params = URI.encode_www_form(app_name: ENV.to_h["NEETODEPLOY_APP_NAME"], process_type: "worker", queue_time: queue_time(worker, job, queue))
+
+      url.query = query_params
+
+      post = Net::HTTP::Post.new(url)
+      post["AuthToken"] = "K0An3O3MSyEEMTCnRd1IHgGjdGQkzy"
+
+      Net::HTTP.start(url.host, url.port, use_ssl: true) do |http|
+        http.request(post)
+      end
+
+      yield
+    end
+
+    private
+
+    def queue_time(worker, job, queue)
+      (Time.now - Time.at(job["created_at"])) * 1000
+    end
+  end
+end

data/lib/neetodeploy/autoscale/version.rb

@@ -2,6 +2,6 @@
 
 module Neetodeploy
   module Autoscale
-    VERSION = "0.5.0"
+    VERSION = "0.7.0"
   end
 end

data/lib/neetodeploy-autoscale.rb

@@ -5,10 +5,29 @@ require_relative "neetodeploy/autoscale/version"
 module Neetodeploy
   module Autoscale
     class Error < StandardError; end
+
+    class << self
+      def setup_middleware
+        require "neetodeploy/autoscale/middleware"
+        Neetodeploy::Autoscale.setup_rails if defined? Rails
+        Neetodeploy::Autoscale.setup_sidekiq if defined?(Sidekiq)
+      end
+
+      def setup_rails
+        require "neetodeploy/autoscale/railtie"
+      end
+
+      def setup_sidekiq
+        require "neetodeploy/autoscale/sidekiq_middleware"
+
+        Sidekiq.configure_server do |config|
+          config.server_middleware do |chain|
+            chain.add Neetodeploy::SidekiqMiddleware
+          end
+        end
+      end
+    end
   end
 end
 
-if ENV.to_h["NEETODEPLOY_APP_NAME"].present?
-  require "neetodeploy/autoscale/middleware"
-  require "neetodeploy/autoscale/railtie" if defined? Rails
-end
+ENV.to_h["NEETODEPLOY_APP_NAME"].present? && Neetodeploy::Autoscale.setup_middleware

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neetodeploy-autoscale
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Sreeram Venkitesh
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-05-07 00:00:00.000000000 Z
+date: 2024-05-13 00:00:00.000000000 Z
 dependencies: []
 description: For automatically scaling your Rails application based on network metrics
 email:
@@ -25,6 +25,7 @@ files:
 - lib/neetodeploy-autoscale.rb
 - lib/neetodeploy/autoscale/middleware.rb
 - lib/neetodeploy/autoscale/railtie.rb
+- lib/neetodeploy/autoscale/sidekiq_middleware.rb
 - lib/neetodeploy/autoscale/version.rb
 homepage: https://neetodeploy.com
 licenses: []
@@ -47,7 +48,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.7
+rubygems_version: 3.3.26
 signing_key:
 specification_version: 4
 summary: neetoDeploy autoscaler gem