COPY csv file with additionnal datas

user_393984 0
Viewed 144

I have the following table :

persons(id,id_tech,name,nationality,id_list)

And a CSV file containing the datas for the column id_tech,name and nationality.

Importing the data like so is working :

\copy persons(id_tech,name,nationality) FROM '/path/to/file.csv' DELIMITER ',' CSV

The id_list value is not in the CSV file because the file has to be imported on many servers where the value of this id can be different.

Is there a way to import the CSV file while providing an additional value for a specific column ? I can't find anything relevant on the documentation.

Edit--
Note that all my command will be perform with pqxx in C++ (multi plateform). I'm trying to avoid editing the file because of it's size.

Edit 2 --
I'm considering the follwing approch :

  • Create a temp table with a correct default value for the field i need
  • Import File into this temp table
  • Copy tmp to fina l table
  • Remove temp table

But i'm unsure of the performance. Biggest import can be close to a 500K lines.

Thanks

libpqxx postgresql-copy csv postgresql
2 Answers

On Linux you could use for example awk to add the additional value field to your data and use psql to read from from stdin:

$ cat copy.sql
\copy persons(id_tech,name,nationality,extra_col) FROM '/dev/stdin' DELIMITER ',' CSV

and then:

$ awk '
BEGIN {
    FS=OFS=","
}
{
    print $1,$2,$3,"additional value"
}' file.csv | psql -h host -d database -f file.sql

(tested on PostgreSQL 12.4)

user_41623560

Found a solution which seems more than ok.

As stated in my OP , i use libpqxx to insert data , so instead of running a COPY sql request directly , i simply use pqxx::stream_to

Which allow me to add addtional field if needed :

pqxx::stream_to stream(w, mTable, std::vector<std::string>{"id_tech","name","nationality","extra_col"});
csv::CSVReader reader(filePath);
for (csv::CSVRow& row : reader) {

    stream << std::make_tuple(row[0].get<long long>(), row[1].get<std::string>(), row[2].get<std::string>(), custom_id);
}
stream.complete();

Taking around 10s to import 300K lines which is fine for my needs.

user_3939840
Related