Dapatkan jumlah catatan yang terpengaruh oleh INSERT atau UPDATE di PostgreSQL

Saya tahu pertanyaan ini oooolllllld dan solusi saya bisa dibilang terlalu rumit, tapi itulah solusi favorit saya!

Bagaimanapun, saya harus melakukan hal yang sama dan membuatnya bekerja seperti ini:

-- 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;

Suatu hari nanti saya benar-benar harus menulis soneta cinta untuk klausa WITH PostgreSQL ...