Patch to add support for text/int arrays for plpythonu

From: "Valtonen, Hannu" <hannu(dot)valtonen(at)hut(dot)fi>
To: pgsql-hackers(at)postgresql(dot)org
Subject: Patch to add support for text/int arrays for plpythonu
Date: 2009-04-26 10:00:54
Message-ID: 49F430D6.4050307@hut.fi
Views: Raw Message | Whole Thread | Download mbox | Resend email
Thread:
Lists: pgsql-hackers

Hi,

Attached is a small patch for adding support for INT[] and TEXT[] to
plpythonu.
There are also some small tests for this copypasted to the end of the mail.

I'd like to get this into CVS but as it's probably a bit late for 8.4 I
wouldn't mind it going into the first commitfest for 8.5.

- Hannu

---SELECT test_int_array[3] FROM return_int_array(ARRAY[1,2,3]) AS
test_int_array;
CREATE OR REPLACE FUNCTION return_int_array(int_array INT[])
RETURNS int[]
AS $$
plpy.info("Type: %s %r" % (type(int_array), int_array))
return int_array
$$ LANGUAGE plpythonu;

---SELECT jee[3] FROM return_text_array(ARRAY['test1','test2','test3'])
AS test_text_array;
CREATE OR REPLACE FUNCTION return_text_array(text_array TEXT[])
RETURNS TEXT[]
AS $$
plpy.info("Type: %s %r" % (type(text_array), text_array))
return text_array
$$ LANGUAGE plpythonu;

--CREATE TABLE test_table (id SERIAL PRIMARY KEY, bar INT[], bar2 TEXT[]);
--SELECT * FROM test_text_array_plan(ARRAY['test1', 'test2']);
CREATE OR REPLACE FUNCTION test_text_array_plan(test_array TEXT[])
RETURNS void
AS $$
plan = plpy.prepare("INSERT INTO test_table (bar) VALUES ($1)", ["TEXT[]"])
results = plpy.execute(plan, [test_array])
$$ LANGUAGE plpythonu;

--SELECT * FROM test_int_array_plan(ARRAY[1,2,3,4,5,6,7,8,9,10]);
CREATE OR REPLACE FUNCTION test_int_array_plan(test_array INT[])
RETURNS void
AS $$
plan = plpy.prepare("INSERT INTO test_table (bar2) VALUES ($1)", ["INT[]"])
results = plpy.execute(plan, [test_array])
$$ LANGUAGE plpythonu;

Attachment Content-Type Size
plpythonarray.diff text/plain 6.0 KB

Responses

Browse pgsql-hackers by date

  From Date Subject
Next Message Sam Halliday 2009-04-26 10:54:55 Re: RFE: Transparent encryption on all fields
Previous Message Magnus Hagander 2009-04-26 06:51:59 Re: Randomly placed definition of S_ISDIR() in psql/copy.c