mapnik/plugins/input/postgis/cursorresultset.hpp

154 lines
3.4 KiB
C++
Raw Normal View History

#ifndef CURSORRESULTSET_HPP
#define CURSORRESULTSET_HPP
#include "connection.hpp"
#include "resultset.hpp"
class CursorResultSet : public IResultSet
{
private:
boost::shared_ptr<Connection> conn_;
std::string cursorName_;
boost::shared_ptr<ResultSet> rs_;
int fetch_size_;
bool is_closed_;
int *refCount_;
2010-11-18 23:42:33 +00:00
void getNextResultSet()
{
std::ostringstream s;
s << "FETCH FORWARD " << fetch_size_ << " FROM " << cursorName_;
#ifdef MAPNIK_DEBUG
2010-11-18 23:42:33 +00:00
std::clog << "Postgis Plugin: " << s.str() << std::endl;
#endif
rs_ = conn_->executeQuery(s.str());
is_closed_ = false;
#ifdef MAPNIK_DEBUG
2010-11-18 23:42:33 +00:00
std::clog << "Postgis Plugin: FETCH result (" << cursorName_ << "): " << rs_->size() << " rows" << std::endl;
#endif
}
public:
CursorResultSet(boost::shared_ptr<Connection> const &conn, std::string cursorName, int fetch_count)
2010-11-18 23:42:33 +00:00
: conn_(conn),
cursorName_(cursorName),
fetch_size_(fetch_count),
is_closed_(false),
refCount_(new int(1))
{
getNextResultSet();
}
CursorResultSet(const CursorResultSet& rhs)
2010-11-18 23:42:33 +00:00
: conn_(rhs.conn_),
cursorName_(rhs.cursorName_),
rs_(rhs.rs_),
fetch_size_(rhs.fetch_size_),
is_closed_(rhs.is_closed_),
refCount_(rhs.refCount_)
{
(*refCount_)++;
}
CursorResultSet& operator=(const CursorResultSet& rhs)
{
if (this==&rhs) return *this;
if (--(refCount_)==0)
{
close();
delete refCount_,refCount_=0;
}
conn_=rhs.conn_;
cursorName_=rhs.cursorName_;
rs_=rhs.rs_;
refCount_=rhs.refCount_;
fetch_size_=rhs.fetch_size_;
is_closed_ = false;
(*refCount_)++;
return *this;
}
virtual void close()
{
2010-11-18 23:42:33 +00:00
if (!is_closed_)
{
rs_.reset();
std::ostringstream s;
s << "CLOSE " << cursorName_;
#ifdef MAPNIK_DEBUG
2010-11-18 23:42:33 +00:00
std::clog << "Postgis Plugin: " << s.str() << std::endl;
#endif
conn_->execute(s.str());
is_closed_ = true;
}
}
virtual ~CursorResultSet()
{
if (--(*refCount_)==0)
{
close();
delete refCount_,refCount_=0;
}
}
virtual int getNumFields() const
{
return rs_->getNumFields();
}
virtual bool next()
{
if (rs_->next()) {
return true;
} else if (rs_->size() == 0) {
return false;
} else {
getNextResultSet();
return rs_->next();
}
}
virtual const char* getFieldName(int index) const
{
return rs_->getFieldName(index);
}
virtual int getFieldLength(int index) const
{
return rs_->getFieldLength(index);
}
virtual int getFieldLength(const char* name) const
{
return rs_->getFieldLength(name);
}
virtual int getTypeOID(int index) const
{
return rs_->getTypeOID(index);
}
virtual int getTypeOID(const char* name) const
{
return rs_->getTypeOID(name);
}
virtual bool isNull(int index) const
{
return rs_->isNull(index);
}
virtual const char* getValue(int index) const
{
return rs_->getValue(index);
}
virtual const char* getValue(const char* name) const
{
return rs_->getValue(name);
}
};
#endif //CURSORRESULTSET_HPP