ResultSetImplPGSQL.H

/* 
 * Copyright (c) 2005-2011, Guillaume Gimenez <guillaume@blackmilk.fr>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of G.Gimenez nor the names of its contributors may
 *       be used to endorse or promote products derived from this software
 *       without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL G.GIMENEZ BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Authors:
 *     * Guillaume Gimenez <guillaume@blackmilk.fr>
 *
 */
#ifndef RAIIRESULTSETIMPLPGSQL_H
#define RAIIRESULTSETIMPLPGSQL_H

#include<libpq-fe.h>

namespace raii {
namespace sql {

class ResultSetImplPGSQL : public ResultSetImpl {

	PGresult *result;
	//interdit
	ResultSetImplPGSQL(const ResultSetImpl& rsi);

	Map<String,int> tupleIndex;
	public:

	ResultSetImplPGSQL(SQLDriverPGSQL& driver, const String& q)
		:	ResultSetImpl(driver,q),
			result(NULL),
			tupleIndex()
			{

                sendQuery(q);
	}

	virtual ~ResultSetImplPGSQL() {

		if ( result )
			PQclear(result);
	}

	virtual void sendQuery(const String& query) {

		cursor=-1;

		if ( result ) PQclear(result);
		result=PQexec((PGconn*)driver.conn,query.c_str());
		if ( result == NULL ) {
			raii::Logger log("raii::sql");
			log.error(String("Attempted query was: ")+query);
			throw PostgreSQLException("result is null");
		}
		if (	( PQresultStatus(result) != PGRES_COMMAND_OK )
		     && ( PQresultStatus(result) != PGRES_TUPLES_OK) ) {
			Logger log("raii::sql");
			log.error(String("Attempted query was: ")+query);
			throw PostgreSQLException(PQerrorMessage((PGconn*)driver.conn));
		}
		
		tupleIndex.clear();
		int nfields = PQnfields(result);
		for ( int i = 0 ; i < nfields ; ++i ) {
			tupleIndex[PQfname(result,i)]=i;
		}
	}

	virtual int rowCount() {

		return PQntuples(result);
	}

	virtual String operator [] (const String& attrName) {

		Map<String,int>::iterator pair = tupleIndex.find(attrName);
		if ( pair == tupleIndex.end() )
			throw PostgreSQLException("Field "+attrName+" doesn't exist.");
		if ( cursor >= 0 && cursor < rowCount() )
			return PQgetvalue(result,cursor,pair->second);
		throw PostgreSQLException("Cursor out of bound");
	}

};

}
}

#endif