How to fake an array of a user-defined type

From: Richard Plotkin <richard(at)richardplotkin(dot)com>
To: pgsql-general(at)postgresql(dot)org
Subject: How to fake an array of a user-defined type
Date: 2005-02-10 14:07:23
Message-ID: 664b06e68f925ffafbeed91ea5790da5@richardplotkin.com
Views: Raw Message | Whole Thread | Download mbox | Resend email
Thread:
Lists: pgsql-general

I've been reading some posts on this list, trying to figure out how to
send a function an array of a user-defined type. I have figured out a
way to do this. To some extent, this serves as a follow-up to
"User-defined type arrays?"
(http://archives.postgresql.org/pgsql-novice/2004-10/msg00132.php)

The general idea is to have a user-defined type paired with a
user-defined cast and user-defined cast conversion function. The
typecasting will change type text[] to your user-defined type, and
since type text[][] is allowed, you can pass text[][] to a function and
then typecast text[1]..text[n] as your user-defined type.

Hope this saves someone some time.

First, assume a created type
CREATE TYPE mytype AS (
attribute1 text,
attribute2 integer
);

Then, assume a failed function (which won't work because mytype[] will
not be accepted as a function parameter)
CREATE OR REPLACE FUNCTION myfunc(mytype[]) RETURNS null AS $$
DECLARE
mytype_array ALIAS FOR $1

BEGIN

RETURN null;

END;
$$ LANGUAGE 'plpgsql';

Now add the following:
CREATE OR REPLACE FUNCTION to_mytype(text[]) RETURNS mytype AS
DECLARE
input ALIAS FOR $1
result mytype;

BEGIN
--create a row using same types that are assigned in mytype
result = ROW(input[1]::text, input[2]::integer);
RETURN result;
END;
$$ LANGUAGE 'plpgsql';

CREATE CAST (text[] as mytype)
WITH FUNCTION to_mytype(text[]);

Now, change your function to the following, where text[][] is an array
of mytype's that will initially be parsed as text's (so type text[][]
is what you'll pass, and type mytype[] is what you'll end up dealing
with. Treat each mytype as a text[])

CREATE OR REPLACE FUNCTION myfunc(text[][]) RETURNS null AS $$
DECLARE
mytype_array ALIAS FOR $1;
mytype_element mytype;
arr_u int;
arr_l int;

BEGIN
--you should loop through all array-type elements of your input array
(text[][]) to pull them all out as mytype's
--and then, once you've pulled them out as mytype's, deal with them
immediately (you can't convert them and array_append into mytype[])
SELECT INTO arr_u array_upper(mytype_array);
SELECT INTO arr_l array_lower(mytype_array);

FOR i IN arr_l..arr_u LOOP
SELECT INTO mytype_element
CAST(mytype_array[i] AS mytype);
--anything else you want to do
END LOOP;
END;
$$ LANGUAGE 'plpgsql';

Browse pgsql-general by date

  From Date Subject
Next Message Tom Lane 2005-02-10 14:53:49 Re: a SELECT FOR UPDATE question
Previous Message Christopher Browne 2005-02-10 13:33:33 Re: Functions with more than 32 parameters