FOREACH for array iteration in PostgreSQL

Introduction

FOREACH helps you look at each item in a list one by one inside a database function. It makes working with lists easy without writing complex loops.

When you want to process each item in a list stored in a database array.

When you need to perform an action for every element in an array inside a PostgreSQL function.

When you want to check or modify each value in an array one by one.

When you want to insert or update rows based on each element of an array.

When you want to avoid writing manual loops and use a simple way to handle arrays.

Syntax

PostgreSQL

FOREACH element_variable IN ARRAY array_variable LOOP
    -- statements using element_variable
END LOOP;

The element_variable holds the current item from the array during each loop.

The array_variable is the array you want to go through.

Examples

This example prints each number from the array one by one.

PostgreSQL

DECLARE
    current_number integer;
    numbers integer[] := ARRAY[1, 2, 3];
BEGIN
    FOREACH current_number IN ARRAY numbers LOOP
        RAISE NOTICE 'Number: %', current_number;
    END LOOP;
END;

This example goes through a text array and prints each fruit.

PostgreSQL

DECLARE
    current_text text;
    texts text[] := ARRAY['apple', 'banana', 'cherry'];
BEGIN
    FOREACH current_text IN ARRAY texts LOOP
        RAISE NOTICE 'Fruit: %', current_text;
    END LOOP;
END;

If the array is empty, the loop does not run at all.

PostgreSQL

DECLARE
    current_item integer;
    empty_array integer[] := ARRAY[]::integer[];
BEGIN
    FOREACH current_item IN ARRAY empty_array LOOP
        RAISE NOTICE 'This will not print';
    END LOOP;
END;

Sample Program

This program shows the array and then prints each value inside it using FOREACH.

PostgreSQL

DO $$
DECLARE
    current_value integer;
    sample_array integer[] := ARRAY[10, 20, 30, 40];
BEGIN
    RAISE NOTICE 'Array before loop: %', sample_array;
    FOREACH current_value IN ARRAY sample_array LOOP
        RAISE NOTICE 'Current value: %', current_value;
    END LOOP;
END $$;

Important Notes

FOREACH runs in O(n) time where n is the number of elements in the array.

It uses constant extra space, just one variable to hold the current element.

A common mistake is forgetting to declare the element variable with the correct type.

Use FOREACH when you want simple, readable code to process arrays instead of manual index loops.

Summary

FOREACH lets you easily go through each item in a PostgreSQL array.

It simplifies looping by handling the array traversal for you.

Use it inside PL/pgSQL functions to work with array elements one by one.

Practice

1. What is the main purpose of the FOREACH statement in PostgreSQL when working with arrays?

easy

A. To convert an array into a string.

B. To loop through each element of an array easily inside PL/pgSQL.

C. To sort the elements of an array in ascending order.

D. To create a new array from existing tables.

Solution

1. Step 1: Understand FOREACH usage context

   FOREACH is used inside PL/pgSQL to iterate over array elements one by one.

2. Step 2: Identify the main purpose

   It simplifies looping through arrays without manual index handling.

3. Final Answer:

   To loop through each element of an array easily inside PL/pgSQL. -> Option B

4. Quick Check:

   FOREACH loops over array elements [OK]

Hint: FOREACH loops over array elements inside PL/pgSQL [OK]

Common Mistakes:

• Confusing FOREACH with array creation or sorting
• Thinking FOREACH works outside PL/pgSQL
• Assuming FOREACH converts arrays to strings

2. Which of the following is the correct syntax to iterate over an integer array arr using FOREACH in PL/pgSQL?

easy

A. FOREACH element IN ARRAY arr LOOP ... END LOOP;

B. FOREACH element FROM arr LOOP ... END LOOP;

C. FOREACH element ON arr LOOP ... END LOOP;

D. FOREACH element OVER arr LOOP ... END LOOP;

Solution

1. Step 1: Recall FOREACH syntax

   The correct syntax uses 'IN ARRAY' to specify the array to loop over.

2. Step 2: Match syntax options

   Only FOREACH element IN ARRAY arr LOOP ... END LOOP; uses 'IN ARRAY' correctly; others use invalid keywords.

3. Final Answer:

   FOREACH element IN ARRAY arr LOOP ... END LOOP; -> Option A

4. Quick Check:

   FOREACH ... IN ARRAY ... is correct syntax [OK]

Hint: Use 'IN ARRAY' to loop over arrays with FOREACH [OK]

Common Mistakes:

• Using FROM, ON, or OVER instead of IN ARRAY
• Omitting LOOP or END LOOP keywords
• Trying to use FOREACH outside PL/pgSQL

3. Consider the following PL/pgSQL block:

DECLARE
  arr integer[] := ARRAY[2, 4, 6];
  sum integer := 0;
BEGIN
  FOREACH val IN ARRAY arr LOOP
    sum := sum + val;
  END LOOP;
  RAISE NOTICE '%', sum;
END;

What will be the output when this block runs?

medium

A. Syntax error

B. 24

C. 12

D. 0

Solution

1. Step 1: Understand the loop iteration

   The FOREACH loops over arr elements: 2, 4, and 6.

2. Step 2: Calculate the sum

   sum starts at 0, then adds 2 + 4 + 6 = 12.

3. Final Answer:

   12 -> Option C

4. Quick Check:

   2 + 4 + 6 = 12 [OK]

Hint: Sum array elements by adding each in FOREACH loop [OK]

Common Mistakes:

• Assuming FOREACH does not add elements
• Confusing sum initialization
• Expecting syntax error due to RAISE NOTICE

4. Identify the error in this PL/pgSQL snippet:

DECLARE
  arr text[] := ARRAY['a', 'b', 'c'];
  ch text;
BEGIN
  FOREACH ch IN arr LOOP
    RAISE NOTICE '%', ch;
  END LOOP;
END;

medium

A. Variable 'ch' must be declared as integer, not text.

B. Array declaration syntax is incorrect.

C. RAISE NOTICE cannot print variables inside loops.

D. FOREACH must use 'IN ARRAY' before the array variable.

Solution

1. Step 1: Check FOREACH syntax

   FOREACH requires 'IN ARRAY' before the array variable to iterate properly.

2. Step 2: Identify the error

   The code uses 'IN arr' instead of 'IN ARRAY arr', causing syntax error.

3. Final Answer:

   FOREACH must use 'IN ARRAY' before the array variable. -> Option D

4. Quick Check:

   FOREACH ... IN ARRAY ... is correct syntax [OK]

Hint: Always write 'FOREACH var IN ARRAY arr' [OK]

Common Mistakes:

• Omitting 'ARRAY' keyword in FOREACH
• Wrong variable type for array elements
• Misunderstanding RAISE NOTICE usage

5. You want to write a PL/pgSQL function that takes an integer array and returns the count of even numbers using FOREACH. Which of the following code snippets correctly implements this?

hard

A. CREATE FUNCTION count_evens(arr integer[]) RETURNS integer AS $$
DECLARE
val integer;
count integer := 0;
BEGIN
FOREACH val IN ARRAY arr LOOP
IF val % 2 = 0 THEN
count := count + 1;
END IF;
END LOOP;
RETURN count;
END;
$$ LANGUAGE plpgsql;

B. CREATE FUNCTION count_evens(arr integer[]) RETURNS integer AS $$
DECLARE
val integer;
count integer := 0;
BEGIN
FOREACH val FROM arr LOOP
IF val % 2 = 0 THEN
count := count + 1;
END IF;
END LOOP;
RETURN count;
END;
$$ LANGUAGE plpgsql;

C. CREATE FUNCTION count_evens(arr integer[]) RETURNS integer AS $$
DECLARE
val integer;
count integer := 0;
BEGIN
FOR val IN ARRAY arr LOOP
IF val % 2 = 0 THEN
count := count + 1;
END IF;
END LOOP;
RETURN count;
END;
$$ LANGUAGE plpgsql;

D. CREATE FUNCTION count_evens(arr integer[]) RETURNS integer AS $$
DECLARE
val integer;
count integer := 0;
BEGIN
FOREACH val ON ARRAY arr LOOP
IF val % 2 = 0 THEN
count := count + 1;
END IF;
END LOOP;
RETURN count;
END;
$$ LANGUAGE plpgsql;

Solution

1. Step 1: Check FOREACH syntax correctness

   CREATE FUNCTION count_evens(arr integer[]) RETURNS integer AS $$ DECLARE val integer; count integer := 0; BEGIN FOREACH val IN ARRAY arr LOOP IF val % 2 = 0 THEN count := count + 1; END IF; END LOOP; RETURN count; END; $$ LANGUAGE plpgsql; uses 'FOREACH val IN ARRAY arr LOOP', which is the correct syntax.

2. Step 2: Verify logic for counting even numbers

   Inside the loop, it checks if val is even and increments count correctly.

3. Final Answer:

   FOREACH val IN ARRAY arr LOOP with IF val % 2 = 0 logic -> Option A

4. Quick Check:

   FOREACH ... IN ARRAY ... and correct IF condition [OK]

Hint: Use 'FOREACH val IN ARRAY arr' and check val % 2 = 0 [OK]

Common Mistakes:

• Using 'FROM' or 'ON ARRAY' instead of 'IN ARRAY'
• Confusing FOREACH with FOR loops
• Missing RETURN statement or wrong logic