Пример #1
0
uint8_t *SparseVector::GetValPtr(int32_t key) {
  int32_t st = 0, end = size_ - 1;
  int32_t found_idx;
  bool found = Search(key, st, end, &found_idx);

  if (found)
    return GetValPtrByIdx(found_idx);
  else if (capacity_ < size_ + 1)
    return 0;
  else if (found_idx < 0) { // empty buffer
    GetKeyByIdx(0) = key;
    ++size_;
    return GetValPtrByIdx(0);
  } else if (GetKeyByIdx(found_idx) > key) {
    InsertAtIndex(found_idx, key);
    return GetValPtrByIdx(found_idx);
  } else if (GetKeyByIdx(found_idx) < key) {
    if (found_idx + 1 < size_) {
      InsertAtIndex(found_idx + 1, key);
    } else {
      GetKeyByIdx(found_idx + 1) = key;
      ++size_;
    }
    return GetValPtrByIdx(found_idx + 1);
  } else {
    LOG(FATAL) << "Impossible to reach here!";
  }
  return 0;
}
Пример #2
0
//返回清除的记录数,写log
ssize_t MemKeyCache::CleanNode(DUMP_MATCH_FUNC fMatchFunc,void* pArg)
{
	char szKey[MAX_KEY_LEN];

	ssize_t iCnt = 0;
	for (ssize_t iBucketIdx=0; iBucketIdx<m_pMemCacheHead->m_iBucketNum; iBucketIdx++)
	{
		ssize_t iNodeIdx = m_piBucket[iBucketIdx];
		while( iNodeIdx >= 0 )
		{
			ssize_t iKeyLen = GetKeyByIdx(iNodeIdx,szKey,MAX_KEY_LEN);
			ssize_t iDataFlag = GetFlag(iNodeIdx);

			//先指向下一个,然后删除
			iNodeIdx = m_NodeObjMng.GetDsIdx(iNodeIdx,m_pMemCacheHead->m_iDSSuffix);

			//符合条件的
			if(fMatchFunc(szKey,iKeyLen,iDataFlag,pArg) == 0)
			{
				Del(szKey,iKeyLen);
				iCnt++;
			}
		}
	}

	return iCnt;
}
Пример #3
0
//返回dump的字节数
ssize_t MemKeyCache::CoreDumpMem(char* pBuffer,const ssize_t BUFFSIZE,DUMP_MATCH_FUNC fDumpMatchFunc,void* pArg)
{
	ssize_t iBufferPos = 0;
	char* pTmpBuffer = new char[BUFFSIZE];

	for (ssize_t iBucketIdx=0; iBucketIdx<m_pMemCacheHead->m_iBucketNum; iBucketIdx++)
	{
		ssize_t iNodeIdx = m_piBucket[iBucketIdx];
		while( iNodeIdx >= 0 )
		{
			ssize_t iKeyLen = GetKeyByIdx(iNodeIdx,pTmpBuffer,BUFFSIZE);
			ssize_t iDataFlag = GetFlag(iNodeIdx);
			char *pKey = pTmpBuffer;

			//符合条件的
			if(fDumpMatchFunc(pKey,iKeyLen,iDataFlag,pArg) == 0)
			{
				ssize_t iBufferLen = m_BuffMng.GetBuffer(iNodeIdx,pTmpBuffer,BUFFSIZE);

				//存入总长度
				memcpy(pBuffer+iBufferPos,&iBufferLen,sizeof(ssize_t));
				iBufferPos += sizeof(ssize_t);

				//存入数据
				memcpy(pBuffer+iBufferPos,pTmpBuffer,iBufferLen);
				iBufferPos += iBufferLen;
			}
			iNodeIdx = m_NodeObjMng.GetDsIdx(iNodeIdx,m_pMemCacheHead->m_iDSSuffix);
		}
	}

	delete []pTmpBuffer;
	return iBufferPos;
}
Пример #4
0
void SparseVector::InsertAtIndex(int32_t index, int32_t key) {
  memmove(GetPtrByIdx(index + 1), GetPtrByIdx(index),
          (size_ - index)*(sizeof(int32_t) + value_size_));
  GetKeyByIdx(index) = key;
  ++size_;
}
Пример #5
0
uint8_t* SparseVector::GetByIdx(int32_t index, int32_t *key) {
  *key = GetKeyByIdx(index);
  return GetValPtrByIdx(index);
}