CASE in PL/pgSQL in PostgreSQL - Time & Space Complexity

Time Complexity: CASE in PL/pgSQL

O(n)

Understanding Time Complexity

We want to understand how the time it takes to run a CASE statement in PL/pgSQL changes as the number of conditions grows.

How does adding more conditions affect the work the database does?

Scenario Under Consideration

Analyze the time complexity of the following PL/pgSQL CASE statement.

DECLARE
  grade CHAR := 'B';
  result TEXT;
BEGIN
  CASE grade
    WHEN 'A' THEN result := 'Excellent';
    WHEN 'B' THEN result := 'Good';
    WHEN 'C' THEN result := 'Average';
    WHEN 'D' THEN result := 'Below Average';
    ELSE result := 'Fail';
  END CASE;
END;
    

This code checks the value of grade and sets result based on matching conditions.

Identify Repeating Operations

Look for repeated checks or comparisons inside the CASE.

• Primary operation: Comparing the input value to each WHEN condition.
• How many times: Up to the number of WHEN conditions, until a match is found.

How Execution Grows With Input

As the number of WHEN conditions increases, the number of comparisons can grow.

Input Size (number of WHENs)	Approx. Comparisons
5	Up to 5
50	Up to 50
500	Up to 500

Pattern observation: The number of comparisons grows roughly in a straight line with the number of conditions.

Final Time Complexity

Time Complexity: O(n)

This means the time to find a match grows linearly with the number of CASE conditions.

Common Mistake

[X] Wrong: "The CASE statement always runs in constant time no matter how many conditions there are."

[OK] Correct: The database checks conditions one by one until it finds a match, so more conditions mean more checks and more time.

Interview Connect

Understanding how CASE statements scale helps you write efficient conditional logic in database functions, a useful skill in many real projects.

Self-Check

"What if we replaced the CASE with a lookup table join? How would the time complexity change?"

Practice

1. What is the main purpose of using CASE in PL/pgSQL?

easy

A. To choose different actions based on conditions

B. To create loops that repeat actions

C. To define new tables in the database

D. To permanently store data in variables

Solution

1. Step 1: Understand the role of CASE

   CASE is used to select one action from many based on conditions, like a traffic light deciding when to stop or go.

2. Step 2: Compare with other options

   Loops repeat actions, table creation defines structure, and variables store data, none of which is the main role of CASE.

3. Final Answer:

   To choose different actions based on conditions -> Option A

4. Quick Check:

   CASE chooses actions based on conditions [OK]

Hint: CASE picks actions by conditions, not loops or storage [OK]

Common Mistakes:

• Confusing CASE with loops
• Thinking CASE creates tables
• Assuming CASE stores data permanently

2. Which of the following is the correct way to end a CASE block in PL/pgSQL?

easy

A. END;

B. STOP;

C. FINISH CASE;

D. END CASE;

Solution

1. Step 1: Recall PL/pgSQL syntax for CASE

   In PL/pgSQL, a CASE block must be closed with END CASE; to mark its end clearly.

2. Step 2: Check other options

   END; ends blocks like functions, but CASE specifically needs END CASE;. FINISH CASE; and STOP; are invalid keywords.

3. Final Answer:

   END CASE; -> Option D

4. Quick Check:

   CASE ends with END CASE; [OK]

Hint: Always close CASE with END CASE; in PL/pgSQL [OK]

Common Mistakes:

• Using END; alone to close CASE
• Writing FINISH CASE; which is invalid
• Using STOP; which is not a PL/pgSQL keyword

3. Consider this PL/pgSQL snippet:

DECLARE
  grade CHAR := 'B';
  result TEXT;
BEGIN
  CASE grade
    WHEN 'A' THEN result := 'Excellent';
    WHEN 'B' THEN result := 'Good';
    ELSE result := 'Average';
  END CASE;
  RETURN result;
END;

What will be the returned value?

medium

A. 'Good'

B. 'Average'

C. 'Excellent'

D. NULL

Solution

1. Step 1: Identify the value of grade

   The variable grade is set to 'B'.

2. Step 2: Match grade in CASE

   CASE checks 'B', matches the second WHEN clause, so result becomes 'Good'.

3. Final Answer:

   'Good' -> Option A

4. Quick Check:

   grade 'B' returns 'Good' [OK]

Hint: Match CASE value to WHEN clause for output [OK]

Common Mistakes:

• Choosing ELSE when a WHEN matches
• Confusing variable assignment inside CASE
• Assuming NULL if no ELSE present

4. Identify the error in this PL/pgSQL CASE block:

DECLARE
  score INT := 85;
  grade TEXT;
BEGIN
  CASE
    WHEN score >= 90 THEN grade := 'A';
    WHEN score >= 80 THEN grade := 'B';
    ELSE grade := 'C';
  END;
  RETURN grade;
END;

medium

A. Incorrect variable declaration

B. Invalid comparison operators

C. Missing END CASE; to close CASE block

D. No ELSE clause present

Solution

1. Step 1: Check CASE block ending

   The CASE block is closed with END; but PL/pgSQL requires END CASE; to close CASE.

2. Step 2: Verify other parts

   Variable declarations and comparisons are correct, and ELSE clause is present.

3. Final Answer:

   Missing END CASE; to close CASE block -> Option C

4. Quick Check:

   CASE must end with END CASE; [OK]

Hint: Close CASE with END CASE;, not just END; [OK]

Common Mistakes:

• Using END; instead of END CASE;
• Thinking ELSE is optional here
• Misreading comparison operators

5. You want to write a PL/pgSQL function that returns 'Pass' if a student's score is 50 or more, 'Fail' if below 50, and 'Invalid' if the score is NULL. Which CASE structure correctly implements this?

hard

A. CASE score
WHEN NULL THEN RETURN 'Invalid';
WHEN >= 50 THEN RETURN 'Pass';
ELSE RETURN 'Fail';
END CASE;

B. CASE
WHEN score IS NULL THEN RETURN 'Invalid';
WHEN score >= 50 THEN RETURN 'Pass';
ELSE RETURN 'Fail';
END CASE;

C. CASE
WHEN score >= 50 THEN RETURN 'Pass';
WHEN score IS NULL THEN RETURN 'Fail';
ELSE RETURN 'Invalid';
END CASE;

D. CASE score
WHEN score >= 50 THEN RETURN 'Pass';
WHEN score < 50 THEN RETURN 'Fail';
ELSE RETURN 'Invalid';
END CASE;

Solution

1. Step 1: Handle NULL explicitly

   Since NULL cannot be matched by simple WHEN, use WHEN score IS NULL to check NULL values.

2. Step 2: Order conditions correctly

   Check NULL first, then score >= 50 for 'Pass', else 'Fail'. This matches

   CASE
     WHEN score IS NULL THEN RETURN 'Invalid';
     WHEN score >= 50 THEN RETURN 'Pass';
     ELSE RETURN 'Fail';
   END CASE;

   .

3. Final Answer:

   CASE with WHEN score IS NULL, then score >= 50, else Fail -> Option B

4. Quick Check:

   Use IS NULL to check NULL in CASE [OK]

Hint: Use WHEN score IS NULL to test NULL in CASE [OK]

Common Mistakes:

• Trying to match NULL with WHEN NULL
• Using CASE score with conditions inside WHEN
• Not checking NULL before other conditions