PostgreSQL存儲過程中怎么利用BEGIN塊處理事務

PostgreSQL存儲過程中怎么利用BEGIN塊處理事務，相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。

1、用BEGIN塊模擬

隨便建個表：

CREATE TABLE a(col1 int);

存儲過程如下：

CREATE OR REPLACE FUNCTION ins_t() RETURNS void
AS $$
BEGIN
  INSERT INTO a VALUES(100);
  BEGIN
    INSERT INTO a VALUES(200);
    RAISE 'any error';
  EXCEPTION
    WHEN others THEN
      null;
  END;
END;
$$ LANGUAGE plpgsql;

測試：

flying=# SELECT ins_t();                                                                                  ins_t
-------

(1 row)

flying=# SELECT * FROM a;
 col1
------
  100
(1 row)

可以看到，內嵌BEGIN塊中的INSERT操作并沒有生效。

2、PL/pgSQL如何實現

塊定義在pl_gram.y中，有興趣可以自己去看，塊的執行代碼位于 src/pl/plpgsql/src/pl_exec.c。

static int
exec_stmt_block(PLpgSQL_execstate *estate, PLpgSQL_stmt_block *block)
{
...

	if (block->exceptions)
	{
...
		BeginInternalSubTransaction(NULL);
...
		PG_TRY();
...
		PG_CATCH();
...
			RollbackAndReleaseCurrentSubTransaction();
...

只有當塊定義有EXCEPTION部分，才會出現創建子事務的動作，否則（它對應的else分支內）只會當做普通語句處理。拋出例外時，就把這個子事務會滾掉。

而如果不拋出例外，而是WARNING或者NOTICE之類，也不會引起它的回滾，嘗試一下就知道。PG怎么知道要不要回滾，關鍵在PG_TRY，這個說起來話長，暫時了解到這里足夠。

3、沒有EXCEPTION處理會怎樣

去掉例外處理部分

CREATE OR REPLACE FUNCTION ins_t() RETURNS void
AS $$
BEGIN
  INSERT INTO a VALUES(100);
  BEGIN
    INSERT INTO a VALUES(200);
    RAISE 'any error';
  END;
END;
$$ LANGUAGE plpgsql;

結果什么都沒有，因為它沒有啟用子事務，所以會全部回滾。

flying=# select ins_t();
ERROR:  any error
CONTEXT:  PL/pgSQL function ins_t() line 6 at RAISE
flying=# select * from a;
 col1
------
(0 rows)

flying=#

4、神奇的PG_TRY和PG_CATCH

它們用到的是sigsetjmp，PG中很多地方用到它，了解這個可能需要一定的匯編語言知識，也需要了解錯誤拋出機制，有機會再詳細講。

看完上述內容，你們掌握PostgreSQL存儲過程中怎么利用BEGIN塊處理事務的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！