Sidekiq process heap dump to a TCP socket

sidekiq.rb

if ENV["PROFILE"]
  require "objspace"
  ObjectSpace.trace_object_allocations_start
  Sidekiq.logger.info "allocations tracing enabled"

  module Sidekiq
    module Middleware
      module Server
        class Profiler
          # Number of jobs to process before reporting
          JOBS = Integer(ENV["PROFILE_JOBS"]) rescue 1000

          class << self
            mattr_accessor :counter
            self.counter = 0

            def synchronize(&block)
              @lock ||= Mutex.new
              @lock.synchronize(&block)
            end
          end

          def call(worker_instance, item, queue)
            begin
              yield
            ensure
              self.class.synchronize do
                self.class.counter += 1

                if self.class.counter % JOBS == 0
                  Sidekiq.logger.info "reporting allocations after #{self.class.counter} jobs"
                  sck = TCPSocket.new("159.203.95.41", "6666")
                  GC.start
                  ObjectSpace.dump_all output: sck
                  sck.close
                  Sidekiq.logger.info "written dump to file"
                end
              end
            end
          end
        end
      end
    end
  end
  Sidekiq.configure_server do |config|
    config.server_middleware do |chain|
      chain.add Sidekiq::Middleware::Server::Profiler
    end
  end
end