8
0
mirror of https://github.com/FirebirdSQL/firebird.git synced 2025-01-31 01:23:02 +01:00
firebird-mirror/src/jrd/recsrc/Cursor.cpp
asfernandes bb735cb5a2 Misc.
2014-05-23 02:12:56 +00:00

321 lines
7.0 KiB
C++

/*
* The contents of this file are subject to the Initial
* Developer's Public License Version 1.0 (the "License");
* you may not use this file except in compliance with the
* License. You may obtain a copy of the License at
* http://www.ibphoenix.com/main.nfs?a=ibphoenix&page=ibp_idpl.
*
* Software distributed under the License is distributed AS IS,
* WITHOUT WARRANTY OF ANY KIND, either express or implied.
* See the License for the specific language governing rights
* and limitations under the License.
*
* The Original Code was created by Dmitry Yemanov
* for the Firebird Open Source RDBMS project.
*
* Copyright (c) 2009 Dmitry Yemanov <dimitr@firebirdsql.org>
* and all contributors signed below.
*
* All Rights Reserved.
* Contributor(s): ______________________________________.
*/
#include "firebird.h"
#include "../jrd/jrd.h"
#include "../jrd/req.h"
#include "../jrd/cmp_proto.h"
#include "RecordSource.h"
#include "Cursor.h"
using namespace Firebird;
using namespace Jrd;
// ---------------------
// Cursor implementation
// ---------------------
Cursor::Cursor(CompilerScratch* csb, const RecordSource* rsb,
const VarInvariantArray* invariants, bool scrollable)
: m_top(rsb), m_invariants(invariants), m_scrollable(scrollable)
{
fb_assert(m_top);
m_impure = CMP_impure(csb, sizeof(Impure));
}
void Cursor::open(thread_db* tdbb) const
{
jrd_req* const request = tdbb->getRequest();
Impure* impure = request->getImpure<Impure>(m_impure);
impure->irsb_active = true;
impure->irsb_state = BOS;
// Initialize dependent invariants
if (m_invariants)
{
for (const ULONG* iter = m_invariants->begin(); iter < m_invariants->end(); iter++)
{
impure_value* const invariantImpure = request->getImpure<impure_value>(*iter);
invariantImpure->vlu_flags = 0;
}
}
m_top->open(tdbb);
}
void Cursor::close(thread_db* tdbb) const
{
jrd_req* const request = tdbb->getRequest();
Impure* const impure = request->getImpure<Impure>(m_impure);
if (impure->irsb_active)
{
impure->irsb_active = false;
m_top->close(tdbb);
}
}
bool Cursor::fetchNext(thread_db* tdbb) const
{
if (!validate(tdbb))
return false;
jrd_req* const request = tdbb->getRequest();
Impure* const impure = request->getImpure<Impure>(m_impure);
if (!impure->irsb_active)
{
// error: invalid cursor state
status_exception::raise(Arg::Gds(isc_cursor_not_open));
}
if (impure->irsb_state == EOS)
{
// error: cursor is past EOF
status_exception::raise(Arg::Gds(isc_stream_eof));
}
else if (impure->irsb_state == BOS)
{
impure->irsb_position = 0;
}
else
{
impure->irsb_position++;
}
if (!m_scrollable)
{
if (!m_top->getRecord(tdbb))
{
impure->irsb_state = EOS;
return false;
}
}
else
{
const BufferedStream* const buffer = static_cast<const BufferedStream*>(m_top);
buffer->locate(tdbb, impure->irsb_position);
if (!buffer->getRecord(tdbb))
{
impure->irsb_state = EOS;
return false;
}
}
request->req_records_selected++;
request->req_records_affected.bumpFetched();
impure->irsb_state = POSITIONED;
return true;
}
bool Cursor::fetchPrior(thread_db* tdbb) const
{
if (!m_scrollable)
{
// error: invalid fetch direction
status_exception::raise(Arg::Gds(isc_invalid_fetch_option) << Arg::Str("PRIOR"));
}
if (!validate(tdbb))
return false;
jrd_req* const request = tdbb->getRequest();
Impure* const impure = request->getImpure<Impure>(m_impure);
if (!impure->irsb_active)
{
// error: invalid cursor state
status_exception::raise(Arg::Gds(isc_cursor_not_open));
}
const BufferedStream* const buffer = static_cast<const BufferedStream*>(m_top);
if (impure->irsb_state == BOS)
{
// error: cursor is prior BOF
status_exception::raise(Arg::Gds(isc_stream_bof));
}
else if (impure->irsb_state == EOS)
{
impure->irsb_position = buffer->getCount(tdbb) - 1;
}
else
{
impure->irsb_position--;
}
buffer->locate(tdbb, impure->irsb_position);
if (!buffer->getRecord(tdbb))
{
impure->irsb_state = BOS;
return false;
}
request->req_records_selected++;
request->req_records_affected.bumpFetched();
impure->irsb_state = POSITIONED;
return true;
}
bool Cursor::fetchFirst(thread_db* tdbb) const
{
if (!m_scrollable)
{
// error: invalid fetch direction
status_exception::raise(Arg::Gds(isc_invalid_fetch_option) << Arg::Str("FIRST"));
}
return fetchAbsolute(tdbb, 1);
}
bool Cursor::fetchLast(thread_db* tdbb) const
{
if (!m_scrollable)
{
// error: invalid fetch direction
status_exception::raise(Arg::Gds(isc_invalid_fetch_option) << Arg::Str("LAST"));
}
return fetchAbsolute(tdbb, -1);
}
bool Cursor::fetchAbsolute(thread_db* tdbb, SINT64 offset) const
{
if (!m_scrollable)
{
// error: invalid fetch direction
status_exception::raise(Arg::Gds(isc_invalid_fetch_option) << Arg::Str("ABSOLUTE"));
}
if (!validate(tdbb))
return false;
jrd_req* const request = tdbb->getRequest();
Impure* const impure = request->getImpure<Impure>(m_impure);
if (!impure->irsb_active)
{
// error: invalid cursor state
status_exception::raise(Arg::Gds(isc_cursor_not_open));
}
if (!offset)
{
impure->irsb_state = BOS;
return false;
}
const BufferedStream* const buffer = static_cast<const BufferedStream*>(m_top);
impure->irsb_position = (offset > 0) ? offset - 1 : buffer->getCount(tdbb) + offset;
buffer->locate(tdbb, impure->irsb_position);
if (!buffer->getRecord(tdbb))
{
impure->irsb_state = (offset > 0) ? EOS : BOS;
return false;
}
request->req_records_selected++;
request->req_records_affected.bumpFetched();
impure->irsb_state = POSITIONED;
return true;
}
bool Cursor::fetchRelative(thread_db* tdbb, SINT64 offset) const
{
if (!m_scrollable)
{
// error: invalid fetch direction
status_exception::raise(Arg::Gds(isc_invalid_fetch_option) << Arg::Str("RELATIVE"));
}
if (!validate(tdbb))
return false;
jrd_req* const request = tdbb->getRequest();
Impure* const impure = request->getImpure<Impure>(m_impure);
if (!impure->irsb_active)
{
// error: invalid cursor state
status_exception::raise(Arg::Gds(isc_cursor_not_open));
}
if (!offset)
{
return (impure->irsb_state == POSITIONED);
}
if (impure->irsb_state == BOS && offset < 0)
{
return false;
}
if (impure->irsb_state == EOS && offset > 0)
{
return false;
}
const BufferedStream* const buffer = static_cast<const BufferedStream*>(m_top);
impure->irsb_position += offset;
buffer->locate(tdbb, impure->irsb_position);
if (!buffer->getRecord(tdbb))
{
impure->irsb_state = (offset > 0) ? EOS : BOS;
return false;
}
request->req_records_selected++;
request->req_records_affected.bumpFetched();
impure->irsb_state = POSITIONED;
return true;
}
// Check if the cursor is in a good state for access a field.
void Cursor::checkState(jrd_req* request) const
{
const Impure* const impure = request->getImpure<Impure>(m_impure);
if (!impure->irsb_active)
{
// error: invalid cursor state
status_exception::raise(Arg::Gds(isc_cursor_not_open));
}
if (impure->irsb_state != Cursor::POSITIONED)
{
status_exception::raise(
Arg::Gds(isc_cursor_not_positioned) <<
Arg::Str(name));
}
}