PL/Python 寫的函數(shù)用通常的 CREATE FUNCTION 語法聲明。比如：

CREATE FUNCTION funcname (argument-list)

RETURNS return-type

AS $$

# PL/Python function body

$$ LANGUAGE plpythonu;

函數(shù)體是一個(gè)簡單的 Python 腳本。當(dāng)該函數(shù)被調(diào)用的時(shí)候，傳遞的參數(shù)將成為 args[] 數(shù)組的元素。命名參數(shù)也會(huì)被傳遞到腳本中并被當(dāng)作普通變量。結(jié)果將使用 return 或 yield 照常返回。

例如，返回兩個(gè)整數(shù)中較大者的函數(shù)：

CREATE FUNCTION pymax (a integer, b integer)

RETURNS integer

AS $$

if a > b:

return a

return b

$$ LANGUAGE plpythonu;

以函數(shù)體形式給出的 Python 轉(zhuǎn)換成 Python 函數(shù)。比如，上面的轉(zhuǎn)換成

def __plpython_procedure_pymax_23456():

if a > b:

return a

return b

這里假設(shè) 23456 是 PostgreSQL 賦予這個(gè)函數(shù)的 OID 。

PostgreSQL 函數(shù)變量可以通過全局的 args 列表獲取。在 pymax 例子里，args[0] 包含第一個(gè)參數(shù)的數(shù)值。args[1] 包含第二個(gè)。除此在外，還可以使用上例中演示的已命名的參數(shù)，這種方法的可讀性更好。

如果向函數(shù)傳遞了一個(gè) NULL 值，參數(shù)值將會(huì)顯示為 None 。上述函數(shù)定義將對 NULL 輸入產(chǎn)生錯(cuò)誤的結(jié)果。我們可以通過添加 STRICT 來進(jìn)行更合理的操作：如果傳入 NULL 值，函數(shù)將根本不會(huì)被調(diào)用，而是立即返回 NULL 。當(dāng)然，也可以在函數(shù)體中檢查輸入?yún)?shù)是否為 NULL ：

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;

如上所示，要從 PL/Python 函數(shù)中返回 NULL ，只要返回 None 即可。無論函數(shù)是否嚴(yán)格，這樣做都有效。

復(fù)合類型參數(shù)將作為 Python 映射進(jìn)行傳遞。映射的元素名是組合類型的元素名。如果被傳遞的行中某個(gè)屬性值為 NULL ，它在映射中的值將是 None 。例如：

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;

有多種從 Python 函數(shù)返回行或者復(fù)合類型的方法。下面的例子假定我們有：

CREATE TYPE named_value AS (

name text,

value integer

);

一個(gè)復(fù)合類型的結(jié)果可以作為下列之一被返回：

序列類型(一個(gè)元組或列表，但不是一個(gè)集合，因?yàn)榧喜豢梢员凰饕?

返回的序列對象中的項(xiàng)數(shù)必須與結(jié)果的復(fù)合類型字段數(shù)相同。索引為 0 的項(xiàng)將被分配給組合類型的第一個(gè)字段，其它依此類推，比如：

CREATE FUNCTION make_pair (name text, value integer)

RETURNS named_value

AS $$

return [ name, value ]

# 或者作為元組返回( name, value )

$$ LANGUAGE plpythonu;

要為任何字段返回 NULL ，就在相應(yīng)的位置插入 None 。

映射(字典)

每個(gè)結(jié)果類型字段的值都重新從映射中使用字段名作為鍵檢索。例如：

CREATE FUNCTION make_pair (name text, value integer)

RETURNS named_value

AS $$

return { "name": name, "value": value }

$$ LANGUAGE plpythonu;

任何額外的字典鍵/值對都將被忽略。丟失的鍵將被當(dāng)作錯(cuò)誤。要為任何字段返回 NULL ，就以相應(yīng)的字段名作為鍵插入 None 。

對象(任何提供 __getattr__ 方法的對象)

它的工作方式和映射一樣。例如：

CREATE FUNCTION make_pair (name text, value integer)

RETURNS named_value

AS $$

class named_value:

def __init__ (self, n, v):

self.name = n

self.value = v

return named_value(name, value)

# 或者簡單地

class nv: pass

nv.name = name

nv.value = value

return nv

$$ LANGUAGE plpythonu;

如果沒有提供返回值，Python 將返回缺省的 None 。PL/Python 將把 None 翻譯成 NULL 。

一個(gè) PL/Python 函數(shù)還可以返回標(biāo)量或組合類型的集合。有多種方法可以達(dá)到這個(gè)目的，因?yàn)榉祷氐膶ο笤趦?nèi)部被轉(zhuǎn)換成一個(gè)迭代器。下面的例子假定我們有一個(gè)復(fù)合類型：

CREATE TYPE greeting AS (

how text,

who text

);

一個(gè)集合結(jié)果可以從下列之一返回：

序列類型(元組、列表、集合)

CREATE FUNCTION greet (how text)

RETURNS SETOF greeting

AS $$

# 返回包含作為符合類型列表的元組

# 所有其它組合也都可以

return ( [ how, "World" ], [ how, "PostgreSQL" ], [ how, "PL/Python" ] )

$$ LANGUAGE plpythonu;

迭代器(任何提供 __iter__ 和 next 方法的對象)

CREATE FUNCTION greet (how text)

RETURNS SETOF greeting

AS $$

class producer:

def __init__ (self, how, who):

self.how = how

self.who = who

self.ndx = -1

def __iter__ (self):

return self

def next (self):

self.ndx += 1

if self.ndx == len(self.who):

raise StopIteration

return ( self.how, self.who[self.ndx] )

return producer(how, [ "World", "PostgreSQL", "PL/Python" ])

$$ LANGUAGE plpythonu;

生成器(yield)

CREATE FUNCTION greet (how text)

RETURNS SETOF greeting

AS $$

for who in [ "World", "PostgreSQL", "PL/Python" ]:

yield ( how, who )

$$ LANGUAGE plpythonu;

警告

當(dāng)前，由于 Python bug #1483133 的原因，一些 Python 2.4 的調(diào)試版本(使用 --with-pydebug 選項(xiàng)編譯)已知會(huì)讓 PostgreSQL 服務(wù)器在使用迭代器返回集合時(shí)崩潰。原始的 Fedora 4 仍然包含這個(gè)臭蟲，但是在生產(chǎn)版本的 Python 或已經(jīng)修補(bǔ)的 Fedora 4 上沒有這個(gè)問題。

全局字典 SD 可以用于在函數(shù)調(diào)用中間存儲(chǔ)數(shù)據(jù)。這個(gè)變量是私有靜態(tài)數(shù)據(jù)。全局字典 SD 是公共數(shù)據(jù)，可以在一個(gè)后端里的所有 Python 函數(shù)之間使用。但在使用時(shí)需要小心。

每個(gè)函數(shù)都在 Python 解釋器里獲得自己的受限制的執(zhí)行對象，因此來自 myfunc 的全局?jǐn)?shù)據(jù)很函數(shù)參數(shù)是 myfunc2 所看不到的。這里的例外是上面提到的 SD 字典。

總結(jié)

以上是生活随笔為你收集整理的plpythonu_PostgreSQL PL/Python 函数的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。