php postgresql pdo копия со стандартного ввода - PullRequest
Новая
       8

php postgresql pdo копия со стандартного ввода

6 голосов
/ 13 февраля 2012 
COPY table_name ( field1, field2, field3) FROM STDIN CSV;
1,2,"q w"
3,4,"a s"
5,6,d
\.

Как выполнить этот запрос с помощью PDO?

Обновление:

Проблема в том, что драйвер PDO выполняет этот запрос как оператор.
Например, если вы вставите его в pgAdmin, он выдаст ошибку.
Мне нужно выполнить это в psql: 

C:\Users\User>psql -e -h localhost -U postgres db_name
psql (9.1.2)
db_name=# COPY table_name ( field1, field2, field3) FROM STDIN CSV;
COPY table_name ( field1, field2, field3) FROM STDIN CSV;
Enter data to be copied followed by a newline.
End with a backslash and a period on a line by itself.
>> 1,2,"q w"
>> 3,4,"a s"
>> 5,6,d
>> \.

Ответы [ 2 ]

5 голосов
/ 17 февраля 2012

Благодаря этой книге

Обратите внимание, что представленная здесь функция эффективно обходит ограничения безопасности, которые существуют по причине.Ваша функция должна проверять путь к файлу и таблицу в соответствии со строгими условиями белого списка.Этот пример также открыт для SQL-инъекций, так как он не корректно вводит свои входные данные.

Создание функции, которая выполняет COPY команду 

CREATE OR REPLACE FUNCTION copy_from_csv_ignoring_security(table_name text, table_fieds text, file_path text, oids boolean DEFAULT false, header boolean DEFAULT false, delimeter text DEFAULT ','::text, "null" text DEFAULT ''::text, quote text DEFAULT '"'::text, escape text DEFAULT '"'::text, force_not_null text DEFAULT ''::text)
  RETURNS void AS
$BODY$

declare statement text;
begin

statement := 'COPY ' || table_name || ' (' || table_fieds || ') ' || 'FROM ''' || file_path || ''' WITH ';
IF oids THEN
 statement := statement || 'OIDS ';
end if;
statement := statement || 'DELIMITER ''' || delimeter || ''' ';
statement := statement || 'NULL ''' || "null" || ''' CSV ';
IF header THEN
 statement := statement || 'HEADER ';
end if;
statement := statement || 'QUOTE ''' || "quote" || ''' ';
statement := statement || 'ESCAPE ''' || "escape" || ''' ';
IF force_not_null <> '' THEN
statement := statement || 'FORCE NOT NULL ''' || force_not_null || ''' ';
end if;
execute statement;
end;
$BODY$
  LANGUAGE plpgsql VOLATILE SECURITY DEFINER
  COST 100;

Предоставление прав на функцию 

revoke all on function copy_from_csv_ignoring_security(text, text, text, boolean, boolean, text, text, text, text, text) from public;
grant execute on function copy_from_csv_ignoring_security(text, text, text, boolean, boolean, text, text, text, text, text) to db_user;

Выполнить из PHP 

$dbh->exec('SELECT copy_from_csv_ignoring_security(...)');

===== Если версия> = 9.1.7, трюк выше не работает.=====

Решение:

создать файл .pgpass (избегать запроса пароля) в домашнем каталоге пользователя, который запускает этот скрипт.

#.pgpass contents (chmod 600 - requred)    
host:port:db_name:user_name:password

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

function executeMetaCommand($dbUser, $dbName, $dbPort, $command)
{    
    $command = sprintf(
        "psql -U %s -p %s -d %s -f - <<EOT\n%s\nEOT\n",
        $dbUser, $dbPort, $dbName, $command
    );
    $streams = array(
        array('pipe', 'r'),// stdin
        array('pipe', 'w'),// stdout
        array('pipe', 'w') // stderr
    );
    $process = proc_open($command, $streams, $pipes);
    if (!is_resource($process)) {
        throw new Exception("Cannot open process:\n$command");
    } else {
        list(, $stdout, $stderr) = $pipes;
        $error = stream_get_contents($stderr);
        fclose($stderr);
        if (strlen($error) > 0) {
            throw new Exception("Process error:\n$error");
        } else {
            $output = stream_get_contents($stdout);
            fclose($stdout);
            $returnCode = proc_close($process);
            if ($returnCode === -1) {
                throw new Exception("Process was completed incorrectly:\n$output");
            } else {
                return array(
                    $returnCode,
                    $output
                );
            }
        }
    }
}

//usage:
$command = sprintf("\\copy table(field1, field2) FROM '%s' WITH CSV", $filePath);
executeMetaCommand('postgres', 'test_db', '5432', $command);
0 голосов
/ 13 февраля 2012

Просто поместите строки ввода в CSV-файл, скажем input.csv, а затем попробуйте следующее: 

<?php
$dbh = new PDO($dsn,$username,$pw);

$num_inserted=$dbh->exec("\copy table_name(field1,field2,field3) from /path/to/input.csv");

if($num_inserted===false)
{
  die("Unable to insert rows into table_name from input.csv\n");
}

echo("Inserted $num_inserted rows.\n");
?>
...