configcache.cpp

   1 #include "configcache.h"
   2
   3 #include <stdlib.h>
   4 #include <iostream>
   5 #include <time.h>
   6 #include <dirent.h>
   7
   8 configcache::configcache(const string& cacheid, const int& nequi, const int& nskip, const string& datadir, char **configmem, const int& configMemSize, const int& cachemode){
   9   NEQUI = nequi;
  10   NSKIP = nskip;
  11   DATADIR = datadir;
  12   CACHEID = cacheid;
  13
  14   configMem = (char*)malloc(configMemSize);
  15   tmpConfig = (char*)malloc(configMemSize);
  16
  17   *configmem = configMem;
  18   configSize = configMemSize;
  19
  20   outBuffer = NULL;
  21   inBuffer = NULL;
  22
  23   MODE = cachemode;
  24
  25   refetchDataFiles = false;
  26 }
  27
  28 string configcache::getFileId(const bool& shortid)
  29 {
  30   stringstream fileid;
  31
  32   if(!shortid) fileid << CACHEID << "_" << NEQUI << "_" << NSKIP;
  33   for(int ipara=0; ipara<Paras.size(); ipara++)
  34     fileid << "_" << Paras[ipara].id << Paras[ipara].val;
  35
  36   return fileid.str();
  37 }
  38
  39 void configcache::fetchDataFiles()
  40 {
  41   struct dirent *de=NULL;
  42   DIR *d=NULL;
  43
  44   d=opendir(DATADIR.c_str());
  45   if(d != NULL){
  46     while(de = readdir(d)){
  47       string filename = de->d_name;
  48       if(isValidInFile(filename))
  49         {
  50           inFiles.push_back(filename);
  51         }
  52     }
  53   }
  54 }
  55
  56 bool configcache::isValidInFile(const string& infile)
  57 {
  58   char *inchar, *inParts;
  59   string truncIn, truncOut;
  60
  61   if( infile.size() < 4 ) return false;
  62
  63   if( infile.substr(infile.size()-3) != "dat" ) return false;
  64
  65   inchar = new char [infile.size()+1];
  66   strcpy (inchar, infile.c_str());
  67
  68   inParts = strtok( inchar, "_" );
  69   for(int iPart=0; inParts!=NULL; iPart++)
  70     {
  71       if( iPart>3 ) { truncIn += "_"; truncIn += inParts; }
  72
  73       switch(iPart)
  74         {
  75         case 1: if(inParts != CACHEID) return false; break;
  76         case 2: if(atoi(inParts) < NEQUI) return false; break;
  77         case 3: if(atoi(inParts) < NSKIP) return false; break;
  78         }
  79       inParts = strtok( NULL, "_");
  80     }
  81   truncIn = truncIn.substr(0, truncIn.size()-4);
  82
  83   delete[] inchar;
  84
  85   if( truncIn.find( getFileId(true) + "_" ) == string::npos ) return false;
  86
  87   return true;
  88 }
  89
  90 bool configcache::readConfig()
  91 {
  92   if(DATADIR == "" || MODE == CACHE_MODE_DISABLED) return false;
  93
  94   if(refetchDataFiles){
  95     refetchDataFiles = false;
  96     fetchDataFiles();
  97   }
  98
  99   while(true)
 100     {
 101       if( (!inFile.is_open()) && inFiles.size() == 0 ) return false;
 102
 103       while( (!inFile.is_open()) && inFiles.size() > 0 )
 104         {
 105           inFile.open( (DATADIR + "/" + inFiles.back()).c_str(), std::ios::binary );
 106           inFiles.pop_back();
 107
 108           if( !inFile.is_open() ) continue;
 109
 110           inBuffer = new boost::iostreams::filtering_istreambuf;
 111           inBuffer->push( boost::iostreams::bzip2_decompressor() );
 112           inBuffer->push(inFile);
 113         }
 114
 115       if( inFile.is_open() )
 116         {
 117           if( boost::iostreams::read(*inBuffer, tmpConfig, configSize) == configSize )
 118             {
 119               memcpy(configMem, tmpConfig, configSize);
 120               return true;
 121             }
 122           else inFile.close();
 123         }
 124     }
 125 }
 126
 127 void configcache::writeConfig()
 128 {
 129   if( DATADIR == "" || MODE < 2 ) return;
 130
 131   if(!outFile.is_open()){
 132     time_t secstamp = time(NULL);
 133
 134     outFileName.str("");
 135     outFileName << DATADIR << "/" << secstamp << "_" << getFileId() << "_.dat.tmp";
 136     outFile.open( outFileName.str().c_str(), std::ios::binary );
 137
 138     outBuffer = new boost::iostreams::filtering_ostreambuf;
 139     outBuffer->push(boost::iostreams::bzip2_compressor());
 140     outBuffer->push(outFile);
 141   }
 142
 143   boost::iostreams::write(*outBuffer, configMem, configSize);
 144 }
 145
 146 void configcache::addPara(const string& parid, const double& val){
 147   parameter newPara;
 148   newPara.id = parid;
 149   newPara.val = val;
 150   Paras.push_back(newPara);
 151 }
 152
 153 int configcache::getParIndex(const string& parid){
 154   for(int ipara=0; ipara<Paras.size(); ipara++)
 155     if(Paras[ipara].id == parid) return ipara;
 156 }
 157
 158 void configcache::setPara(const string& parid, const double& value){
 159   Paras[getParIndex(parid)].val = value;
 160   finishOutFile();
 161   if(inBuffer != NULL) { delete inBuffer; inBuffer=NULL; }
 162   inFile.close();
 163   inFiles.clear();
 164
 165   refetchDataFiles = true;
 166 }
 167
 168 configcache::~configcache()
 169 {
 170   finishOutFile();
 171   delete inBuffer;
 172   inBuffer = NULL;
 173 }
 174
 175 void configcache::finishOutFile()
 176 {
 177   if( outBuffer != NULL )
 178     {
 179       delete outBuffer;
 180       outBuffer = NULL;
 181     }
 182
 183   if( outFile.is_open() )
 184     {
 185       outFile.close();
 186       rename( outFileName.str().c_str(), outFileName.str().substr(0, outFileName.str().size()-4).c_str() );
 187     }
 188 }