1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
# frozen-string-literal: true
module Sequel
class Database
# ---------------------
# :section: 6 - Methods relating to logging
# This methods affect relating to the logging of executed SQL.
# ---------------------
# Numeric specifying the duration beyond which queries are logged at warn
# level instead of info level.
attr_accessor :log_warn_duration
# Array of SQL loggers to use for this database.
attr_accessor :loggers
# Whether to include information about the connection in use when logging queries.
attr_accessor :log_connection_info
# Log level at which to log SQL queries. This is actually the method
# sent to the logger, so it should be the method name symbol. The default
# is :info, it can be set to :debug to log at DEBUG level.
attr_accessor :sql_log_level
# Log a message at error level, with information about the exception.
def log_exception(exception, message)
log_each(:error, "#{exception.class}: #{exception.message.strip if exception.message}: #{message}")
end
# Log a message at level info to all loggers.
def log_info(message, args=nil)
log_each(:info, args ? "#{message}; #{args.inspect}" : message)
end
# Yield to the block, logging any errors at error level to all loggers,
# and all other queries with the duration at warn or info level.
def log_connection_yield(sql, conn, args=nil)
return yield if skip_logging?
sql = "#{connection_info(conn) if conn && log_connection_info}#{sql}#{"; #{args.inspect}" if args}"
timer = Sequel.start_timer
begin
yield
rescue => e
log_exception(e, sql)
raise
ensure
log_duration(Sequel.elapsed_seconds_since(timer), sql) unless e
end
end
# Remove any existing loggers and just use the given logger:
#
# DB.logger = Logger.new($stdout)
def logger=(logger)
@loggers = Array(logger)
end
private
# Determine if logging should be skipped. Defaults to true if no loggers
# have been specified.
def skip_logging?
@loggers.empty?
end
# String including information about the connection, for use when logging
# connection info.
def connection_info(conn)
"(conn: #{conn.__id__}) "
end
# Log the given SQL and then execute it on the connection, used by
# the transaction code.
def log_connection_execute(conn, sql)
log_connection_yield(sql, conn){conn.public_send(connection_execute_method, sql)}
end
# Log message with message prefixed by duration at info level, or
# warn level if duration is greater than log_warn_duration.
def log_duration(duration, message)
log_each((lwd = log_warn_duration and duration >= lwd) ? :warn : sql_log_level, "(#{sprintf('%0.6fs', duration)}) #{message}")
end
# Log message at level (which should be :error, :warn, or :info)
# to all loggers.
def log_each(level, message)
@loggers.each{|logger| logger.public_send(level, message)}
end
end
end
|
