mdbhtindex.cpp

一个自行开发的内存数据管理的包

//----------------------------------------------------------------------------
// 程序名称：   HtIndex.cpp
// 程序说明：   CTable类实现
// 程序作者：   陈立峰
// 程序版本：   1.0
// 开始日期：   2005-06-13
//----------------------------------------------------------------------------
#include "Exception.h"
#include "MdbHtIndex.h"
#include <iostream>
#include <strings.h>

using namespace std;
//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
CMdbHtIndex::CMdbHtIndex(MALLOC *pcAlloc,CMdbDict *pcDbDict,CMdbTbl *pcTable,CDbIndexMeta *psIndexMeta):m_pcAlloc(pcAlloc),m_pcDbDict(pcDbDict),m_psIndexMeta(psIndexMeta),m_pcTable(pcTable)
{
	void *ptr;
	if((ptr = psIndexMeta->m_psIndexEntry) == NULL)
	{
		if((ptr = m_pcAlloc->malloc(sizeof(CHtIndexEntry))) == NULL)
		{
	  		throwException(MDB_EC_NORMAL,MDB_ET_MEM_ALLOC_ERROR,"CMdbHtIndex::CMdbHtIndex Error,Can not Alloc Memory for CHtIndexEntry,Idx:%s\n",m_psIndexMeta->m_szIndexName);
	  	}
//		m_psIndexEntry = reinterpret_cast<CHtIndexEntry *>(ptr);
		m_psIndexEntry = new (ptr) CHtIndexEntry();
		for(int i=0;i<HT_BUCKET_CNT;i++)
		{
			(m_psIndexEntry->m_psHtIndexNodes)[i] = (CHtIndexNode *)NULL;
		}
		psIndexMeta->m_psIndexEntry = reinterpret_cast<void *>(m_psIndexEntry);
	}
	else
	{
		m_psIndexEntry = reinterpret_cast<CHtIndexEntry *>(ptr);
	}
}


//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
void CMdbHtIndex::insert(CRecordNode *pRec)
{
	unsigned long nHashVal = 0;
	for(int i=0;i<m_pcDbDict->getIndexColCnt(m_psIndexMeta);i++)
	{
		CDbColumnMeta *psColumnMeta = m_pcDbDict->getIndexColMeta(m_psIndexMeta,i);
		int nFieldType;
		UVal uVal;
		m_pcTable->getAutoFieldValByMeta(pRec,psColumnMeta,nFieldType,(void *)&uVal);
		nHashVal += calcInnerHashVal(nFieldType,&(uVal));
	}
	//cerr << "insert nHashVal:" << nHashVal;
	unsigned long nBucket = nHashVal % HT_BUCKET_CNT;

	void *ptr;
	if((ptr = m_pcAlloc->malloc(sizeof(CHtIndexNode))) == NULL)
	{
	  	throwException(MDB_EC_NORMAL,MDB_ET_MEM_ALLOC_ERROR,"CMdbHtIndex::insert Error,Can not Alloc Memory for CHtIndexNode,Idx:%s\n",m_psIndexMeta->m_szIndexName);
	}
//	CHtIndexNode *psIndexNode = reinterpret_cast<CHtIndexNode *>(ptr);
	CHtIndexNode *psIndexNode = new (ptr) CHtIndexNode();
	psIndexNode->m_psRecord = pRec;
	psIndexNode->m_psNext = (m_psIndexEntry->m_psHtIndexNodes)[nBucket];
	(m_psIndexEntry->m_psHtIndexNodes)[nBucket] = psIndexNode;
}

//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
unsigned long CMdbHtIndex::calcHashVal(int nFieldType,const void *pVal)
{
	
	if(nFieldType == FT_INT8)
	{
		char c = *(char *)pVal;
		if(c < 0)
			return (unsigned long)(-c);
		else
			return (unsigned long)c;
	}
	else if(nFieldType == FT_UINT8)
	{
		return (unsigned long)(*(unsigned char *)pVal);
	}
	else if(nFieldType == FT_INT16)
	{
		short s = *(short *)pVal;
		if(s < 0)
			return (unsigned long)(-s);
		else
			return (unsigned long)s;
	}
	else if(nFieldType == FT_UINT16)
	{
		return (unsigned long)(*(unsigned short *)pVal);
	}
	else if(nFieldType == FT_INT32)
	{
		int i = *(int *)pVal;
		if(i < 0)
			return (unsigned long)(-i);
		else
			return (unsigned long)i;
	}
	else if(nFieldType == FT_UINT32)
	{
		return (unsigned long)(*(unsigned int *)pVal);
	}
	else if(nFieldType == FT_INT64)
	{
		long long ll = *(long long *)pVal;
		if(ll < 0)
			return (unsigned long)(-ll);
		else
			return (unsigned long)ll;
	}
	else if(nFieldType == FT_INT64 || nFieldType == FT_UINT64)
	{
		return (unsigned long)(*(unsigned long long *)pVal);  //保留低32位,2^32
	}
	else if(nFieldType == FT_STRING)
	{
		unsigned char *pChar = (unsigned char *)pVal;
		//cerr << "*p:" << p << endl;
		unsigned long nHashVal = 0;
		if(pChar)
		{
			while(*pChar)
			{
				nHashVal = nHashVal*4 + (unsigned long)(*pChar);
				pChar++;
			}
		}
		return nHashVal;
	}
	else
	{
		throwException(MDB_EC_NORMAL,MDB_ET_INVALID_FIELD_TYPE,"CMdbHtIndex::calcHashVal Error,Invalid nFieldType,Idx:%s,FieldType:%d\n",m_psIndexMeta->m_szIndexName,nFieldType);
		return 0;
	}
}

//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
unsigned long CMdbHtIndex::calcInnerHashVal(int nFieldType,const UVal *pVal)
{
	if(nFieldType == FT_INT8)
	{
		char c = *(char *)pVal;
		if(c < 0)
			return (unsigned long)(-c);
		else
			return (unsigned long)c;
	}
	else if(nFieldType == FT_INT16)
	{
		short s = *(short *)pVal;
		if(s < 0)
			return (unsigned long)(-s);
		else
			return (unsigned long)s;
	}
	else if(nFieldType == FT_INT32)
	{
		int i = *(int *)pVal;
		if(i < 0)
			return (unsigned long)(-i);
		else
			return (unsigned long)i;
	}
	else if(nFieldType == FT_INT64)
	{
		long long ll = *(long long *)pVal;
		if(ll < 0)
			return (unsigned long)(-ll);
		else
			return (unsigned long)ll;
	}
	else if(nFieldType == FT_STRING)
	{
		unsigned char *pChar = *(unsigned char **)pVal;
		unsigned long nHashVal = 0;
		if(pChar)
		{
			while(*pChar)
			{
				nHashVal = nHashVal *4 + (unsigned long)(*pChar);
				pChar++;
			}
		}
		return nHashVal;
	}
	else
	{
		throwException(MDB_EC_NORMAL,MDB_ET_INVALID_FIELD_TYPE,"CMdbHtIndex::calcInnerHashVal Error,Invalid nFieldType,Idx:%s,FieldType:%d\n",m_psIndexMeta->m_szIndexName,nFieldType);
		return 0;
	}
}

/*
//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
CRecordNode *CMdbHtIndex::query(CHtIndexNode *&psPrevIndexNode,int nFieldCnt,const char **szFieldNames,int *pnFieldTypes,void **ppVals,bool &bHit)
{
	CHtIndexNode *psIndexNode;
	if(!psPrevIndexNode)
	{
		if(m_pcDbDict->getIndexColCnt(m_psIndexMeta) != nFieldCnt)
		{
			bHit = false;
			return NULL;   //查询字段个数不同于索引字段个数
		}
		unsigned long nHashVal = 0;
		for(int i=0;i<m_pcDbDict->getIndexColCnt(m_psIndexMeta) ;i++)
		{
			CDbColumnMeta *psColumnMeta = m_pcDbDict->getIndexColMeta(m_psIndexMeta,i);
			int j;
			for(j=0;j<nFieldCnt;j++)
			{
				if(strcasecmp(szFieldNames[j],psColumnMeta->m_szColName) == 0)
				{
					break;
				}
			}
			if(j == nFieldCnt)
			{
				bHit = false;
				return NULL;  //查询字段名不同于索引字段名
			}
			else
			{
				//cerr << "query char *" << endl;
				nHashVal += calcHashVal(pnFieldTypes[j],ppVals[j]);
			}
		}
		//cerr << "nQueryHashVal:" << nHashVal << endl;
		unsigned long nBucket = nHashVal % HT_BUCKET_CNT;
		LOG4CPLUS_DEBUG(logger, "nBucket: " << nBucket << ", nHashVal: " << nHashVal);		
		//cerr << "nBucket:" << nBucket << endl;
		psIndexNode = (m_psIndexEntry->m_psHtIndexNodes)[nBucket];
	}
	else
	{
		psIndexNode = psPrevIndexNode->m_psNext;
	}
	
	for(int i=0;i<m_pcDbDict->getIndexColCnt(m_psIndexMeta);i++)
	{
		CDbColumnMeta *psColumnMeta = m_pcDbDict->getIndexColMeta(m_psIndexMeta,i);
		int j;
		for(j=0;j<nFieldCnt;j++)
		{
			if(strcasecmp(szFieldNames[j],psColumnMeta->m_szColName) == 0)
			{
				break;
			}
		}
		if(j == nFieldCnt)
		{
			throwException(MDB_EC_NORMAL,MDB_ET_LOGIC_ERROR,"CMdbHtIndex::query Logic Error,Idx:%s\n",m_psIndexMeta->m_szIndexName);
		}
		while(psIndexNode)
		{
			LOG4CPLUS_DEBUG(logger, "in psIndexNode");
			psPrevIndexNode = psIndexNode;
			
			int nFieldType;
			UVal uVal;
			CRecordNode *psRecordNode = psIndexNode->m_psRecord;
			m_pcTable->getAutoFieldValByMeta(psRecordNode,psColumnMeta,nFieldType,(void *)&uVal);
			LOG4CPLUS_DEBUG(logger, "i: " << i << ", j: " << j);
			if(compareKey(nFieldType,ppVals[j],psColumnMeta->m_szColName,psColumnMeta->m_nInternalType,&uVal) == false)
			{
				LOG4CPLUS_DEBUG(logger, "compareKey failed");
				psIndexNode = psIndexNode->m_psNext;
				continue;
			}
			else
			{
				//cerr << "one fieldMatch" << endl;
				break;
			}
		}
		if(!psIndexNode)
		{
			bHit = true;
			return NULL;
		}
	}
	bHit = true;
	return (CRecordNode *)(psIndexNode->m_psRecord);
}
*/

//----------------------------------------------------------------------------
// 函数原型： CWorkFlow::CWorkFlow(int nBucketNum)
// 函数功能： CWorkFlow    构造函数
// 传入参数： 无
// 传出参数： 无
// 函数返回： 无
// 注意事项： 无
//---------------------------------------------------------------------------- 
CRecordNode *CMdbHtIndex::query(CHtIndexNode *&psPrevIndexNode,int nFieldCnt,const char **szFieldNames,int *pnFieldTypes,void **ppVals,bool &bHit)
{
	CHtIndexNode *psIndexNode;
	if(!psPrevIndexNode)
	{
		if(m_pcDbDict->getIndexColCnt(m_psIndexMeta) != nFieldCnt)
		{
			bHit = false;
			return NULL;   //查询字段个数不同于索引字段个数
		}
		unsigned long nHashVal = 0;
		for(int i=0;i<m_pcDbDict->getIndexColCnt(m_psIndexMeta) ;i++)
		{
			CDbColumnMeta *psColumnMeta = m_pcDbDict->getIndexColMeta(m_psIndexMeta,i);
			int j;
			for(j=0;j<nFieldCnt;j++)
			{
				if(strcasecmp(szFieldNames[j],psColumnMeta->m_szColName) == 0)
				{
					break;
				}
			}
			if(j == nFieldCnt)
			{
				bHit = false;
				return NULL;  //查询字段名不同于索引字段名
			}
			else
			{
				//cerr << "query char *" << endl;
				nHashVal += calcHashVal(pnFieldTypes[j],ppVals[j]);
			}
		}
		//cerr << "nQueryHashVal:" << nHashVal << endl;
		unsigned long nBucket = nHashVal % HT_BUCKET_CNT;
		//cerr << "nBucket:" << nBucket << endl;
		psIndexNode = (m_psIndexEntry->m_psHtIndexNodes)[nBucket];
	}
	else
	{
		psIndexNode = psPrevIndexNode->m_psNext;
	}

	for(;psIndexNode;psIndexNode = psIndexNode->m_psNext)
	{
		psPrevIndexNode = psIndexNode;
		bool bFound = true;
		for(int i=0;i<m_pcDbDict->getIndexColCnt(m_psIndexMeta);i++)
		{
			//匹配字段名
			CDbColumnMeta *psColumnMeta = m_pcDbDict->getIndexColMeta(m_psIndexMeta,i);
			int j;
			for(j=0;j<nFieldCnt;j++)