Mostrar el último mensaje en una matriz de mensajes

Dado que está utilizando Rails 5, puede aprovechar su or método:

Message.where(to: current_user.id).or(
 Message.where(user_id: current_user.id)
).order(connection: :desc, created_at: :desc)
 .select('distinct on (connection) *')