JDB.hh

Go to the documentation of this file.

#ifndef __JDB_JDB__
#define __JDB_JDB__

#include <string>
#include <fstream>

#include "KM3NeTDBClient.h"

#include "JSystem/JStat.hh"


/**
 * \author mdejong
 */

namespace KM3NETDB {

  using KM3NeT::DB::DBException;
  using KM3NeT::DB::Server;
  using KM3NeT::DB::Client;
  using KM3NeT::DB::ResultSet;
  using KM3NeT::DB::Selector;

  /**
   * Cookie prefix.
   */
  static const char PREFIX_COOKIE = '_';


  /**
   * Auxiliary class for connection to data base.
   */
  class JDB :
    public std::shared_ptr<Client>
  {
  public:
    /**
     * Get connection to database.
     *
     * \return               database connection
     */
    static JDB& get()
    {
      static JDB db;

      return db;
    }


    /**
     * Reset connection to database.
     */
    static void reset()
    {
      JDB& db = JDB::get();

      if (db.get() != NULL) {
        db->Close();
        db. reset();
      }
    }


    /**
     * Reset connection to database.
     *
     * \param  usr           user name
     * \param  pwd           pass word
     */
    static void reset(const std::string& usr, const std::string& pwd)
    {
      static_cast<std::shared_ptr<Client>&>(JDB::get()) = Client::Create(Server::Default, usr.c_str(), pwd.c_str());
    }

    
    /**
     * Reset connection to database.
     *
     * \param  cookie        persistent cookie
     */
    static void reset(const std::string& cookie)
    {
      using namespace std;
      using namespace JPP;

      string buffer = cookie;
      
      if (getFileStatus(cookie.c_str())) {

        ifstream in(cookie.c_str());

        getline(in, buffer);

        string::size_type pos = buffer.find(PREFIX_COOKIE);

        if (pos != string::npos) {
          buffer.erase(0,pos);
        }

        in.close();
      }
      
      static_cast<std::shared_ptr<Client>&>(JDB::get()) = Client::Create(Server::Default, buffer.c_str());
    }


  private:
    /**
     * Default constructor.
     */
    JDB()
    {}
    

    JDB(const JDB&);
    JDB& operator=(const JDB&);
  };
}

#endif