Sha256: 28fd70c0280dfead71d42bd3b376491dd49dcc4c52de019a2cbf09d6de4b1e7f
Contents?: true
Size: 1.96 KB
Versions: 10
Compression:
Stored size: 1.96 KB
Contents
# Copyright (c) 2022 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true
module Contrast
module Logger
# Our decorator for the Ougai logger allowing for timing and with_level
# methods.
module Time
# Log the message at the given level.
#
# @param level [String] the name of the method to use. Should be one of trace, debug, info, warn, error
# @param message [String] the message to log
def with_level level, message
send(level.to_sym, message)
end
# Log, at the debug level, the action with a message including the time it took for the wrapped function to
# complete. If not logging to debug, simply yield the given block.
#
# @param msgs [Array<Object>] the arguments to pass to the logger. msgs[0] will be modified to include the elapsed
# time.
# @param block [Block, Proc] the block to execute
def debug_with_time *msgs, &block
if debug?
log_with_time(:debug, *msgs, &block)
elsif block
yield
end
end
# Log, at the trace level, the action with a message including the time it took for the wrapped function to
# complete. If not logging to debug, simply yield the given block.
#
# @param msgs [Array<Object>] the arguments to pass to the logger. msgs[0] will be modified to include the elapsed
# time.
# @param block [Block, Proc] the block to execute
def trace_with_time *msgs, &block
if trace?
log_with_time(:trace, *msgs, &block)
elsif block
yield
end
end
private
def log_with_time level, *msgs
a = Contrast::Utils::Timer.now_ms
ret = yield if block_given?
z = Contrast::Utils::Timer.now_ms
msgs[0] = "#{ msgs[0] }: pid=#{ Process.pid }, elapsed=#{ z - a }ms"
send(level, *msgs)
ret
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems