Design a generic key-value store | Databricks Interview Question