module Semian::Mysql2

def acquire_semian_resource(**)
  super
rescue ::Mysql2::Error => error
  if error.is_a?(PingFailure) || (!error.is_a?(::Mysql2::SemianError) && error.message.match?(CONNECTION_ERROR))
    semian_resource.mark_failed(error)
    error.semian_identifier = semian_identifier
  end
  raise
end