写代码老容易分心，不够专注怎么治？

写代码老容易分心，效率不是很高 老是写完一段代码或者一个小功能点，或者头脑里对一个难点有解决方案了，就想停下来休息一下刷刷 v2 zhihu 啥的，不能一鼓作气搞定，有点拖延，到了下班然后象征性加加班搞搞，虽然也能搞定，但总觉得要是专注一点效率高一点能更好更快搞好吧。

不能沉迷写代码啊，讲道理，可能是自己没有对代码爱的深沉吧，总之不够专注，你们有什么好的治疗方案么？可以讨论一下。