PostgreSQL 數(shù)據(jù)值

在調(diào)用一個(gè) PL/Python 函數(shù)時(shí)，它的參數(shù)會被從 PostgreSQL 的數(shù)據(jù)類型轉(zhuǎn)換成相應(yīng)的 Python 類型：

當(dāng)一個(gè) PL/Python 函數(shù)返回時(shí)，會按照下列規(guī)則把它的返回值轉(zhuǎn)換成該函數(shù)聲明的 PostgreSQL 返回?cái)?shù)據(jù)類型：

注意所聲明的 PostgreSQL 返回類型和實(shí)際返回對象的 Python 數(shù)據(jù)類型之間的邏輯失配不會被標(biāo)志，無論怎樣該值都會被轉(zhuǎn)換。

如果一個(gè) SQL 空值 被傳遞給一個(gè)函數(shù)，該參數(shù)值將作為 Python 中的None出現(xiàn)。例如，第 45.2 節(jié)中展示的pymax的函數(shù)定義對于空值輸入將會返回錯(cuò)誤的回答。我們可以為函數(shù)定義增加STRICT讓 PostgreSQL做得更加合理：如果一個(gè)空值被傳入，該函數(shù)將根本不會被調(diào)用，而只是自動(dòng)地返回一個(gè)空結(jié)果。此外，我們可以在函數(shù)體中檢查空輸入：

CREATE FUNCTION pymax (a integer, b integer)
  RETURNS integer
AS $$
  if (a is None) or (b is None):
    return None
  if a > b:
    return a
  return b
$$ LANGUAGE plpythonu;

如前所示，要從一個(gè) PL/Python 函數(shù)返回一個(gè) SQL 空值，可返回值None。不管該函數(shù)嚴(yán)格與否都可以這樣做。

SQL 數(shù)組會被作為一個(gè) Python 列表傳遞到 PL/Python 中。要從一個(gè) PL/Python 函數(shù)中返回出一個(gè) SQL 數(shù)組值，可返回一個(gè)Python列表：

CREATE FUNCTION return_arr()
  RETURNS int[]
AS $$
return [1, 2, 3, 4, 5]
$$ LANGUAGE plpythonu;

SELECT return_arr();
 return_arr  
-------------
 {1,2,3,4,5}
(1 row)

多維數(shù)組被當(dāng)做嵌套的Python列表傳入PL/Python。例如，一個(gè)2維數(shù)組是一個(gè)列表的列表。在把一個(gè)多維SQL數(shù)組從PL/Python函數(shù)返回出去時(shí)，每一層的內(nèi)層列表都必須是相同的尺寸。例如：

CREATE FUNCTION test_type_conversion_array_int4(x int4[]) RETURNS int4[] AS $$
plpy.info(x, type(x))
return x
$$ LANGUAGE plpythonu;

SELECT * FROM test_type_conversion_array_int4(ARRAY[[1,2,3],[4,5,6]]);
INFO:  ([[1, 2, 3], [4, 5, 6]], <type 'list'>)
 test_type_conversion_array_int4 
---------------------------------
 {{1,2,3},{4,5,6}}
(1 row)

為了與PostgreSQL的9.6以及更低版本的向后兼容性，當(dāng)不支持多維數(shù)組時(shí)，也接受元組之類的其他Python序列。不過，它們總是被當(dāng)做一維數(shù)組，因?yàn)樗鼈儠徒M合類型混淆。出于同樣的原因，當(dāng)一個(gè)組合類型被用在多維數(shù)組中時(shí)，它必須被表示為一個(gè)元組而不是一個(gè)列表。

注意在 Python 中，串是序列，這可能產(chǎn)生與 Python 程序員所熟悉的不同的效果：

CREATE FUNCTION return_str_arr()
  RETURNS varchar[]
AS $$
return "hello"
$$ LANGUAGE plpythonu;

SELECT return_str_arr();
 return_str_arr
----------------
 {h,e,l,l,o}
(1 row)

組合類型參數(shù)被作為 Python 映射傳遞給函數(shù)。映射的元素名稱就是組合類型的屬性名。如果被傳遞的行中有一個(gè)屬性是空值，在映射中它的值是None。這里是一個(gè)例子：

CREATE TABLE employee (
  name text,
  salary integer,
  age integer
);

CREATE FUNCTION overpaid (e employee)
  RETURNS boolean
AS $$
  if e["salary"] > 200000:
    return True
  if (e["age"] < 30) and (e["salary"] > 100000):
    return True
  return False
$$ LANGUAGE plpythonu;

有多種方式從一個(gè) Python 函數(shù)返回行或者組合類型。下面的例子假設(shè)我們有：

CREATE TYPE named_value AS (
  name   text,
  value  integer
);

一個(gè)組合結(jié)果可以被返回為：

也支持具有OUT參數(shù)的函數(shù)。例如：

CREATE FUNCTION multiout_simple(OUT i integer, OUT j integer) AS $$
return (1, 2)
$$ LANGUAGE plpythonu;

SELECT * FROM multiout_simple();

過程的輸出參數(shù)會以同樣的方式傳回。例如：

CREATE PROCEDURE python_triple(INOUT a integer, INOUT b integer) AS $$
return (a * 3, b * 3)
$$ LANGUAGE plpythonu;

CALL python_triple(5, 10);

PL/Python函數(shù)也能返回標(biāo)量類型或者組合類型的集合。有多種方法可以做到這一點(diǎn)，因?yàn)楸环祷氐膶ο笤趦?nèi)部會被轉(zhuǎn)變成一個(gè)迭代器。下面的例子假設(shè)我們有組合類型：

CREATE TYPE greeting AS (
  how text,
  who text
);

可從以下類型返回集合結(jié)果：

也支持有OUT參數(shù)的集合返回函數(shù)（使用RETURNS SETOF record）。例如：

CREATE FUNCTION multiout_simple_setof(n integer, OUT integer, OUT integer) RETURNS SETOF record AS $$
return [(1, 2)] * n
$$ LANGUAGE plpythonu;

SELECT * FROM multiout_simple_setof(3);