Function writeScript

src/restartscript.cpp:342–621

;

string
writeScript(const string &ckptDir,
            bool uniqueCkptFilenames,
            const time_t &ckptTimeStamp,
            const uint32_t theCheckpointInterval,
            const int thePort,
            const UniquePid &compId,
            const map<string, vector<string> > &restartFilenames,
            const map<string, vector<string> >& rshCmdFileNames,
            const map<string, vector<string> >& sshCmdFileNames)
{
  ostringstream o;
  string uniqueFilename;

  o << string(ckptDir) << "/"
    << RESTART_SCRIPT_BASENAME << "_" << compId;
  if (uniqueCkptFilenames) {
    o << "_" << std::setw(5) << std::setfill('0') <<
        compId.computationGeneration();
  }
  o << "." << RESTART_SCRIPT_EXT;
  uniqueFilename = o.str();

  const bool isSingleHost = ((rshCmdFileNames.size() == 0) && (sshCmdFileNames.size() == 0) && (restartFilenames.size() == 1));

  map<string, vector<string> >::const_iterator host;

  size_t numPeers = 0;
  for (host = restartFilenames.begin();
       host != restartFilenames.end();
       host++) {
    numPeers += host->second.size();
  }
  for (host = rshCmdFileNames.begin();
       host != rshCmdFileNames.end();
       host++) {
    numPeers += host->second.size();
  }
  for (host = sshCmdFileNames.begin();
       host != sshCmdFileNames.end();
       host++) {
    numPeers += host->second.size();
  }


  vector<string>::const_iterator file;

  char hostname[80];
  char timestamp[80];
  gethostname(hostname, 80);

  JTRACE("writing restart script") (uniqueFilename);

  FILE *fp = fopen(uniqueFilename.c_str(), "w");
  JASSERT(fp != 0)(JASSERT_ERRNO)(uniqueFilename)
  .Text("failed to open file");

  fprintf(fp, "%s", header);