Renamed version4 to flaim and version5 to xflaim
git-svn-id: https://svn.code.sf.net/p/flaim/code/trunk@7 0109f412-320b-0410-ab79-c3e0c5ffbbe6
This commit is contained in:
615
xflaim/src/fbtrset.cpp
Normal file
615
xflaim/src/fbtrset.cpp
Normal file
@@ -0,0 +1,615 @@
|
||||
//------------------------------------------------------------------------------
|
||||
// Desc: This file contains routines that implement a result set using
|
||||
// a temporary XFLAIM database.
|
||||
//
|
||||
// Tabs: 3
|
||||
//
|
||||
// Copyright (c) 1991-2006 Novell, Inc. All Rights Reserved.
|
||||
//
|
||||
// This program is free software; you can redistribute it and/or
|
||||
// modify it under the terms of version 2 of the GNU General Public
|
||||
// License as published by the Free Software Foundation.
|
||||
//
|
||||
// This program is distributed in the hope that it will be useful,
|
||||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
// GNU General Public License for more details.
|
||||
//
|
||||
// You should have received a copy of the GNU General Public License
|
||||
// along with this program; if not, contact Novell, Inc.
|
||||
//
|
||||
// To contact Novell about this file by physical or electronic mail,
|
||||
// you may find current contact information at www.novell.com
|
||||
//
|
||||
// $Id: fbtrset.cpp 3111 2006-01-19 13:10:50 -0700 (Thu, 19 Jan 2006) dsanders $
|
||||
//------------------------------------------------------------------------------
|
||||
|
||||
#include "flaimsys.h"
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
F_BtResultSet::~F_BtResultSet()
|
||||
{
|
||||
// Free the collection table if it was ever created.
|
||||
|
||||
if (m_ppCollectionTbl)
|
||||
{
|
||||
FLMUINT uiLoop;
|
||||
|
||||
for (uiLoop = 0; uiLoop < BT_MAX_COLLECTION_TBL_SIZ; uiLoop++)
|
||||
{
|
||||
if (m_ppCollectionTbl[ uiLoop] != NULL)
|
||||
{
|
||||
BT_COLLECTION_XREF * pTmp;
|
||||
|
||||
while (m_ppCollectionTbl[ uiLoop] != NULL)
|
||||
{
|
||||
pTmp = m_ppCollectionTbl[ uiLoop];
|
||||
m_ppCollectionTbl[ uiLoop] = pTmp->pNext;
|
||||
if (pTmp && pTmp->pCompare)
|
||||
{
|
||||
pTmp->pCompare->Release();
|
||||
}
|
||||
f_free( &pTmp);
|
||||
}
|
||||
}
|
||||
}
|
||||
f_free( &m_ppCollectionTbl);
|
||||
}
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getBTree(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
F_Btree ** ppBTree)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMUINT uiCollHash;
|
||||
BT_COLLECTION_XREF * pCollPtr = NULL;
|
||||
F_RandomGenerator * pRandGen = NULL;
|
||||
F_Database * pDatabase;
|
||||
FLMUINT uiCollection;
|
||||
|
||||
if (RC_BAD( rc = m_pBtPool->btpReserveBtree( ppBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if (pSrcIxd)
|
||||
{
|
||||
if (!m_ppCollectionTbl)
|
||||
{
|
||||
if (RC_BAD( rc = f_calloc(
|
||||
BT_MAX_COLLECTION_TBL_SIZ * sizeof(BT_COLLECTION_XREF),
|
||||
&m_ppCollectionTbl)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
uiCollHash = pSrcIxd->uiIndexNum % BT_MAX_COLLECTION_TBL_SIZ;
|
||||
|
||||
pCollPtr = m_ppCollectionTbl[ uiCollHash];
|
||||
|
||||
// Verify that we have the right collection
|
||||
while (pCollPtr && pCollPtr->uiKeyNum != pSrcIxd->uiIndexNum)
|
||||
{
|
||||
pCollPtr = pCollPtr->pNext;
|
||||
}
|
||||
|
||||
if (!pCollPtr)
|
||||
{
|
||||
pDatabase = m_pResultSetDb->m_pDatabase;
|
||||
|
||||
// Will need a random number generator.
|
||||
|
||||
if (( pRandGen = f_new F_RandomGenerator) == NULL)
|
||||
{
|
||||
rc = RC_SET( NE_XFLM_MEM);
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
pRandGen->randomSetSeed( (FLMINT32)pSrcIxd->uiIndexNum);
|
||||
|
||||
// Allocate a new collection key context and create a new
|
||||
// collection for it.
|
||||
|
||||
if (RC_BAD( rc = f_calloc( sizeof(BT_COLLECTION_XREF), &pCollPtr)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
// Insert into the table at the head of the list.
|
||||
|
||||
pCollPtr->pCompare = NULL;
|
||||
pCollPtr->pNext = m_ppCollectionTbl[ uiCollHash];
|
||||
m_ppCollectionTbl[ uiCollHash] = pCollPtr;
|
||||
|
||||
TryAgain:
|
||||
|
||||
// Randomly select a collection number to use.
|
||||
|
||||
uiCollection = pRandGen->randomChoice( 100, XFLM_MAX_COLLECTION_NUM);
|
||||
|
||||
// Check to see if it already exists.
|
||||
if (RC_BAD( rc = pDatabase->lFileCreate( m_pResultSetDb,
|
||||
&pCollPtr->Collection.lfInfo, &pCollPtr->Collection,
|
||||
uiCollection, XFLM_LF_COLLECTION, FALSE, TRUE,
|
||||
pSrcIxd->lfInfo.uiEncId)))
|
||||
{
|
||||
if (rc != NE_XFLM_EXISTS)
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
rc = NE_XFLM_OK;
|
||||
goto TryAgain;
|
||||
}
|
||||
|
||||
pCollPtr->uiKeyNum = pSrcIxd->uiIndexNum;
|
||||
pCollPtr->uiCollection = uiCollection;
|
||||
|
||||
// Set up the comparison object.
|
||||
|
||||
if ((pCollPtr->pCompare = f_new IXKeyCompare) == NULL)
|
||||
{
|
||||
rc = RC_SET( NE_XFLM_MEM);
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
pCollPtr->pCompare->setIxInfo( pSrcDb, pSrcIxd);
|
||||
|
||||
// Open the btree and use the specified collection.
|
||||
|
||||
if (RC_BAD( rc = (*ppBTree)->btOpen( m_pResultSetDb,
|
||||
&pCollPtr->Collection.lfInfo,
|
||||
FALSE, TRUE, pCollPtr->pCompare)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
if (RC_BAD( rc = (*ppBTree)->btOpen( m_pResultSetDb,
|
||||
&m_Collection.lfInfo,
|
||||
FALSE, TRUE, NULL)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if (pRandGen)
|
||||
{
|
||||
pRandGen->Release();
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::addEntry(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyLength,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
F_Btree * pBTree = NULL;
|
||||
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyLength <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btInsertEntry( pucKey, uiKeyLength,
|
||||
uiKeyLength, pucEntry, uiEntryLength, TRUE, TRUE)))
|
||||
{
|
||||
if (rc == NE_XFLM_NOT_UNIQUE)
|
||||
{
|
||||
rc = NE_XFLM_OK;
|
||||
}
|
||||
else
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if (pBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::modifyEntry(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyLength,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
F_Btree * pBTree = NULL;
|
||||
|
||||
if (RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyLength <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btReplaceEntry( pucKey, uiKeyLength,
|
||||
uiKeyLength, pucEntry, uiEntryLength, TRUE, TRUE)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( pBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::deleteEntry(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
F_Btree * pBTree = NULL;
|
||||
|
||||
if (RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyLength <= MAX_KEY_SIZ);
|
||||
|
||||
if (RC_BAD( rc = pBTree->btRemoveEntry( pucKey,
|
||||
uiKeyLength,
|
||||
uiKeyLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if (pBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
return rc;
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::findEntry(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyBufLen,
|
||||
FLMUINT * puiKeyLen,
|
||||
FLMBYTE * pucBuffer,
|
||||
FLMUINT uiBufferLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMUINT uiLengthRV;
|
||||
F_Btree * pBTree = NULL;
|
||||
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyBufLen <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btLocateEntry( pucKey, uiKeyBufLen, puiKeyLen,
|
||||
XFLM_EXACT, NULL, &uiLengthRV)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if( pucBuffer)
|
||||
{
|
||||
// Get the entry ...
|
||||
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, uiKeyBufLen, *puiKeyLen,
|
||||
pucBuffer, uiBufferLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
else if( puiReturnLength)
|
||||
{
|
||||
*puiReturnLength = uiLengthRV;
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( pBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getCurrent(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyLength,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
F_Btree * pBTree = NULL;
|
||||
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyLength <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, uiKeyLength, uiKeyLength,
|
||||
pucEntry, uiEntryLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( pBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getNext(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
F_Btree * pBTree,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyBufLen,
|
||||
FLMUINT * puiKeyLen,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMBOOL bFreeBTree = FALSE;
|
||||
|
||||
if( !pBTree)
|
||||
{
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
bFreeBTree = TRUE;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyBufLen <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btNextEntry( pucKey, uiKeyBufLen, puiKeyLen,
|
||||
puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if( pucEntry)
|
||||
{
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, *puiKeyLen, *puiKeyLen,
|
||||
pucEntry, uiEntryLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( bFreeBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getPrev(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
F_Btree * pBTree,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyBufLen,
|
||||
FLMUINT * puiKeyLen,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMBOOL bFreeBTree = FALSE;
|
||||
|
||||
if( !pBTree)
|
||||
{
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
bFreeBTree = TRUE;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyBufLen <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btPrevEntry( pucKey, uiKeyBufLen, puiKeyLen,
|
||||
puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if( pucEntry)
|
||||
{
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, *puiKeyLen, *puiKeyLen,
|
||||
pucEntry, uiEntryLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( bFreeBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getFirst(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
F_Btree * pBTree,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyBufLen,
|
||||
FLMUINT * puiKeyLen,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMBOOL bFreeBTree = FALSE;
|
||||
|
||||
if( !pBTree)
|
||||
{
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
bFreeBTree = TRUE;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyBufLen <= MAX_KEY_SIZ);
|
||||
|
||||
pBTree->btResetBtree();
|
||||
|
||||
if( RC_BAD( rc = pBTree->btFirstEntry( pucKey, uiKeyBufLen, puiKeyLen,
|
||||
puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if( pucEntry)
|
||||
{
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, *puiKeyLen, *puiKeyLen,
|
||||
pucEntry, uiEntryLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( bFreeBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
|
||||
/****************************************************************************
|
||||
Desc:
|
||||
****************************************************************************/
|
||||
RCODE F_BtResultSet::getLast(
|
||||
F_Db * pSrcDb,
|
||||
IXD * pSrcIxd,
|
||||
F_Btree * pBTree,
|
||||
FLMBYTE * pucKey,
|
||||
FLMUINT uiKeyBufLen,
|
||||
FLMUINT * puiKeyLen,
|
||||
FLMBYTE * pucEntry,
|
||||
FLMUINT uiEntryLength,
|
||||
FLMUINT * puiReturnLength)
|
||||
{
|
||||
RCODE rc = NE_XFLM_OK;
|
||||
FLMBOOL bFreeBTree = FALSE;
|
||||
|
||||
if( !pBTree)
|
||||
{
|
||||
if( RC_BAD( rc = getBTree( pSrcDb, pSrcIxd, &pBTree)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
bFreeBTree = TRUE;
|
||||
}
|
||||
|
||||
flmAssert( uiKeyBufLen <= MAX_KEY_SIZ);
|
||||
|
||||
if( RC_BAD( rc = pBTree->btLastEntry( pucKey, uiKeyBufLen, puiKeyLen,
|
||||
puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
|
||||
if( pucEntry)
|
||||
{
|
||||
if( RC_BAD( rc = pBTree->btGetEntry( pucKey, *puiKeyLen, *puiKeyLen,
|
||||
pucEntry, uiEntryLength, puiReturnLength)))
|
||||
{
|
||||
goto Exit;
|
||||
}
|
||||
}
|
||||
|
||||
Exit:
|
||||
|
||||
if( bFreeBTree)
|
||||
{
|
||||
m_pBtPool->btpReturnBtree( &pBTree);
|
||||
}
|
||||
|
||||
return( rc);
|
||||
}
|
||||
Reference in New Issue
Block a user