cutoff 0.4.1 → 0.4.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5ac27ce016591276e55afa82f64c08de370a969cb6552cc0d5ae69093d3a9e0c
-  data.tar.gz: 31e989979ae04de967a22e8d2457d6da654862603f380dd7521e415c9220148b
+  metadata.gz: d00ff1f831beec4fed582b5faa9ed42c350c13b32217bdcab0b850445ba9c358
+  data.tar.gz: 77d466db54a822240d2d27cfdb120f98aa7f7435ec62ae15191292702580de79
 SHA512:
-  metadata.gz: 4241c63fc4c647c44860aa0781e2966a5eadc1d7edea259851018308d1877d664c46a8d90bd5553f725d2e1fa5f767995b0f27327482bf2ce5128701908aa684
-  data.tar.gz: 64d599559055f6d8b3c384b88b5268b02317784bd44fd6be2384fd3c56b1f9da5a9cfb435c4af05c083ba20af6d8865f0c73a8fc46afff261ad70d43fc9c35e4
+  metadata.gz: 9821d7010a4663ce25497a129c5a04166b5f701bd30d1d7a20f183cd274444605f3ac14099857c22117d45ac980056f4ccbaf09e8b65403e83a93abc923ced90
+  data.tar.gz: ec1f98a8885f8938ba421d87cb82f7f16a79161a778d2e4c230204f2243009b001e4c5de8c95ca1153746a8eebb90f969ca5efe80390cd08f901ee943a06581f

data/CHANGELOG.md

@@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.4.2] - 2021-10-14
+
+- Add sidekiq middleware
+- Select checkpoints to enable or disable
+
 ## [0.4.1] - 2021-10-02
 
 - Fix Net::HTTP patch to override timeouts given to start
@@ -34,7 +39,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Mysql2 patch
 
 [Unreleased]: https://github.com/justinhoward/cutoff/compare/v0.4.1...HEAD
-[0.4.0]: https://github.com/justinhoward/cutoff/compare/v0.4.0...v0.4.1
+[0.4.2]: https://github.com/justinhoward/cutoff/compare/v0.4.1...v0.4.2
+[0.4.1]: https://github.com/justinhoward/cutoff/compare/v0.4.0...v0.4.1
 [0.4.0]: https://github.com/justinhoward/cutoff/compare/v0.3.0...v0.4.0
 [0.3.0]: https://github.com/justinhoward/cutoff/compare/v0.2.0...v0.3.0
 [0.2.0]: https://github.com/justinhoward/cutoff/compare/v0.1.0...v0.2.0

data/README.md

@@ -217,6 +217,29 @@ Cutoff.wrap(3) do
 end
 ```
 
+Selecting Checkpoints
+---------------------------
+
+In some cases, you may want to select some checkpoints to use, but not others.
+For example, you may want to run some code that contains MySQL queries, but not
+use the mysql2 patch. The `exclude` and `only` options support this.
+
+```ruby
+Cutoff.wrap(10, exclude: :mysql2) do
+  # The mysql2 patch won't be used here
+end
+
+Cutoff.wrap(10, only: %i[foo bar]) do
+  # These checkpoints will be used
+  Cutoff.checkpoint!(:foo)
+  Cutoff.checkpoint!(:bar)
+
+  # These checkpoints will be skipped
+  Cutoff.checkpoint!(:asdf)
+  Cutoff.checkpoint!
+end
+```
+
 Timing a Rails Controller
 ---------------------------
 
@@ -277,6 +300,26 @@ class ApplicationController < ActionController::Base
 end
 ```
 
+Timing Sidekiq Workers
+------------
+
+If Sidekiq is loaded, Cutoff includes middleware to support a `:cutoff` option.
+
+```ruby
+class MyWorker
+  include Sidekiq::Worker
+
+  sidekiq_options cutoff: 6.0
+
+  def perform
+    # ...
+    Cutoff.checkpoint!
+    # ...
+  end
+end
+```
+
+
 Disabling Cutoff for Testing and Development
 ------------
 

data/lib/cutoff/patch/mysql2.rb

@@ -6,7 +6,8 @@ require 'mysql2'
 class Cutoff
   module Patch
     # Sets the max execution time for SELECT queries if there is an active
-    # cutoff and it has time remaining
+    # cutoff and it has time remaining. You can select this patch with
+    # `exclude` or `only` using the checkpoint name `:mysql2`.
     module Mysql2
       # Overrides `Mysql2::Client#query` to insert a MAX_EXECUTION_TIME query
       # hint with the remaining cutoff time
@@ -19,9 +20,9 @@ class Cutoff
       #   be executed in this case.
       def query(sql, options = {})
         cutoff = Cutoff.current
-        return super unless cutoff
+        return super unless cutoff&.selected?(:mysql2)
 
-        cutoff.checkpoint!
+        cutoff.checkpoint!(:mysql2)
         sql = QueryWithMaxTime.new(sql, cutoff.ms_remaining.ceil).to_s
         super
       end

data/lib/cutoff/patch/net_http.rb

@@ -4,6 +4,9 @@ require 'net/http'
 
 class Cutoff
   module Patch
+    # Set checkpoints for Ruby HTTP requests. Also sets the Net::HTTP timeouts
+    # to the remaining cutoff time. You can select this patch with
+    # `exclude` or `only` using the checkpoint name `:net_http`.
     module NetHttp
       def self.gen_timeout_method(name)
         <<~RUBY
@@ -23,13 +26,13 @@ class Cutoff
       # @see Net::HTTP#start
       module_eval(<<~RUBY, __FILE__, __LINE__ + 1)
         def start
-          if (cutoff = Cutoff.current)
+          if (cutoff = Cutoff.current) && cutoff.selected?(:net_http)
             remaining = cutoff.seconds_remaining
             #{gen_timeout_method('open_timeout')}
             #{gen_timeout_method('read_timeout')}
             #{gen_timeout_method('write_timeout') if use_write_timeout?}
             #{gen_timeout_method('continue_timeout')}
-            Cutoff.checkpoint!
+            Cutoff.checkpoint!(:net_http)
           end
           super
         end

data/lib/cutoff/rails.rb

@@ -1,7 +1,3 @@
 # frozen_string_literal: true
 
-# Namespace for Rails integration
-module Rails
-end
-
 require 'cutoff/rails/controller' if Gem.loaded_specs['actionpack']

data/lib/cutoff/sidekiq.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require 'sidekiq'
+
+class Cutoff
+  module Sidekiq
+    # Add an option `cutoff` for sidekiq workers
+    #
+    # @example
+    #   class MyWorker
+    #     include Sidekiq::Worker
+    #
+    #     sidekiq_options cutoff: 6.0
+    #
+    #     def perform
+    #       # ...
+    #     end
+    #   end
+    class ServerMiddleware
+      # @param worker [Object] the worker instance
+      # @param _job [Hash] the full job payload
+      # @param _queue [String] queue the name of the queue the job was pulled
+      #   from
+      # @yield the next middleware in the chain or worker `perform` method
+      # @return [void]
+      def call(worker, _job, _queue)
+        allowed_seconds = worker.class.sidekiq_options['cutoff']
+        return yield if allowed_seconds.nil?
+
+        Cutoff.wrap(allowed_seconds) { yield }
+      end
+    end
+  end
+end
+
+::Sidekiq.configure_server do |config|
+  config.server_middleware do |chain|
+    chain.add(Cutoff::Sidekiq::ServerMiddleware)
+  end
+end

data/lib/cutoff/version.rb

@@ -3,6 +3,6 @@
 class Cutoff
   # @return [Gem::Version] The current version of the cutoff gem
   def self.version
-    Gem::Version.new('0.4.1')
+    Gem::Version.new('0.4.2')
   end
 end

data/lib/cutoff.rb

@@ -1,10 +1,14 @@
 # frozen_string_literal:true
 
+require 'set'
+
 require 'cutoff/version'
 require 'cutoff/error'
 require 'cutoff/patch'
 require 'cutoff/timer'
+
 require 'cutoff/rails'
+require 'cutoff/sidekiq' if Gem.loaded_specs['sidekiq']
 
 class Cutoff
   CURRENT_STACK_KEY = 'cutoff_deadline_stack'
@@ -25,13 +29,14 @@ class Cutoff
     # If a cutoff is already started for this thread, then `start` uses the
     # minimum of the current remaining time and the given time
     #
-    # @param seconds [Float, Integer] The number of seconds for the cutoff. May
-    #   be overridden if there is an active cutoff and it has less remaining
-    #   time.
+    # @param (see #initialize)
     # @return [Cutoff] The {Cutoff} instance
-    def start(seconds)
-      seconds = [seconds, current.seconds_remaining].min if current
-      cutoff = Cutoff.new(seconds)
+    def start(allowed_seconds, **options)
+      if current
+        allowed_seconds = [allowed_seconds, current.seconds_remaining].min
+      end
+
+      cutoff = new(allowed_seconds, **options)
       Thread.current[CURRENT_STACK_KEY] ||= []
       Thread.current[CURRENT_STACK_KEY] << cutoff
       cutoff
@@ -68,9 +73,10 @@ class Cutoff
     #
     # @see .start
     # @see .stop
+    # @param (see #initialize)
     # @return The value that returned from the block
-    def wrap(seconds)
-      cutoff = start(seconds)
+    def wrap(allowed_seconds, **options)
+      cutoff = start(allowed_seconds, **options)
       yield cutoff
     ensure
       stop(cutoff)
@@ -82,11 +88,11 @@ class Cutoff
     #
     # @raise CutoffExceededError If there is an active expired cutoff
     # @return [void]
-    def checkpoint!
+    def checkpoint!(name = nil)
       cutoff = current
       return unless cutoff
 
-      cutoff.checkpoint!
+      cutoff.checkpoint!(name)
     end
 
     # Disable Cutoff globally. Useful for testing and debugging
@@ -107,7 +113,7 @@ class Cutoff
       @disabled = false
     end
 
-    # True if cutoff was disabled with {#disable!}
+    # True if cutoff was disabled with {.disable!}
     #
     # @return [Boolean] True if disabled
     def disabled?
@@ -122,10 +128,22 @@ class Cutoff
   #
   # The timer starts immediately upon creation
   #
-  # @param allowed_seconds [Integer, Float] The total number of seconds to allow
-  def initialize(allowed_seconds)
+  # @param allowed_seconds [Float, Integer] The total number of seconds to allow
+  # @param exclude [Enumberable<Symbol>, Symbol, nil] If given a name or
+  #   list of checkpoint names to skip
+  # @param only [Enumberable<Symbol>, Symbol, nil] If given a name or
+  #   list of checkpoint names to allow
+  def initialize(allowed_seconds, exclude: nil, only: nil)
     @allowed_seconds = allowed_seconds.to_f
     @start_time = Cutoff.now
+
+    if exclude
+      @exclude = Set.new(exclude.is_a?(Enumerable) ? exclude : [exclude])
+    end
+
+    if only
+      @only = Set.new(only.is_a?(Enumerable) ? only : [only])
+    end
   end
 
   # The number of seconds left on the clock
@@ -162,9 +180,29 @@ class Cutoff
   #
   # @raise CutoffExceededError If there is an active expired cutoff
   # @return [void]
-  def checkpoint!
+  def checkpoint!(name = nil)
+    unless name.nil? || name.is_a?(Symbol)
+      raise ArgumentError, 'name must be a symbol'
+    end
+
+    return unless selected?(name)
     raise CutoffExceededError, self if exceeded?
 
     nil
   end
+
+  # True if the named checkpoint is selected by the `exclude` and `only`
+  # options. If these options are not given, a checkpoint is considered to be
+  # selected. If the checkpoint is not named, it is also considered to be
+  # selected.
+  #
+  # @param name [Symbol, nil] The name of the checkpoint
+  # @return [Boolean] True if the checkpoint is selected
+  def selected?(name)
+    return true if name.nil? && @exclude
+    return false if @exclude&.include?(name)
+    return false if @only && !@only.include?(name)
+
+    true
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cutoff
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors:
 - Justin Howard
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-10-02 00:00:00.000000000 Z
+date: 2021-10-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -104,6 +104,7 @@ files:
 - lib/cutoff/patch/net_http.rb
 - lib/cutoff/rails.rb
 - lib/cutoff/rails/controller.rb
+- lib/cutoff/sidekiq.rb
 - lib/cutoff/timer.rb
 - lib/cutoff/version.rb
 homepage: https://github.com/justinhoward/cutoff
@@ -111,7 +112,7 @@ licenses:
 - MIT
 metadata:
   changelog_uri: https://github.com/justinhoward/cutoff/blob/master/CHANGELOG.md
-  documentation_uri: https://www.rubydoc.info/gems/cutoff/0.4.1
+  documentation_uri: https://www.rubydoc.info/gems/cutoff/0.4.2
 post_install_message: 
 rdoc_options: []
 require_paths: