PostgreSQL: Przekaż tabelę jako argument w funkcji

11

Ja odkrywając TYPEw PostgreSQL. Mam coś, TABLE TYPEco niektóre tabele muszą respektować (interfejs). Na przykład:

CREATE TYPE dataset AS(
    ChannelId INTEGER
   ,GranulityIdIn INTEGER
   ,GranulityId INTEGER
   ,TimeValue TIMESTAMP
   ,FloatValue FLOAT
   ,Status BIGINT
   ,QualityCodeId INTEGER
   ,DataArray FLOAT[]
   ,DataCount BIGINT
   ,Performance FLOAT
   ,StepCount INTEGER
   ,TableRegClass regclass
   ,Tags TEXT[]
   ,WeightedMean FLOAT
   ,MeanData FLOAT
   ,StdData FLOAT
   ,MinData FLOAT
   ,MaxData FLOAT
   ,MedianData FLOAT
   ,Percentiles FLOAT[]
);

Mogę utworzyć tabelę za pomocą tego szablonu z:

CREATE TABLE test OF dataset;

Widziałem wiele opcji w interfejsie API , ale jestem trochę zagubiony. Chciałbym wiedzieć, czy możliwe jest przypisanie tego typu do INPUT/OUTPUTparametrów funkcji .

Powiedzmy, że mam FUNCTIONwywołanie, processktóre pobiera próbkę rekordów z zestawu danych TABLE source, przetwarza je, a następnie zwraca to TABLE sinksamo TYPE.

Chciałbym wiedzieć, czy można stworzyć taki, TYPEktóry zachowuje się tak:

CREATE FUNCTION process(
    input dataset
) RETURNS dataset
AS ...

I można to tak nazwać:

SELECT
    *
FROM
    source, process(input := source) AS sink;

Zastanawiam się, czy jest to możliwe dzięki PostgreSQL i pytam, jak to zrobić. Czy ktoś z was wie?


Oto MWE tego, co próbuję zrobić:

DROP TABLE IF EXISTS source;
DROP FUNCTION IF EXISTS process(dataset);
DROP TYPE dataset;

CREATE TYPE dataset AS (
    id INTEGER
   ,t  TIMESTAMP
   ,x  FLOAT
);


CREATE TABLE source OF dataset;
ALTER TABLE source ADD PRIMARY KEY(Id);
INSERT INTO source VALUES
    (1, '2016-01-01 00:00:00', 10.0)
   ,(2, '2016-01-01 00:30:00', 11.0)
   ,(3, '2016-01-01 01:00:00', 12.0)
   ,(4, '2016-01-01 01:30:00',  9.0)
   ;

CREATE OR REPLACE FUNCTION process(
    _source dataset
)
RETURNS SETOF dataset
AS
$BODY$
SELECT * FROM source;
$BODY$
LANGUAGE SQL;

SELECT * FROM process(source);

Ale to się nie udaje, to tak, jakby źródło było postrzegane jako kolumna zamiast SETOF RECORDSz typem zestawu danych.

jlandercy
źródło

Odpowiedzi:

13

Twój parametr _source w dodanym MWE nie jest nigdzie wymieniony. Identyfikator sourcew ciele funkcji nie ma wiodącego podkreślenia i jest niezależnie interpretowany jako stała nazwa tabeli.

Co ważniejsze, i tak by to nie działało. SQL pozwala jedynie na parametryzację wartości w instrukcjach DML. Szczegóły w tej powiązanej odpowiedzi:

Rozwiązanie

Nadal możesz sprawić, aby działał za pomocą dynamicznego SQL z EXECUTEfunkcją plpgsql. Detale:

Lub spróbuj wyszukać podobne pytania i odpowiedzi

CREATE TYPE dataset AS (id integer, t timestamp, x float);
CREATE TABLE source OF dataset (PRIMARY KEY(Id));  -- add constraints in same command

INSERT INTO source VALUES
    (1, '2016-01-01 00:00:00', 10.0)
   ,(2, '2016-01-01 00:30:00', 11.0);

CREATE OR REPLACE FUNCTION process(_tbl regclass)
  RETURNS SETOF dataset AS
$func$
BEGIN
RETURN QUERY EXECUTE 'SELECT * FROM ' || _tbl;
END
$func$  LANGUAGE plpgsql;

SELECT * FROM process('source');  -- table name as string literal 

Możesz nawet sprawić, aby działało to dla dowolnej tabeli:

CREATE OR REPLACE FUNCTION process2(_tbl anyelement)
  RETURNS SETOF anyelement AS
$func$
BEGIN
RETURN QUERY EXECUTE 'SELECT * FROM ' || pg_typeof(_tbl);
END
$func$  LANGUAGE plpgsql;

SELECT * FROM process2(NULL::source);  -- note the call syntax!!

Szczegółowe wyjaśnienie:

Erwin Brandstetter
źródło
Dzięki za odpowiedź. Sprawdzę to za kilka godzin. Wystarczy wiedzieć przed testem, czy Twoje rozwiązanie akceptuje otrzymywanie wierszy od as SELECT. Mam na myśli SELECT * FROM process((SELECT * FROM source WHERE cond)).
jlandercy
@ j: Nie, podajesz nazwę tabeli . Nie ma możliwości przekazania samej tabeli (bez zmiennej tabeli). Jest na to kilka sposobów. Powiązane: stackoverflow.com/a/27853965/939860 lub stackoverflow.com/a/31167928/939860 . Do pracy nad wynikiem zapytania użyłbym kursora lub tabeli tymczasowej ...
Erwin Brandstetter
0

To zrobi, co chcesz bez potrzeby dynamicznego SQL :

drop table if exists source cascade;
drop function if exists process(dataset) cascade;
drop type if exists dataset cascade;

create type dataset as (
    id integer
   ,t  timestamp
   ,x  float
);

create table source of dataset;
alter table source add primary key(id);
insert into source values
   (1, '2016-01-01 00:00:00', 10.0)
 , (2, '2016-01-01 00:30:00', 11.0)
;

create or replace function process(
    x_source dataset[]
)
returns setof dataset
as
$body$
select * from unnest(x_source);
$body$
language sql;

select *
from
  process(
    array(
      select
        row(id, t, x)::dataset
      from source
    )
  );

O ile mogę powiedzieć (po obszernym googlowaniu, ponieważ miałem ten sam problem), nie można przekazać tabeli bezpośrednio do funkcji.

Jednak, jak pokazano, możesz przekształcić tabelę w tablicę []typu niestandardowego, która składa się z kilku podstawowych typów (podobnych do definicji tabeli).

Następnie możesz przekazać tę tablicę i odkleić ją z powrotem do tabeli, gdy znajdziesz się w funkcji.

Sam Fed
źródło