如何實現PostgreSQL函數自定義例外處理

這篇文章主要為大家展示了“如何實現PostgreSQL函數自定義例外處理”，內容簡而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶領大家一起研究并學習一下“如何實現PostgreSQL函數自定義例外處理”這篇文章吧。

代碼搬運也需要發揮想象力，讓不可能變為可能，這里講一個例子。

1、 有人問PostgreSQL有沒有自定義例外，Oracle是有的：

--定義
myex Exception;
--拋出
RAISE myex;
--捕獲
WHEN myex THEN

簡單易用

2、再來看PostgreSQL的PL/pgSQL

RAISE [ level ] condition_name [ USING option = expression [, ... ] ];
RAISE [ level ] SQLSTATE 'sqlstate' [ USING option = expression [, ... ] ];

這兩種語法似乎有一定的靈活性，實際上它們只能使用預定義的例外，文檔中有說明，如果不能識別會報錯。

ERROR:  unrecognized exception condition "xxxxxxx"
CONTEXT:  compilation of PL/pgSQL function "func_a" near line 3

ERROR:  invalid SQLSTATE code at or near "'12345'"
LINE 5: RAISE SQLSTATE '12345' USING MESSAGE = 'zzz';

3、代碼實現

這段是強行加戲，防止篇幅過小的一個例外處理，完全可以跳過而不會有任何影響。

	for (i = 0; exception_label_map[i].label != NULL; i++)
	{
		if (strcmp(condname, exception_label_map[i].label) == 0)
			return exception_label_map[i].sqlerrstate;
	}

這里可以看出，能使用的名字或編碼在編譯時已經確定，不能自己定義。

4、但是，我們仍有辦法可以做到

盡量選擇一個上下文無關的錯誤，也就是這段代碼不可能會拋出的例外，避免程序出錯被錯誤攔截。

比如：0A000 feature_not_supported

我們可以拋出一個例外，用錯誤信息標記它的特殊性：

RAISE SQLSTATE '0A000' USING MESSAGE = 'flying0001:failed to ...';

在捕獲它時可以按條件處理：

WHEN SQLSTATE '0A000' THEN
  DECLARE
    m text;
  BEGIN
    GET STACKED DIAGNOSTICS m = MESSAGE_TEXT;
    IF ( strpos(m, 'flying0001:')=1) THEN
      RAISE WARNING 'got flying0001';
    ELSIF
      ...
    END IF;
  END;

雖然啰嗦，但確實是實現了同樣的功能。

5、完整演示

也是加戲，湊篇幅用。

CREATE OR REPLACE FUNCTION func_a()
RETURNS void AS
$$
BEGIN
RAISE SQLSTATE '0A000' USING MESSAGE = 'flying0001:Do one thing at a time, and do well.';
EXCEPTION
WHEN feature_not_supported THEN
  DECLARE
    m text;
  BEGIN
    GET STACKED DIAGNOSTICS m = MESSAGE_TEXT;
    IF (strpos(m, 'flying0001:')=1) THEN
      RAISE WARNING 'got flying0001';
    END IF;
  END;
END;
$$
LANGUAGE plpgsql;

以上是“如何實現PostgreSQL函數自定義例外處理”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！