Sha256: 8067927558a6dc3ec5b07e9f66f6fafef185fd544336b2e4f87db53f6fa3e7fb
Contents?: true
Size: 1.39 KB
Versions: 59
Compression:
Stored size: 1.39 KB
Contents
# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.
module NewRelic
module Agent
class Harvester
attr_accessor :starting_pid
# Inject target for after_fork call to avoid spawning thread in tests
def initialize(events, after_forker=NewRelic::Agent)
# We intentionally don't set our pid as started at this point.
# Startup routines must call mark_started when they consider us set!
@starting_pid = nil
@after_forker = after_forker
if events
events.subscribe(:start_transaction, &method(:on_transaction))
end
end
def on_transaction(*_)
return unless restart_in_children_enabled? &&
needs_restart? &&
harvest_thread_enabled?
restart_harvest_thread
end
def mark_started(pid = Process.pid)
@starting_pid = pid
end
def needs_restart?(pid = Process.pid)
@starting_pid != pid
end
def restart_in_children_enabled?
NewRelic::Agent.config[:restart_thread_in_children]
end
def harvest_thread_enabled?
!NewRelic::Agent.config[:disable_harvest_thread]
end
def restart_harvest_thread
@after_forker.after_fork(:force_reconnect => true)
end
end
end
end
Version data entries
59 entries across 59 versions & 2 rubygems