La forma correcta de actualizar un campo atómicamente en Postgres basándose en su valor actual

Tengo una tabla my_table en PostgreSQL v 12 que tiene dos campos llamados id y status.
Necesito escribir un procedimiento almacenado updateStatus que tenga tres parámetros (id, oldStatus y newStatus). Debería actualizar el estado de la fila a newStatus solo si actualmente se encuentra en oldStatus (de lo contrario, hacer algo más como lanzar una excepción). El procedimiento debería funcionar bien en modo concurrente.
¿Cuál es la mejor manera de lograr esto?
Pensé en el método sugerido aquí: https://stackoverflow.com/questions/11532550/atomic-update-select-in-postgres

¿Está bien? ¿O tal vez hay una forma más fácil de lograr el mismo efecto?

LOOP
   UPDATE my_table
   SET    status = _new_status_
   WHERE  id = (SELECT id FROM my_table WHERE id = _id_ and status = _old_status_
                FOR UPDATE SKIP LOCKED);  -- pg 9.5

CONTINUE WHEN FOUND; 

UPDATE my_table
   SET    status = _new_status_
   WHERE  id = (SELECT id FROM my_table WHERE id = _id_ and status = _old_status_
                FOR UPDATE);

EXIT WHEN NOT FOUND;  -- lanzar algún error tal vez, etc.
END LOOP;