Uzyskaj liczbę rekordów, których dotyczy INSERT lub UPDATE w PostgreSQL

Wiem, że to pytanie brzmi oooolllllld, a moje rozwiązanie jest prawdopodobnie zbyt skomplikowane, ale to mój ulubiony rodzaj rozwiązania!

W każdym razie musiałem zrobić to samo i sprawić, by działało tak:

-- Get count from INSERT
WITH rows AS (
    INSERT INTO distributors
        (did, dname)
    VALUES
        (DEFAULT, 'XYZ Widgets'),
        (DEFAULT, 'ABC Widgets')
    RETURNING 1
)
SELECT count(*) FROM rows;

-- Get count from UPDATE
WITH rows AS (
    UPDATE distributors
    SET dname = 'JKL Widgets'
    WHERE did <= 10
    RETURNING 1
)
SELECT count(*) FROM rows;

Któregoś dnia naprawdę muszę zabrać się za napisanie sonetu miłosnego do klauzuli WITH PostgreSQL...