if điều kiện trong truy vấn cập nhật máy chủ sql

Một cái gì đó như thế này sẽ hoạt động:

UPDATE
    table_Name
SET 
  column_A = CASE WHEN @flag = '1' THEN column_A + @new_value ELSE column_A END,
  column_B = CASE WHEN @flag = '0' THEN column_B + @new_value ELSE column_B END
WHERE
    ID = @ID