DTConfigPluginHandler.cc

Go to the documentation of this file.

/*
 *  See header file for a description of this class.
 *
 *  \author Paolo Ronchese INFN Padova
 *
 */

//-----------------------
// This Class' Header --
//-----------------------
#include "CondCore/DTPlugins/interface/DTConfigPluginHandler.h"

//-------------------------------
// Collaborating Class Headers --
//-------------------------------
#include "CondFormats/DTObjects/interface/DTKeyedConfig.h"
#include "CondFormats/DataRecord/interface/DTKeyedConfigListRcd.h"
#include "CondCore/DBOutputService/interface/KeyedElement.h"
#include "CondCore/CondDB/interface/KeyList.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"

//---------------
// C++ Headers --
//---------------
#include <cstdio>
#include <iostream>

//-------------------
// Initializations --
//-------------------
const int DTConfigPluginHandler::maxBrickNumber  = 5000;
const int DTConfigPluginHandler::maxStringNumber = 100000;
const int DTConfigPluginHandler::maxByteNumber   = 10000000;
//DTConfigPluginHandler::handler_map DTConfigPluginHandler::handlerMap;

//----------------
// Constructors --
//----------------
DTConfigPluginHandler::DTConfigPluginHandler():
  cachedBrickNumber(  0 ),
  cachedStringNumber( 0 ),
  cachedByteNumber(   0 ) {
//  std::cout << "===============================" << std::endl;
//  std::cout << "=                             =" << std::endl;
//  std::cout << "=  new DTConfigPluginHandler  =" << std::endl;
//  std::cout << "=                             =" << std::endl;
//  std::cout << "===============================" << std::endl;
//  if ( instance == 0 ) instance = this;
}


//--------------
// Destructor --
//--------------
DTConfigPluginHandler::~DTConfigPluginHandler() {
  purge();
}


//--------------
// Operations --
//--------------
void DTConfigPluginHandler::build() {
  if ( instance == 0 ) instance = new DTConfigPluginHandler;
}


int DTConfigPluginHandler::get( const edm::EventSetup& context,
                                int cfgId, const DTKeyedConfig*& obj ) {
  return get( context.get<DTKeyedConfigListRcd>(), cfgId, obj );
}


int DTConfigPluginHandler::get( const DTKeyedConfigListRcd& keyRecord,
                                int cfgId, const DTKeyedConfig*& obj ) {

  bool cacheFound = false;
  int cacheAge = 999999999;
  std::map<int,counted_brick>::iterator cache_iter = brickMap.begin();
  std::map<int,counted_brick>::iterator cache_icfg = brickMap.find( cfgId );
  std::map<int,counted_brick>::iterator cache_iend = brickMap.end();
  if ( cache_icfg != cache_iend ) {
    std::pair<const int,counted_brick>& entry = *cache_icfg;
    counted_brick& cBrick = entry.second;
    cacheAge = cBrick.first;
    obj = cBrick.second;
    cacheFound = true;
  }

  std::map<int,const DTKeyedConfig*> ageMap;
  if ( cacheFound ) {
    if ( !cacheAge ) return 0;
    while ( cache_iter != cache_iend ) {
      std::pair<const int,counted_brick>& entry = *cache_iter++;
      counted_brick& cBrick = entry.second;
      int& brickAge = cBrick.first;
      if ( brickAge < cacheAge ) brickAge++;
      if ( entry.first == cfgId ) brickAge = 0;
    }
    return 0;
  }
  else {
    while ( cache_iter != cache_iend ) {
      std::pair<const int,counted_brick>& entry = *cache_iter++;
      counted_brick& cBrick = entry.second;
      ageMap.insert( std::pair<int,const DTKeyedConfig*>( 
                     ++cBrick.first, entry.second.second ) );
    }
  }

// get dummy brick list
  edm::ESHandle<cond::persistency::KeyList> klh;
  keyRecord.get( klh );
  cond::persistency::KeyList const &  kl= *klh.product();
  cond::persistency::KeyList* keyList = const_cast<cond::persistency::KeyList*>( &kl );
  if ( keyList == 0 ) return 999;

  std::vector<unsigned long long> checkedKeys;
  boost::shared_ptr<DTKeyedConfig> kBrick;
  checkedKeys.push_back( cfgId );
  bool brickFound = false;
  try {
    keyList->load( checkedKeys );
    kBrick = keyList->get<DTKeyedConfig>( 0 );
    if ( kBrick.get() ) brickFound = ( kBrick->getId() == cfgId );
  }
  catch ( std::exception const & e ) {
  }
  if ( brickFound ) {
    counted_brick cBrick( 0, obj = new DTKeyedConfig( *kBrick ) );
    brickMap.insert( std::pair<int,counted_brick>( cfgId, cBrick ) );
    DTKeyedConfig::data_iterator d_iter = kBrick->dataBegin();
    DTKeyedConfig::data_iterator d_iend = kBrick->dataEnd();
    cachedBrickNumber++;
    cachedStringNumber += ( d_iend - d_iter );
    while ( d_iter != d_iend ) cachedByteNumber += ( *d_iter++ ).size();
  }
  std::map<int,const DTKeyedConfig*>::reverse_iterator iter = ageMap.rbegin();
  while ( ( cachedBrickNumber  > maxBrickNumber  ) ||
          ( cachedStringNumber > maxStringNumber ) ||
          ( cachedByteNumber   > maxByteNumber   ) ) {
    const DTKeyedConfig* oldestBrick = iter->second;
    int oldestId = oldestBrick->getId();
    cachedBrickNumber--;
    DTKeyedConfig::data_iterator d_iter = oldestBrick->dataBegin();
    DTKeyedConfig::data_iterator d_iend = oldestBrick->dataEnd();
    cachedStringNumber -= ( d_iend - d_iter );
    while ( d_iter != d_iend ) cachedByteNumber -= ( *d_iter++ ).size();
    brickMap.erase( oldestId );
    delete iter->second;
    iter++;
  }

  return 999;

}


void DTConfigPluginHandler::getData( const edm::EventSetup& context, int cfgId,
                                     std::vector<std::string>& list ) {
  getData( context.get<DTKeyedConfigListRcd>(), cfgId, list );
  return;
}


void DTConfigPluginHandler::getData( const DTKeyedConfigListRcd& keyRecord,
                                     int cfgId,
                                     std::vector<std::string>& list ) {
  const DTKeyedConfig* obj = 0;
  get( keyRecord, cfgId, obj );
  if ( obj == 0 ) return; 
  DTKeyedConfig::data_iterator d_iter = obj->dataBegin();
  DTKeyedConfig::data_iterator d_iend = obj->dataEnd();
  while ( d_iter != d_iend ) list.push_back( *d_iter++ );
  DTKeyedConfig::link_iterator l_iter = obj->linkBegin();
  DTKeyedConfig::link_iterator l_iend = obj->linkEnd();
  while ( l_iter != l_iend ) getData( keyRecord, *l_iter++, list );
  return;
}


void DTConfigPluginHandler::purge() {
  std::cout << "DTConfigPluginHandler::purge "
            << this << " "
            << cachedBrickNumber  << " "
            << cachedStringNumber << " "
            << cachedByteNumber   << std::endl;
  std::map<int,counted_brick>::const_iterator iter = brickMap.begin();
  std::map<int,counted_brick>::const_iterator iend = brickMap.end();
  while ( iter != iend ) {
    delete iter->second.second;
    iter++;
  }
  brickMap.clear();
  cachedBrickNumber  = 0;
  cachedStringNumber = 0;
  cachedByteNumber   = 0;
  return;
}