StormStorageMaker.cc

Go to the documentation of this file.

#define ML_DEBUG 1
#include "Utilities/StorageFactory/interface/StorageMaker.h"
#include "Utilities/StorageFactory/interface/StorageMakerFactory.h"
#include "Utilities/StorageFactory/interface/StorageFactory.h"
#include "Utilities/StorageFactory/interface/File.h"
#include "FWCore/Utilities/interface/EDMException.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include <cstdlib>
#include <cstdio>
#include <sys/stat.h>
 
class StormStorageMaker : public StorageMaker {
  /* getTURL: Executes a prepare to get script and extracts the physical file path */
  std::string getTURL(const std::string &surl) const {
    std::string client;
    if (char *p = std::getenv("CMS_STORM_PTG_CLIENT"))
      client = p;
    else
      throw cms::Exception("StormStorageMaker") << "$CMS_STORM_PTG_CLIENT has no value";
 
    // Command
    std::string comm(client + " srm:" + surl + " 2>&1");
    LogDebug("StormStorageMaker") << "command: " << comm << std::endl;
 
    FILE *pipe = popen(comm.c_str(), "r");
    if (!pipe)
      throw cms::Exception("StormStorageMaker") << "failed to execute PtG command: " << comm;
 
    // Get output
    int ch;
    std::string output;
    while ((ch = getc(pipe)) != EOF)
      output.push_back(ch);
    pclose(pipe);
 
    LogDebug("StormStorageMaker") << "output: " << output << std::endl;
 
    // Extract TURL if possible.
    size_t start = output.find("FilePath:", 0);
    if (start == std::string::npos)
      throw cms::Exception("StormStorageMaker") << "no turl found in command '" << comm << "' output:\n" << output;
 
    start += 9;
    std::string turl(output, start, output.find_first_of("\n", start) - start);
    LogDebug("StormStorageMaker") << "file to open: " << turl << std::endl;
    return turl;
  }
 
public:
  std::unique_ptr<Storage> open(const std::string &proto,
                                const std::string &surl,
                                int mode,
                                const AuxSettings &) const override {
    const StorageFactory *f = StorageFactory::get();
    StorageFactory::ReadHint readHint = f->readHint();
    StorageFactory::CacheHint cacheHint = f->cacheHint();
 
    if (readHint != StorageFactory::READ_HINT_UNBUFFERED || cacheHint == StorageFactory::CACHE_HINT_STORAGE)
      mode &= ~IOFlags::OpenUnbuffered;
    else
      mode |= IOFlags::OpenUnbuffered;
 
    std::string path = getTURL(surl);
    auto file = std::make_unique<File>(path, mode);
    return f->wrapNonLocalFile(std::move(file), proto, path, mode);
  }
 
  bool check(const std::string & /*proto*/,
             const std::string &path,
             const AuxSettings &,
             IOOffset *size = nullptr) const override {
    struct stat st;
    if (stat(getTURL(path).c_str(), &st) != 0)
      return false;
 
    if (size)
      *size = st.st_size;
 
    return true;
  }
};
 
DEFINE_EDM_PLUGIN(StorageMakerFactory, StormStorageMaker, "storm");