# HG changeset patch
# User Julien Cristau <julien.cristau@logilab.fr>
# Date 1391522785 -3600
# Node ID e45bf9baa7b73f8eb783da40caf79ac06ee3faef
# Parent  927d035b6921cab54b397f3556ce9a337a5546d7
Add a db-statement-timeout option for postgresql sources

Use that new option to set the statement_timeout option when creating
connections, except for maintenance operations (shell and upgrade).

Closes #2547026

diff -r 927d035b6921 -r e45bf9baa7b7 server/sources/native.py
--- a/server/sources/native.py	Tue Jan 07 15:10:57 2014 +0100
+++ b/server/sources/native.py	Tue Feb 04 15:06:25 2014 +0100
@@ -323,10 +323,16 @@
                   'want trusted authentication for the database connection',
           'group': 'native-source', 'level': 2,
           }),
+        ('db-statement-timeout',
+         {'type': 'int',
+          'default': 0,
+          'help': 'sql statement timeout, in milliseconds (postgres only)',
+          'group': 'native-source', 'level': 2,
+          }),
     )
 
     def __init__(self, repo, source_config, *args, **kwargs):
-        SQLAdapterMixIn.__init__(self, source_config)
+        SQLAdapterMixIn.__init__(self, source_config, repairing=repo.config.repairing)
         self.authentifiers = [LoginPasswordAuthentifier(self)]
         if repo.config['allow-email-login']:
             self.authentifiers.insert(0, EmailPasswordAuthentifier(self))
diff -r 927d035b6921 -r e45bf9baa7b7 server/sqlutils.py
--- a/server/sqlutils.py	Tue Jan 07 15:10:57 2014 +0100
+++ b/server/sqlutils.py	Tue Feb 04 15:06:25 2014 +0100
@@ -299,7 +299,7 @@
     """
     cnx_wrap = ConnectionWrapper
 
-    def __init__(self, source_config):
+    def __init__(self, source_config, repairing=False):
         try:
             self.dbdriver = source_config['db-driver'].lower()
             dbname = source_config['db-name']
@@ -328,6 +328,14 @@
         if self.dbdriver == 'sqlite':
             self.cnx_wrap = SqliteConnectionWrapper
             self.dbhelper.dbname = abspath(self.dbhelper.dbname)
+        if not repairing:
+            statement_timeout = int(source_config.get('db-statement-timeout', 0))
+            if statement_timeout > 0:
+                def set_postgres_timeout(cnx):
+                    cnx.cursor().execute('SET statement_timeout to %d' % statement_timeout)
+                    cnx.commit()
+                postgres_hooks = SQL_CONNECT_HOOKS['postgres']
+                postgres_hooks.append(set_postgres_timeout)
 
     def wrapped_connection(self):
         """open and return a connection to the database, wrapped into a class