PL/pgSQL CREATE или REPLACE в рамках EXECUTE

У меня есть следующий скрипт для динамического создания представлений в базе данных PostgreSQL.

CREATE OR REPLACE FUNCTION cs_refresh_mviews() RETURNS integer AS $$
DECLARE
    mviews RECORD;
    query text;
    park_name text;
    ppstatements int;
BEGIN
    RAISE NOTICE 'Creating views...';

    FOR mviews IN SELECT name FROM "Canadian_Parks" LOOP
        park_name := mviews.name;
        RAISE NOTICE 'Creating or replace view %s...', mviews.name; 
        query := 'CREATE OR REPLACE VIEW %_view AS
          SELECT * from "Canadian_Parks" where name=''%'';
          ALTER TABLE %_view OWNER TO postgres', park_name, park_name, park_name;
      --  RAISE NOTICE query;
        EXECUTE query;
    END LOOP;

    RAISE NOTICE 'Done refreshing materialized views.';
    RETURN 1;
END;
$$ LANGUAGE plpgsql;

Я подтвердил целостность строки, например

CREATE OR REPLACE VIEW Saguenay_St__Lawrence_view AS
SELECT * from "Canadian_Parks" where name='Saguenay_St__Lawrence';
ALTER TABLE Saguenay_St__Lawrence_view OWNER TO postgres

присваивается переменной запроса, вручную отправляя ее в базу данных и получая успешный ответ.

Однако, если я попытаюсь выполнить функцию, используя

SELECT cs_refresh_mviews();

отображается следующая ошибка:

ERROR:  query "SELECT 'CREATE OR REPLACE VIEW %_view AS SELECT * from "Canadian_Parks" where name=''%''; ALTER TABLE %_view OWNER TO postgres', park_name, park_name, park_name" returned 4 columns
CONTEXT:  PL/pgSQL function "cs_refresh_mviews" line 32 at assignment

********** Error **********

ERROR: query "SELECT 'CREATE OR REPLACE VIEW %_view AS SELECT * from "Canadian_Parks" where name=''%''; ALTER TABLE %_view OWNER TO postgres', park_name, park_name, park_name" returned 4 columns
SQL state: 42601
Context: PL/pgSQL function "cs_refresh_mviews" line 32 at assignment

Почему это было преобразовано в оператор SELECT, а не в чистый CREATE?


postgresql dynamic-sql materialized-views plpgsql
person Klaus Nji    schedule 09.10.2012    source источник

Ответы (2)


arrow_upward
4
arrow_downward

У вас довольно запутанная установка. Зачем сохранять часть имени представления в составном типе таблицы, а не в текстовом столбце?

Во всяком случае, это может работать так:

Установите соответствующий вопрос:

CREATE SCHEMA x;  -- demo in test schema
SET search_path = x;
CREATE TYPE mviews AS (id int, name text); -- composite type used in table

CREATE TABLE "Canadian_Parks" (name mviews);
INSERT INTO "Canadian_Parks"(name) VALUES
 ('(1,"canadian")')
,('(2,"islandic")');  -- composite types, seriously?

SELECT name, (name).* from "Canadian_Parks";

CREATE OR REPLACE FUNCTION cs_refresh_mviews()
  RETURNS int LANGUAGE plpgsql SET search_path = x AS  -- search_path for test
$func$
DECLARE
    _parkname text;
BEGIN

FOR _parkname IN SELECT (name).name FROM "Canadian_Parks" LOOP
   EXECUTE format('
      CREATE OR REPLACE VIEW %1$I AS
      SELECT * FROM "Canadian_Parks" WHERE (name).name = %2$L;
      ALTER TABLE %1$I OWNER TO postgres'
      , _parkname || '_view', _parkname);
END LOOP;

RETURN 1;

END
$func$;

SELECT cs_refresh_mviews();

DROP SCHEMA x CASCADE; -- clean up

Основные моменты

  • Поскольку вы выполняете текст с помощью execute, вам необходимо защититься от инъекций SQL. Я использую функцию format() для < em>идентификаторы и литерал

  • Я использую синтаксис SELECT (name).name, чтобы справиться с вашей странной настройкой и сразу извлечь name, который нам нужен.

  • Точно так же VIEW необходимо прочитать WHERE (name).name = .., чтобы работать в этой настройке.

  • Я удалил много шума, который не имеет отношения к вопросу.

  • Также, вероятно, бессмысленно иметь функцию RETURN 1. Просто определите функцию с помощью RETURNS void. Однако я сохранил его, чтобы соответствовать вопросу.

Распутанная установка

Как, наверное, должно быть:

CREATE SCHEMA x;
SET search_path = x;

CREATE TABLE canadian_parks (id serial primary key, name text);
INSERT INTO canadian_parks(name) VALUES ('canadian'), ('islandic');

SELECT * from canadian_parks;

CREATE OR REPLACE FUNCTION cs_refresh_mviews()
  RETURNS void LANGUAGE plpgsql SET search_path = x AS
$func$
DECLARE
    parkname text;
BEGIN

FOR parkname IN SELECT name FROM canadian_parks LOOP
   EXECUTE format('
      CREATE OR REPLACE VIEW %1$I AS
      SELECT * FROM canadian_parks WHERE name = %2$L;
      ALTER TABLE %1$I OWNER TO postgres'
      , parkname || '_view', parkname);
END LOOP;

END
$func$;

SELECT cs_refresh_mviews();

DROP SCHEMA x CASCADE;
person Erwin Brandstetter    schedule 10.10.2012

arrow_upward
2
arrow_downward

Вы неправильно поняли использование запятых в выражении присваивания. Он превращает query в массив (RECORD) вместо скаляра. Используйте конкатенацию:

park_name := quote_ident(mviews.name||'_view');
query := 'CREATE OR REPLACE VIEW '||park_name||' AS SELECT * from "Canadian_Parks" where name='||quote_literal(mviews.name)||'; ALTER TABLE '||park_name||' OWNER TO postgres';
person SadSamurai    schedule 09.10.2012