Current section

Files

Jump to
new_relic_agent lib new_relic transaction monitor.ex
Raw

lib/new_relic/transaction/monitor.ex

defmodule NewRelic.Transaction.Monitor do
use GenServer
alias NewRelic.Transaction
alias NewRelic.DistributedTrace
# This GenServer watches transaction processes for
# - :trace messages
# - :DOWN messages
#
# :trace messages signal that a transaction process has
# spwaned another process. We want to track that process
# so we can track attributes on the original transaction
#
# :DOWN signals that the transaction process is complete, so
# we tell the reporter to purge its data
@moduledoc false
def start_link do
GenServer.start_link(__MODULE__, :ok, name: __MODULE__)
end
def init(:ok) do
NewRelic.sample_process()
enable_trace_patterns()
{:ok, %{pids: %{}}}
end
# API
def add(pid), do: GenServer.call(__MODULE__, {:add, pid})
# Server
def handle_call({:add, pid}, _from, %{pids: pids} = state) do
pids =
pids
|> Map.has_key?(pid)
|> setup_monitor(pids, pid)
{:reply, :ok, %{state | pids: pids}}
end
# Trace messages
def handle_info({:trace_ts, source, :spawn, pid, _mfa, timestamp}, state) do
Transaction.Reporter.track_spawn(source, pid, NewRelic.Util.time_to_ms(timestamp))
{:noreply, state}
end
def handle_info({:trace_ts, pid, :exit, _reason, timestamp}, state) do
Transaction.Reporter.track_exit(pid, NewRelic.Util.time_to_ms(timestamp))
{:noreply, state}
end
def handle_info(
{:trace_ts, source, :return_from, {:poolboy, :checkout, _}, pid, timestamp},
state
) do
Transaction.Reporter.track_spawn(source, pid, NewRelic.Util.time_to_ms(timestamp))
{:noreply, state}
end
def handle_info({:DOWN, _ref, :process, pid, _reason}, state) do
Transaction.Reporter.ensure_purge(pid)
Transaction.Reporter.complete(pid)
DistributedTrace.Tracker.cleanup(pid)
{:noreply, %{state | pids: Map.delete(state.pids, pid)}}
end
def handle_info(_msg, state) do
# Ignore other :trace messages
{:noreply, state}
end
# Helpers
defp setup_monitor(false, pids, pid) do
Process.monitor(pid)
enable_trace_flags(pid)
Map.put(pids, pid, true)
end
defp setup_monitor(true, pids, _) do
pids
end
def enable_trace_flags(pid) do
# Trace process events to notice when a process is spawned
# Trace function calls so we can install specific trace_patterns
# http://erlang.org/doc/man/erlang.html#trace-3
:erlang.trace(pid, true, [:procs, :call, :set_on_spawn, :timestamp])
rescue
# Process is already dead
ArgumentError ->
nil
end
def enable_trace_patterns do
# Use function tracers to notice when Async work has been kicked off
# http://erlang.org/doc/man/erlang.html#trace_3_trace_messages_return_from
# http://erlang.org/doc/apps/erts/match_spec.html
trace_poolboy_checkout()
end
defp trace_poolboy_checkout do
:erlang.trace_pattern({:poolboy, :checkout, :_}, [{:_, [], [{:return_trace}]}], [])
end
end