Eso de agarrar el teléfono por 5 minutos para revisar un mensaje y terminar perdiendo media hora viendo videos nos pasa a todos. Para mí, configurar bien las notificaciones y silenciar lo innecesario es lo más práctico, la diferencia se nota muchísimo para no perder el foco. ¿Probaste usar el modo "No Molestar" o limitar el tiempo de pantalla desde los ajustes del sistema?