Life Will Go On!

Protagonist dies and goes to the afterlife, weirdly similar to Soviet sanatorium. He finds friends and love interest there, but farewell is soon - doctors are going to resuscitate him