Initial import of CVS-v2_8_branch

2025-11-26 15:20:29 +01:00 · 2014-08-12 15:13:02 -04:00
commit afd67c7af8
1508 changed files with 405304 additions and 0 deletions
--- a/lib/rddb.cpp
+++ b/lib/rddb.cpp
@@ -0,0 +1,149 @@
+// rddb.cpp
+//
+//   Database driver with automatic reconnect
+//
+//   (C) Copyright 2007 Dan Mills <dmills@exponent.myzen.co.uk>
+//
+//      $Id: rddb.cpp,v 1.13.4.2.2.2 2014/03/21 15:41:45 cvs Exp $
+//
+//   This program is free software; you can redistribute it and/or modify
+//   it under the terms of the GNU General Public License version 2 as
+//   published by the Free Software Foundation.
+//
+//   This program is distributed in the hope that it will be useful,
+//   but WITHOUT ANY WARRANTY; without even the implied warranty of
+//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+//   GNU General Public License for more details.
+//
+//   You should have received a copy of the GNU General Public
+//   License along with this program; if not, write to the Free Software
+//   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+//
+
+#include <qobject.h>
+#include <qstring.h>
+#include <qtextcodec.h>
+#include <qtranslator.h>
+#include <qserversocket.h>
+#include <qsqldatabase.h>
+#include <qsqlerror.h>
+#include <assert.h>
+
+#include "rddb.h"
+#include "rddbheartbeat.h"
+
+static QSqlDatabase *db = NULL;
+static RDSqlDatabaseStatus * dbStatus = NULL;
+
+QSqlDatabase *RDInitDb (unsigned *schema,QString *error)
+{
+  static bool firsttime = true;
+
+  *schema=0;
+  RDConfig *cf = RDConfiguration();
+  cf->load();
+  assert (cf);
+  if (!db){
+    db=QSqlDatabase::addDatabase(cf->mysqlDriver());
+    if(!db) {
+      if (error){
+	(*error) += QString(QObject::tr("Couldn't initialize QSql driver!"));
+      }
+      return NULL;
+    }
+    db->setDatabaseName(cf->mysqlDbname());
+    db->setUserName(cf->mysqlUsername());
+    db->setPassword(cf->mysqlPassword());
+    db->setHostName(cf->mysqlHostname());
+    if(!db->open()) {
+      if (error){
+	(*error) += QString(QObject::tr("Couldn't open mySQL connection!"));
+      }
+      db->removeDatabase(cf->mysqlDbname());
+      db->close();
+      return NULL;
+    }
+  }
+  if (firsttime){
+    new RDDbHeartbeat(cf->mysqlHeartbeatInterval());
+    firsttime = false;
+  }
+  //  QSqlQuery *q=new QSqlQuery("set character_set_results='utf8'");
+  //  delete q;
+
+  QSqlQuery *q=new QSqlQuery("select DB from VERSION");
+  if(q->first()) {
+    *schema=q->value(0).toUInt();
+  }
+  delete q;
+
+  return db;
+}
+
+RDSqlQuery::RDSqlQuery (const QString &query, QSqlDatabase *dbase):
+  QSqlQuery (query,dbase)
+{
+  //printf("lastQuery: %s\n",(const char *)lastQuery());
+
+  // With any luck, by the time we get here, we have already done the biz...
+  unsigned schema;
+  if (!isActive()){ //DB Offline?
+    QString err=QObject::tr("invalid SQL or failed DB connection")+
+      +"["+lastError().text()+"]: "+query;
+
+    fprintf(stderr,"%s\n",(const char *)err);
+#ifndef WIN32
+    syslog(LOG_ERR,(const char *)err);
+#endif  // WIN32
+    QSqlDatabase *ldb = QSqlDatabase::database();
+    // Something went wrong with the DB, trying a reconnect
+    ldb->removeDatabase(RDConfiguration()->mysqlDbname());
+    ldb->close();
+    db = NULL;
+    RDInitDb (&schema);
+    QSqlQuery::prepare (query);
+    QSqlQuery::exec ();
+    if (RDDbStatus()){
+      if (isActive()){
+	RDDbStatus()->sendRecon();
+      } else {
+	RDDbStatus()->sendDiscon(query);
+      }
+    }
+  } else {
+    RDDbStatus()->sendRecon();
+  }
+}
+void RDSqlDatabaseStatus::sendRecon()
+{
+  if (discon){
+    discon = false;
+    emit reconnected();
+    fprintf (stderr,"Database connection restored.\n");
+    emit logText(RDConfig::LogErr,QString(tr("Database connection restored.")));
+  }
+}
+
+void RDSqlDatabaseStatus::sendDiscon(QString query)
+{
+  if (!discon){
+    emit connectionFailed();
+    fprintf (stderr,"Database connection failed: %s\n",(const char *)query);
+    emit logText(RDConfig::LogErr,
+		 QString(tr("Database connection failed : ")) + query);
+    discon = true;
+  }
+}
+
+RDSqlDatabaseStatus::RDSqlDatabaseStatus()
+{
+  discon = false;
+}
+
+RDSqlDatabaseStatus * RDDbStatus()
+{
+  if (!dbStatus){
+    dbStatus = new RDSqlDatabaseStatus;
+  }
+  return dbStatus;
+}