Challenge: Generate matching shard id

Here is something that came up that I liked as task to handle in a job interview. We have the following function, generating a shard id from a document key: There is a need to create a new document that must reside on the same shard as another document.