Add source.on_shutdown.