Início / dba / Perguntas / 24518
Accepted
dezso
Asked: 2012-09-20 00:32:46 +0800 CST

Como parar condicionalmente um script psql (com base em um valor de variável)?

  • 772

Vamos considerar o seguinte exemplo (do início de um script psql):

\c :db_to_run_on

TRUNCATE the_most_important_table;
-- tried to avoid similarities to anything that exists out there

Agora, se for executado pelo comando

psql [connection details] -v db_to_run_on=\'dev_database\'

então ele simplesmente roda e o usuário fica feliz. Mas e se ele decidir especificar -v db_to_run_on=production_database? (Vamos supor que isso possa acontecer, assim como as pessoas executam rm -rf / # don't try this at home!!!ocasionalmente.) Espero que haja um novo backup dessa tabela ...

Então surge a pergunta: como verificar as variáveis ​​passadas para um script e interromper o processamento posterior com base em seu valor?

postgresql psql

6 respostas

  1. Existe uma opção em psqlque para de executar comandos em caso de erro, é ON_ERROR_STOP. Se pudéssemos gerar um erro de alguma forma, isso faria o que queremos.

    O problema é que temos que testar a variável e produzir um erro de alguma forma. Como não se pode usar estruturas de controle em psql(porque não existem) *, minha única ideia era usar SQL para teste. Bem, produzir um erro condicionalmente é algo em que pl/pgsqlé muito bom, então escrevi uma função que geraria um erro. Agora posso chamar essa função a partir de uma CASEestrutura simples. Um exemplo simples:

    -- let's assume for clarity that there is no function with this name in the database
CREATE OR REPLACE FUNCTION error_generator()
RETURNS boolean AS
$body$
BEGIN
    RAISE 'Meaningful error message here';
    RETURN FALSE; -- just for aesthetical purposes
END;
$body$
LANGUAGE plpgsql;

\set ON_ERROR_STOP on

BEGIN;

-- test for the variable value
-- notice that if :var is not set, it fails as well (with a syntax error)
SELECT CASE WHEN 1 = :var THEN error_generator() ELSE TRUE END;

INSERT INTO test_table (integer_value, text_value)
VALUES (:var, 'something');

COMMIT;

    *: Você pode usar qualquer comando shell depois \!e condicionais do shell, mas como \!abre um novo shell, executar qualquer coisa lá não tem nenhum efeito para o script psql atual.

    • 17
  2. Best Answer

    PostgreSQL 10

    PostgreSQL 10 traz condicionais para psql. Isso não é mais um problema.

    \if :db_to_run_on = 'dev_database'
  TRUNCATE the_most_important_table;
\endif

    Eu acho que você também poderia usar DO..

    \if :db_to_run_on != 'dev_database'
do $$
  BEGIN
    RAISE 'Meaningful error message here';
  END;
$$ LANGUAGE plpgsql;
\endif
    • 8

  3. Uma versão mais concisa da resposta de dezso:

    CREATE OR REPLACE FUNCTION pg_temp.err(msg varchar) RETURNS boolean     
AS $$ BEGIN RAISE '%',msg; END; $$ LANGUAGE plpgsql;

    Você pode então chamar isso como:

    \set ON_ERROR_STOP on

SELECT CASE WHEN (
  SELECT COUNT(*) FROM mytable
) > 0 THEN pg_temp.err('Already loaded') END;
    • 2

  4. O que descobri que funciona muito bem para mim é usar uma linguagem de script para gerar um arquivo SQL que eu canalizo para o psql, algo assim:

    #!/usr/bin/env ruby

raise "Not a good database name: #{ARGV.first.inspect}" unless ARGV.first =~ /^(dev|test)/

puts "\\timing off"
puts "set client_min_messages='warning';"
puts
puts "TRUNCATE the_most_important_table;"
puts "-- more commands"

    Então, eu chamo isso de um script de driver:

    #!/bin/bash
/usr/bin/ruby generator ${1} | /usr/bin/psql --dbname=${1} --file=- --single-transaction

    Meu script de driver geralmente é um arquivo Rake, mas você entendeu.

    • 1

  5. Para sair psqlde um script, você pode simplesmente usar o \quitcomando meta ou gerar um erro enquanto estiver ON_ERROR_STOPativo. As diferenças são:

    • \quitfará uma saída limpa com status 0ou encerrará apenas o script em execução se estiver usando scripts aninhados ou\i
    • produzir um erro ao ON_ERROR_STOPser ativado sairá com status 3e se propagará para o script de nível superior no caso de scripts aninhados.

    Uma alternativa mais fácil para gerar um erro e executar RAISEa partir de uma função ou bloco de código anônimo é apenas um comando SQL inválido como: INVALID;.

    Para exibir a mensagem de erro, você pode usar o \warnmeta-comando que a envia para stderr(começando na página 13) ou simplesmente \echopara stdout.

    Combinado com \if(começando na página 10), minha tentativa para o problema do OP seria:

    SELECT :'db_to_run_on' = 'production_database' AS forbidden \gset
\if :forbidden
    \warn 'You better not do this!'
    \set ON_ERROR_STOP on
    INVALID COMMAND;
\endif
\unset forbidden

\c :db_to_run_on
TRUNCATE the_most_important_table;

    Infelizmente, não conheço uma maneira de suprimir a saída de 'erro de sintaxe' gerada pelo comando inválido. Portanto, dependendo do seu caso de uso, combinar \warn 'You better not do this!'e \quitou \set ON_ERROR_STOP one DO $$BEGIN RAISE 'You better not do this!'; END$$ LANGUAGE plpgsql;pode ser mais adequado.

    • 0

  6. Em um script de consulta (que eu acho que é de alguma forma diferente de um script psql), \iffoi considerado um erro de sintaxe, mas um código como este funcionou para mim:

    DO $$ BEGIN
    IF problematic_value = 'undesired thing' THEN
        RAISE EXCEPTION 'Aborting: % is no good', problematic_value;
    END IF;
END $$;

    Testado no PostgreSQL 13.2. DOO bloco externo não é necessário dentro de um procedimento armazenado.

    • 0

relate perguntas