module Appsignal::Hooks::DataMapperLogListener

def log(message)
  # If scheme is SQL-like, try to sanitize it, otherwise clear the body
  if SQL_CLASSES.include?(self.class.to_s)
    body_content = message.query
    body_format = Appsignal::EventFormatter::SQL_BODY_FORMAT
  else
    body_content = ""
    body_format = Appsignal::EventFormatter::DEFAULT
  end
  # Record event
  Appsignal::Transaction.current.record_event(
    "query.data_mapper",
    "DataMapper Query",
    body_content,
    message.duration,
    body_format
  )
  super
end