I'm working on a database migration script written in python, which will take data from a MySQL database and insert them into a PostgreSQL database with a different Schema (different table structures, different datatypes and so on).
Performances does matter, since I will have to handle with sizable databases.
I use mysql.connector
and psycopg2
adapters in order to make python talks with the two databases.
My problem is that I need performance, but also the power of modifying/converting mysql data before inserting rows in the new fresh database. I will give you an example in order to show why there actually is a conflict of interest.
Following is a snippet of code without performance optimization, but at least, where transformation/conversion of data was possible:
cur_msql.execute("SELECT customer_id, customer_name, contact_name, address, city, postal_code, country FROM mysqlcustomers")
for row in cur_msql:
# /!\ checking if customer_name is NULL (NOT NULL field in destination database)
if row['customer_name'] == NULL:
row['customer_name'] = row['contact_name']
try:
cur_psql.execute("INSERT INTO psqlcustomers (customer_id, customer_name, contact_name, address, city, postal_code, country) \
VALUES (%(customer_id)s, %(customer_name)s, %(contact_name)s, %(address)s, %(city)s, %(postal_code)s, %(country)s)", row)
except psycopg2.Error as e:
print "cannot execute that query", e.pgerror
sys.exit("leaving early the script")
As you can see here I have the possibility to make an if
statement for each row retrieved from mysql, to e.g. put the value of a field inside another field (or casting field datatype and so on...). Which is something that I really need to do in my scenario (elaboration of db row).
Meanwhile, I have found out that performing many insert statement as above (each mysql row is inserted separately) is killing the performance.
So I'm thinking to switch to something like that, which prepares all the blocks of statements, and then insert it all in one block:
cur_msql.execute("SELECT customer_id, customer_name, contact_name, address, city, postal_code, country FROM mysqlcustomers")
args_str = ','.join(cur_psql.mogrify("(%(customer_id)s, %(customer_name)s, %(contact_name)s, %(address)s, %(city)s, %(postal_code)s, %(country)s)", x) for x in cur_msql)
try:
cur_psql.execute("INSERT INTO psqlcustomers (customer_id, customer_name, contact_name, address, city, postal_code, country) \
VALUES " + args_str)
except psycopg2.Error as e:
print "cannot execute that query", e.pgerror
sys.exit("leaving early the script")
With this approach the script can be about 100x faster! But the problem with this is, that I can no more elaborate the data (transformation/conversion/casting...) because all the rows get inserted into an object and I'm not able to iterate through it anymore...
Is it possible to approach the problem in a way where I can:
- Not kill the performance
- Still elaborate/transform the data of each row?