Change volatile -> atomic.

Move flow shutdown before lock to fix rare locking issue where lock never gets acquired while output thread keeps locking/unlocking.