Contents

#
# Copyright (C) 2016 Conjur Inc.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of
# this software and associated documentation files (the "Software"), to deal in
# the Software without restriction, including without limitation the rights to
# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
# the Software, and to permit persons to whom the Software is furnished to do so,
# subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
require "puma_backlog_detector/version"
require "puma_backlog_detector/railtie" if defined? Rails

class PumaBacklogDetector
  def initialize flag_path, max_backlog = 16
    @flag = FlagFile.new flag_path
    @max_backlog = max_backlog
  end

  def check
    if server = Puma::Server.current
      @flag.set server.backlog > @max_backlog
    end
  end

  def check_periodically interval_s = 0.01
    while true
      check
      sleep interval_s
    end
  end

  def check_in_background interval_s = 0.01
    @thread = Thread.new { check_periodically interval_s }
  end

  def stop
    if @thread
      @thread.exit
      @thread.join
      @thread = nil
    end
  end

  class FlagFile
    def initialize path
      @path = path
      @exists = File.exists? path
      set false
    end

    def set value
      if @exists != value
        FileUtils.send((value ? :touch : :rm_f), @path)
        @exists = value
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
puma_backlog_detector-0.1.1	lib/puma_backlog_detector.rb