Aşağıdaki örneği ele alalım (psql betiğinin başından itibaren):

\c :db_to_run_on

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

Şimdi bunu komut tarafından çalıştırılıyorsa

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

sonra sadece çalışır ve kullanıcı mutlu. Peki ya belirtmeye karar verirse -v db_to_run_on=production_database? (Diyelim ki, tıpkı insanlar zaman zaman koşuyor gibi rm -rf / # don't try this at home!!!.) Umarım bu tablonun yeni bir yedeği vardır ...

Böylece soru ortaya çıkıyor: bir betiğe geçirilen değişkenler nasıl kontrol edilir ve değerlerine göre daha fazla işlem nasıl durdurulur?

psqlHata halinde komutları çalıştırmayı durduran bir seçenek var , bu ON_ERROR_STOP. Bir şekilde bir hata oluşturabilirsek, bu istediğimizi yapar.

Sorun, değişkeni test etmemiz ve bir şekilde bir hata üretmemiz gerektiğidir. Biri kontrol yapılarını kullanamadığı için psql(hiçbiri olmadığı için) *, tek fikrim test için SQL kullanmaktı. Şartlı olarak bir hata üretmek pl/pgsqloldukça iyi bir şeydir , bu yüzden hata oluşturacak bir fonksiyon yazdım. Şimdi bu işlevi basit bir CASEyapıdan çağırabilirim . Basit bir örnek:

-- 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;

*: Kabuktan sonraki herhangi bir kabuk komutunu \!ve koşulunu kullanabilirsiniz , ancak \!yeni bir kabuk açtığından, herhangi bir şey yürütmenin geçerli psql betiği için herhangi bir etkisi yoktur.

PostgreSQL 10

PostgreSQL 10, psql'ye koşul getiriyor. Bu artık sorun değil.

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

Sanırım sen de kullanabilirsin DO..

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

Bulduğum şey benim için çok iyi çalışıyor, daha sonra psql içine boru gibi bir SQL dosyası oluşturmak için bir komut dosyası dili kullanmaktır:

#!/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"

Sonra, bunu bir sürücü komut dosyasından çağırıyorum:

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

Sürücü betiğim genellikle bir Rake dosyasıdır, ancak fikri anlarsınız.

Dezso'nun cevabının daha özlü bir versiyonu:

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

Daha sonra bunu şöyle çağırabilirsiniz:

\set ON_ERROR_STOP on

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