Add hash map to track in-flight UID assignments so that we avoid leaking UIDs
with the new async code.