I have updated your example and used row selection persistence to maintain the selection state in the Grid. Here is the result that meets your requirements:

https://stackblitz.com/edit/angular-pe3m9kdq-i1qt5ben

I hope this helps.