Revert making shutdown wait on Watch/Command/Balance

This broke tests and made it way too easy for balancer and coordinator
implementations to deadlock the consumer on shutdown. Balancer and
Coordinator implementations should be easy to write and misbehaving
implementations shouldn't be able to trivially deadlock the consumer.