bcachefs: Fix locking in bch2_alloc_write_key()