Reference for Berakhot 9:14

פליגי בה רבי יעקב בר אידי ורבי אחא בר חנינא חד אמר אלו הם יסורין של אהבה כל שאין בהן בטול תורה שנאמר אשרי הגבר אשר תיסרנו יה ומתורתך תלמדנו

R. Jacob b. Iddi and Rab Aha b. Hannina [differ in opinion]. One of them says : Those are to be considered chastenings of love which do not involve neglect of Torah; as it is said, " Happy is the man whom Thou chastenest, O Lord, and teachest him out of Thy law " (Ps. xciv. 12).

Jastrow