INSERT with RETURNING clause in PostgreSQL - Time & Space Complexity

Time Complexity: INSERT with RETURNING clause

O(n)

Understanding Time Complexity

When we add new rows to a database table using INSERT, it is helpful to know how the time to do this grows as we add more rows.

We want to understand how the database handles inserting data and returning results as the input size changes.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

INSERT INTO employees (name, department, salary)
VALUES ('Alice', 'Sales', 50000),
       ('Bob', 'HR', 45000),
       ('Carol', 'IT', 60000)
RETURNING id, name;
    

This code inserts three new employee records and returns their generated IDs and names.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Inserting each row into the table and writing it to disk.
• How many times: Once for each row in the VALUES list (3 times here, but can be many more).

How Execution Grows With Input

As the number of rows to insert grows, the work grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	About 10 insert operations and 10 returned rows
100	About 100 insert operations and 100 returned rows
1000	About 1000 insert operations and 1000 returned rows

Pattern observation: The time grows linearly as you add more rows to insert and return.

Final Time Complexity

Time Complexity: O(n)

This means the time to insert and return rows grows directly with the number of rows you add.

Common Mistake

[X] Wrong: "The RETURNING clause makes the insert take much longer regardless of how many rows are inserted."

[OK] Correct: Returning data adds some work, but it scales with the number of rows inserted, not independently. The main cost is still inserting each row.

Interview Connect

Understanding how inserting multiple rows and returning data scales helps you explain database performance clearly and confidently in real situations.

Self-Check

"What if we changed the INSERT to add rows one at a time inside a loop instead of all at once? How would the time complexity change?"

Practice

1. What does the RETURNING clause do in a PostgreSQL INSERT statement?

easy

A. It creates a backup of the inserted data.

B. It deletes rows after insertion.

C. It updates existing rows instead of inserting.

D. It returns specified columns of the newly inserted rows immediately.

Solution

1. Step 1: Understand the purpose of RETURNING

   The RETURNING clause in PostgreSQL is used to get back specified columns from the rows that were just inserted.

2. Step 2: Compare with other options

   The other options describe actions unrelated to RETURNING. RETURNING does not delete, update, or backup data.

3. Final Answer:

   It returns specified columns of the newly inserted rows immediately. -> Option D

4. Quick Check:

   RETURNING returns inserted data [OK]

Hint: RETURNING fetches inserted row data instantly [OK]

Common Mistakes:

• Thinking RETURNING deletes or updates data
• Confusing RETURNING with SELECT
• Assuming RETURNING creates backups

2. Which of the following is the correct syntax to insert a row and return the generated id in PostgreSQL?

easy

A. INSERT INTO users VALUES ('Alice') RETURNING id;

B. INSERT INTO users (name) RETURN id VALUES ('Alice');

C. INSERT INTO users (name) VALUES ('Alice') RETURNING id;

D. INSERT INTO users (name) VALUES ('Alice') RETURN id;

Solution

1. Step 1: Check correct RETURNING syntax

   The correct syntax places RETURNING after VALUES and specifies the column to return, like RETURNING id.

2. Step 2: Validate each option

   INSERT INTO users (name) VALUES ('Alice') RETURNING id; matches the correct syntax. Placing RETURNING before VALUES or using RETURN instead of RETURNING is incorrect. INSERT INTO users VALUES ('Alice') RETURNING id; misses column list before VALUES.

3. Final Answer:

   INSERT INTO users (name) VALUES ('Alice') RETURNING id; -> Option C

4. Quick Check:

   RETURNING follows VALUES with column list [OK]

Hint: RETURNING goes right after VALUES clause [OK]

Common Mistakes:

• Placing RETURNING before VALUES
• Omitting column list in INSERT
• Using RETURN instead of RETURNING

3. Given the table products(id SERIAL PRIMARY KEY, name TEXT, price NUMERIC), what will this query return?

INSERT INTO products (name, price) VALUES ('Pen', 1.5) RETURNING id, price;

medium

A. The new product's id and price, e.g., (5, 1.5)

B. Only the id of the new product

C. An error because RETURNING can't return multiple columns

D. Nothing, INSERT does not return rows

Solution

1. Step 1: Understand RETURNING with multiple columns

   RETURNING can return one or more columns from the inserted row(s). Here, id and price are requested.

2. Step 2: Predict output

   The query inserts a new product and returns its auto-generated id and the price 1.5 as a row, e.g., (5, 1.5).

3. Final Answer:

   The new product's id and price, e.g., (5, 1.5) -> Option A

4. Quick Check:

   RETURNING multiple columns returns all requested [OK]

Hint: RETURNING can list multiple columns separated by commas [OK]

Common Mistakes:

• Assuming RETURNING returns only one column
• Expecting no output from INSERT
• Thinking RETURNING causes syntax error with multiple columns

4. What is wrong with this query?

INSERT INTO orders (product_id, quantity) VALUES (10, 3) RETURNING;

medium

A. VALUES clause is incorrectly formatted.

B. RETURNING clause is missing column names to return.

C. INSERT statement cannot have RETURNING clause.

D. product_id and quantity columns do not exist.

Solution

1. Step 1: Check RETURNING syntax

   RETURNING must specify at least one column or * to return. Here, it is empty, which is invalid syntax.

2. Step 2: Verify other parts

   VALUES clause and columns look correct. INSERT supports RETURNING. Assuming columns exist, no error there.

3. Final Answer:

   RETURNING clause is missing column names to return. -> Option B

4. Quick Check:

   RETURNING needs columns or * [OK]

Hint: RETURNING must specify columns or * [OK]

Common Mistakes:

• Leaving RETURNING empty
• Confusing RETURNING with WHERE
• Assuming RETURNING is optional but empty

5. You want to insert multiple rows into employees(name, department) and get back their generated id values. Which query correctly does this and returns all new ids?

hard

A. INSERT INTO employees (name, department) VALUES ('John', 'HR'), ('Jane', 'IT') RETURNING id;

B. INSERT INTO employees (name, department) RETURNING id VALUES ('John', 'HR'), ('Jane', 'IT');

C. INSERT INTO employees (name, department) VALUES ('John', 'HR'); RETURNING id; INSERT INTO employees (name, department) VALUES ('Jane', 'IT'); RETURNING id;

D. INSERT INTO employees (name, department) VALUES ('John', 'HR'), ('Jane', 'IT') RETURN id;

Solution

1. Step 1: Understand multi-row INSERT with RETURNING

   PostgreSQL allows inserting multiple rows in one statement and RETURNING returns all inserted rows' specified columns.

2. Step 2: Analyze each option

   INSERT INTO employees (name, department) VALUES ('John', 'HR'), ('Jane', 'IT') RETURNING id; correctly inserts two rows and returns their ids. INSERT INTO employees (name, department) RETURNING id VALUES ('John', 'HR'), ('Jane', 'IT'); misplaces RETURNING before VALUES. INSERT INTO employees (name, department) VALUES ('John', 'HR'); RETURNING id; INSERT INTO employees (name, department) VALUES ('Jane', 'IT'); RETURNING id; splits inserts and RETURNING incorrectly. INSERT INTO employees (name, department) VALUES ('John', 'HR'), ('Jane', 'IT') RETURN id; uses invalid RETURN keyword.

3. Final Answer:

   INSERT INTO employees (name, department) VALUES ('John', 'HR'), ('Jane', 'IT') RETURNING id; -> Option A

4. Quick Check:

   Multi-row INSERT with RETURNING returns all rows [OK]

Hint: Use RETURNING after multi-row VALUES to get all ids [OK]

Common Mistakes:

• Placing RETURNING before VALUES
• Splitting inserts instead of multi-row insert
• Using RETURN instead of RETURNING