SqlQuery.cpp

Go to the documentation of this file.

#include <string>
#include <vector>
#include <iostream>
#include "DbProfile.h"
#include "Database.h"
#include "IString.h"
#include "SqlQuery.h"
#include "SqlRecord.h"
#include <QString>

using namespace std;

namespace Isis {

  SqlQuery::SqlQuery() : QSqlQuery(Database()), _throwIfFailed(true) { }

  SqlQuery::SqlQuery(Database &db) : QSqlQuery(db), _throwIfFailed(true) { }


  SqlQuery::SqlQuery(const std::string &query, Database db) : QSqlQuery(db),
    _throwIfFailed(true) {
//  Execute with error detector
    exec(query);
  }

  SqlQuery::SqlQuery(const SqlQuery &other) : QSqlQuery(other),
    _throwIfFailed(other._throwIfFailed) { }


  bool SqlQuery::exec(const std::string &query) {
    bool ok = this->QSqlQuery::exec(IString::ToQt(query));
    if((!ok) && isThrowing()) {
      string mess = "Query \'" + query + "\' failed to execute";
      tossQueryError(mess, _FILEINFO_);
    }
    return (ok);
  }

  std::string SqlQuery::getQuery() const {
    std::string lastq = IString::ToStd(lastQuery());
    if(lastq.empty()) lastq = IString::ToStd(executedQuery());
    return (lastq);
  }

  int SqlQuery::nFields() const {
    return (record().count());
  }

  std::string SqlQuery::fieldName(int index) const {
    return (IString::ToStd(record().fieldName(index)));
  }

  int SqlQuery::fieldIndex(const std::string &name) const {
    return(record().indexOf(IString::ToQt(name)));
  }

  std::vector<std::string> SqlQuery::fieldNameList() const {
    std::vector<std::string> fields;
    QSqlRecord rec = record();
    for(int i = 0 ; i < rec.count() ; i++) {
      fields.push_back(IString::ToStd(rec.fieldName(i)));
    }
    return (fields);
  }

  std::vector<std::string> SqlQuery::fieldTypeList() const {
    std::vector<std::string> types;
    SqlRecord rec = getRecord();
    for(int i = 0 ; i < rec.count() ; i++) {
      types.push_back(rec.getType(i).toAscii().data());
    }
    return (types);
  }

  int SqlQuery::nRows() const {
    return (size());
  }

  SqlRecord SqlQuery::getRecord() const {
    return (SqlRecord(*this));
  }

  void SqlQuery::tossQueryError(const std::string &message, const char *f, int l) const {
    string errmess = message + " - QueryError = " +
                     IString::ToStd(lastError().text());
    throw IException(IException::User, errmess, f, l);
  }

}