DBParams.hpp

Go to the documentation of this file.

#ifndef __OPENTREP_DBPARAMS_HPP
#define __OPENTREP_DBPARAMS_HPP
 
// //////////////////////////////////////////////////////////////////////
// Import section
// //////////////////////////////////////////////////////////////////////
// STL
#include <sstream>
#include <string>
// OpenTrep
#include <opentrep/OPENTREP_Types.hpp>
#include <opentrep/OPENTREP_Abstract.hpp>
#include <opentrep/DBType.hpp>
 
namespace OPENTREP {

  typedef std::list<std::string> DBParamsNameList_T;
 
  
  struct DBParams : public OPENTREP_Abstract {
  public:
    // ///////////////////// Getters //////////////////////
    const DBType& getType() const {
      return _dbtype;
    }
    
    const std::string& getUser() const {
      return _user;
    }
    
    const std::string& getPassword() const {
      return _passwd;
    }
    
    const std::string& getHost() const {
      return _host;
    }
    
    const std::string& getPort() const {
      return _port;
    }
    
    const std::string& getDBName() const {
      return _dbname;
    }
    
    
  public:
    // /////////////////////// Setters ///////////////////////
    void setType (const DBType& iType) {
      _dbtype = iType;
    }

    void setUser (const std::string& iUser) {
      _user = iUser;
    }

    void setPassword (const std::string& iPasswd) {
      _passwd = iPasswd;
    }

    void setHost (const std::string& iHost) {
      _host = iHost;
    }

    void setPort (const std::string& iPort) {
      _port = iPort;
    }

    void setDBName (const std::string& iDBName) {
      _dbname = iDBName;
    }
 
 
  public:
    // ///////////////////// Busines methods ////////////////////
    bool checkSQLite () const {
      if (_dbname.empty() == true) {
        return false;
      }
      return true;
    }
    
    bool checkMySQL () const {
      if (_user.empty() == true || _passwd.empty() == true
          || _host.empty() == true || _port.empty()
          || _dbname.empty() == true) {
        return false;
      }
      return true;
    }

    bool checkPG () const {
      if (_user.empty() == true || _passwd.empty() == true
          || _dbname.empty() == true) {
        return false;
      }
      return true;
    }
    
  public:
    // //////////////////// Display methods //////////////////////
    void toStream (std::ostream& ioOut) const {
      ioOut << toString();
    }

    void fromStream (std::istream&) {
    }

    std::string toShortString() const {
      std::ostringstream oStr;
      oStr << _dbname << "." << _user << "@" << _host << ":" << _port;
      return oStr.str();
    }
    
    std::string toString() const {
      std::ostringstream oStr;
      oStr << _dbname << "." << _user << "@" << _host << ":" << _port;
      return oStr.str();
    }

    std::string toMySQLConnectionString() const {
      std::ostringstream oStr;
      oStr << "db=" << _dbname << " user=" << _user << " password=" << _passwd
           << " port=" << _port << " host=" << _host;
      return oStr.str();
    }

    std::string toPGConnectionString() const {
      std::ostringstream oStr;
      oStr << "dbname=" << _dbname << " user=" << _user
           << " password=" << _passwd;
      if (!_host.empty()) {
        oStr << " host=" << _host;
      }
      if (!_port.empty()) {
        oStr << " port=" << _port;
      }
      return oStr.str();
    }

    std::string toSQLiteConnectionString() const {
      std::ostringstream oStr;
      oStr << "db=" << _dbname;
      return oStr.str();
    }
 
 
  public:
    // /////////////// Constructors and Destructors ///////////////////
    DBParams (const DBType& iDBType,
              const std::string& iDBUser, const std::string& iDBPasswd,
              const std::string& iDBHost, const std::string& iDBPort,
              const std::string& iDBName)
      : _dbtype (iDBType), _user (iDBUser), _passwd (iDBPasswd),
        _host (iDBHost), _port (iDBPort), _dbname (iDBName) {
    }
    DBParams (const DBType& iDBType, const std::string& iDBName)
      : _dbtype (iDBType), _dbname (iDBName) {
    }

    // DBParams();
    // DBParams (const DBParams&);
    
    virtual ~DBParams() {}
 
    
  private:
    // ///////////////////////// Attributes ////////////////////////
    DBType _dbtype;

    std::string _user;

    std::string _passwd;

    std::string _host;

    std::string _port;

    std::string _dbname;
  };
 
}
#endif // __OPENTREP_DBPARAMS_HPP