Postgresql 函数返回受影响的行数

PostgreSQL 函数返回受影响的行数

sql
UPDATE employees SET salary = salary * 1.1 WHERE department = 'IT' RETURNING *;

sql
CREATE OR REPLACE FUNCTION update_employee_salary(department_name text, salary_increase numeric)
RETURNS integer AS $$
DECLARE
    affected_rows integer;
BEGIN
    UPDATE employees SET salary = salary * salary_increase WHERE department = department_name RETURNING * INTO affected_rows;
    
    IF affected_rows > 0 THEN
        RAISE NOTICE 'Salary updated for % employees in % department.', affected_rows, department_name;
    ELSE
        RAISE NOTICE 'No employees found in % department.', department_name;
    END IF;
    
    RETURN affected_rows;
END;
$$ LANGUAGE plpgsql;