OA - In-Memory Database

The "In-Memory Database" problem is a classic technical interview challenge used by Snowflake to evaluate a candidate's ability to design and implement complex data structures, handle concurrency, and manage data persistence. PracHub +2 0 4 3

Problem Statement

Design and implement a simplified In-Memory Database that supports basic data operations, transactional integrity (like BEGIN, COMMIT, ROLLBACK), and potentially advanced features like Time-to-Live (TTL) or data versioning. Snowflake Docs +1 1 0

Core Functional Requirements

1. • SET(name, value): Set the variable name to value.
   • GET(name): Return the current value of name.
   • DELETE(name) : Remove the variable name.
   • COUNT(value): Return the number of variables currently set to value.
2. SET(name, value): Set the variable name to value.
3. GET(name): Return the current value of name.
4. DELETE(name) : Remove the variable name.
5. COUNT(value): Return the number of variables currently set to value.
6. • BEGIN: Open a new transaction block.
   • COMMIT: Apply all changes from the current transaction and close it.
   • ROLLBACK: Undo all changes made in the most recent transaction block and close it.
7. BEGIN: Open a new transaction block.
8. COMMIT: Apply all changes from the current transaction and close it.
9. ROLLBACK: Undo all changes made in the most recent transaction block and close it.
10. • Ensure the database is thread-safe to handle multiple simultaneous read and write requests. java-success.com
11. Ensure the database is thread-safe to handle multiple simultaneous read and write requests. java-success.com

Implementation Steps

1. Define Primary Data StructuresUse a hash map (e.g., dict in Python or HashMap in Java) to store the current state for 𝑂(1) average-time complexity on lookups and updates. To support the COUNT command efficiently, maintain a secondary frequency map that tracks how many times each value appears across all keys.
2. Manage Transactional StateUse a stack to store transaction states. Each BEGIN command pushes a new "undo" or "delta" map onto the stack. This allows you to track changes made within that specific scope without immediately overwriting the global state.
3. Implement Rollback LogicWhen ROLLBACK is called, pop the top of the transaction stack. If you are using an undo-log approach, use the popped map to restore the previous values in the primary data structure.
4. Implement Commit LogicWhen COMMIT is called, collapse all transaction blocks in the stack into the permanent global state and clear the stack.

Example Execution

CommandActionResult/StateSET a 10Global update{a∶10}BEGINNew transactionStack: [{}]SET a 20Update in txn{a∶20}, Undo: {a∶10}ROLLBACKRevert txn{a∶10}, Stack: []GET aLookup106 lines; Line 1: Command Action Result/State; Line 2: SET a 10 Global update the set of all a such that 10 end-set; Line 3: BEGIN New transaction Stack: open bracket the empty set close bracket; Line 4: SET a 20 Update in txn the set of all a such that 20 end-set comma Undo: the set of all a such that 10 end-set; Line 5: ROLLBACK Revert txn the set of all a such that 10 end-set comma Stack: open bracket close bracket; Line 6: GET a Lookup 10 end-lines;CommandActionResult/StateSET 𝑎 10Global update{𝑎∶10}BEGINNew transactionStack: [{}]SET 𝑎 20Update in txn{𝑎∶20}, Undo: {𝑎∶10}ROLLBACKRevert txn{𝑎∶10}, Stack: []GET 𝑎Lookup10

Final Answer

The "In-Memory Database" interview question requires building a system that can manage key-value pairs with nested transaction support. The primary technical challenge lies in efficiently managing the stack of transactions so that ROLLBACK and COMMIT operations correctly restore or finalize state without degrading the 𝑂(1) performance of basic GET and SET operations.

Would you like a sample implementation of this database in a specific programming language like Python or Java?

[0] - PracHub - Practice Tech Interview Questions (Updated 2026) [1] - Transactions - Snowflake Documentation [2] - 18 Java scenarios based interview Questions and Answers [3] - Anthropic Coding & Algorithms Questions (Updated 2026) - PracHub [4] - Software Engineer Coding & Algorithms Interview Questions ...