This is a very beautiful dua to make as much as possible. We should ask Allah to only take our life when He is pleased with us. 

Allahumma tawaffani wa anta radin ‘anni.

O Allah, take my life while you are pleased with me.