Getting a (useful) notification and drifting off to other stuff is definitely a problem. Interestingly, a smart watch has helped me with that. I now read notifications on the watch, and about 75% of the time I don't have to take out my phone anymore.