From 5ccb80491346bab81414f74cd8297285083a5d8f Mon Sep 17 00:00:00 2001 From: Helmut Grohne Date: Sat, 8 Mar 2014 13:30:20 +0100 Subject: enable result buffering for postgres --- dedup/utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'dedup') diff --git a/dedup/utils.py b/dedup/utils.py index e163a00..1452edc 100644 --- a/dedup/utils.py +++ b/dedup/utils.py @@ -22,3 +22,13 @@ def sqlite_add_version_compare(engine): def add_version_compare(connection, _): connection.create_collation("debian_version", version_compare) connection.create_function("debian_version_compare", 2, version_compare) + +def postgres_enable_result_streaming(engine): + if engine.dialect.driver != "psycopg2": + return + engine.server_side_cursors = True + +def configure_database_engine(engine): + enable_sqlite_foreign_keys(engine) + sqlite_add_version_compare(engine) + postgres_enable_result_streaming(engine) -- cgit v1.2.3