One strategy that has helped with the "getting back to work" part is if I'm mid thought or in anyway aware of what comes next, I write that down on a post it note and stick it to my monitor.

It might be like

"Deploy X and test case Foo"

"Add widgetX to list of running widgets"

"self Review code and submit PR"