Add ability to set reconnect flag on MySQL connections

Add database server ping method to SQL classes

2 files changed, 24 insertions, 0 deletions

diff --git a/src/sql/drivers/psql/qsql_psql.cpp b/src/sql/drivers/psql/qsql_psql.cpp
index 7fe1a91..9d028c0 100644
--- a/src/sql/drivers/psql/qsql_psql.cpp
+++ b/src/sql/drivers/psql/qsql_psql.cpp
@@ -679,6 +679,29 @@ void QPSQLDriver::close()
     }
 }
 
+bool QPSQLDriver::ping()
+{
+    if ( !isOpen() ) {
+        return FALSE;
+    }
+
+    PGresult *res = NULL;
+
+    // Send ping
+    res = PQexec( d->connection, "" );
+    PQclear(res);
+
+    // Check connection status
+    if ( PQstatus( d->connection ) != CONNECTION_OK ) {
+	PQreset( d->connection );
+	if ( PQstatus( d->connection ) != CONNECTION_OK ) {
+	    setLastError( qMakeError("Unable to execute ping", QSqlError::Statement, d ) );
+	    return FALSE;
+	}
+    }
+    return TRUE;
+}
+
 QSqlQuery QPSQLDriver::createQuery() const
 {
     return QSqlQuery( new QPSQLResult( this, d ) );
diff --git a/src/sql/drivers/psql/qsql_psql.h b/src/sql/drivers/psql/qsql_psql.h
index fd9cbe6..b0e4f8c 100644
--- a/src/sql/drivers/psql/qsql_psql.h
+++ b/src/sql/drivers/psql/qsql_psql.h
@@ -98,6 +98,7 @@ public:
 				const QString & host = QString::null,
 				int port = -1 );
     void		close();
+    bool		ping();
     QSqlQuery		createQuery() const;
     QStringList		tables( const QString& user ) const;
     QSqlIndex		primaryIndex( const QString& tablename ) const;