i usually have the main method wait on a future at the end, and in my shutdown handler i just resolve that future