CrestFunctions.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
// @file CrestFunctions.cxx
// Implementation for CrestFunctions utilities
// @author Shaun Roe
// @date 1 July 2019
 
#include "CrestFunctions.h"
#include "CrestApi/CrestApi.h"
#include "CrestApi/CrestApiFs.h"
#include <iostream>
#include <exception>
#include <regex>
#include "IOVDbStringFunctions.h"
#include <string>
#include <algorithm>
#include <map>
 
namespace IOVDbNamespace{
 
  using namespace Crest;
 
  CrestFunctions::CrestFunctions(const std::string & crest_path){
    setURLBase(crest_path);
 
    const std::string prefix1 = "http://";
    const std::string prefix2 = "https://";
    if (crest_path.starts_with(prefix1) || crest_path.starts_with(prefix2)){
      m_crestCl = std::unique_ptr<Crest::CrestApi>(new Crest::CrestApi(getURLBase()));
    }
    else{
      m_crestCl = std::unique_ptr<Crest::CrestApiFs>(new Crest::CrestApiFs(true,getURLBase()));
    }
  }
 
  const std::string &
  CrestFunctions::getURLBase(){
    return m_CREST_PATH;
  }
 
  void 
  CrestFunctions::setURLBase(const std::string & crest_path){
    m_CREST_PATH = crest_path;
  }
 
 
  std::string
  CrestFunctions::extractHashFromJson(const std::string & jsonReply){
    std::string hash{};
    try{
      std::string_view signature="payloadHash\":\"";
      auto signaturePosition=jsonReply.rfind(signature);
      if (signaturePosition == std::string::npos) throw std::runtime_error("signature "+std::string(signature)+" not found");
      auto startOfHash=signaturePosition + signature.size();
      auto endOfHash=jsonReply.find('\"',startOfHash);
      auto len=endOfHash-startOfHash;
      if (startOfHash > jsonReply.size()) throw std::runtime_error("Hash start is beyond end of string");
      hash=jsonReply.substr(startOfHash, len);
    } catch (std::exception & e){
      std::cerr<<__FILE__<<":"<<__LINE__<< ": "<<e.what()<<" while trying to find the hash in "<<jsonReply<<std::endl;
    }
    return hash;
  }
 
  
  std::string 
  CrestFunctions::getLastHashForTag(const std::string & tag){
    char tu[] = "";
    strfry(tu);
    std::string reply = "";
 
    try{
      IovSetDto dto = m_crestCl->selectIovs(tag, 0, -1, 0, 10000, 0, "id.since:ASC");
 
      nlohmann::json iov_data = dto.toJson();
      nlohmann::json iov_list = getResources(iov_data);
      reply = iov_list.dump();
    } catch (std::exception & e){
      std::cerr<<__FILE__<<":"<<__LINE__<< ": "<<e.what()<<" while trying to find the IOVs"<<std::endl;
      return "";
    }
 
    return extractHashFromJson(reply);
  }
 
  std::string 
  CrestFunctions::getPayloadForHash(const std::string & hash){
    std::string reply = "";  
    
    try{
    reply = m_crestCl->getPayload(hash);
    } catch (std::exception & e){
        std::cerr<<__FILE__<<":"<<__LINE__<< ": "<<e.what()<<" while trying to find the payload"<<std::endl;
        return "";
    }
    
    return reply;
  }
 
 
  nlohmann::json 
  CrestFunctions::getResources(nlohmann::json& js) {
    nlohmann::json js2 = json::array();
    nlohmann::json result = js.value("resources", js2);
    return result;
  }
 
  std::vector<uint64_t>
  CrestFunctions::getIovGroups(const std::string & tag){
    std::vector<uint64_t> v;
    try{
      IovSetDto dto = m_crestCl->selectGroups(tag, 0, 10000, 0, "id.since:ASC");
      const std::vector<IovDto> & res = dto.getResources();
      for (const IovDto & item_iov: res){
        v.emplace_back(item_iov.getSince());
      }
    } catch (std::exception & e){
      std::cerr<<__FILE__<<":"<<__LINE__<< ": "<<e.what()<<" while trying to find the IOVs"<<std::endl;
      return {};
    }
    return v;
  }
 
 
  std::pair<uint64_t,uint64_t>
  CrestFunctions::getSinceUntilPair(const std::vector<uint64_t>& v, const uint64_t since, const uint64_t until){
    uint64_t new_since = 0;
    uint64_t new_until = 0;
    std::pair<uint64_t,uint64_t> answer = std::make_pair(0,0);
 
    if (until < since){
    std::cerr << "Wrong since/until." << std::endl;
    return answer;
    }
 
    int N = v.size();
    for (int i = 0; i < N; i++) {
      if(v[i] <= since && since < v[i+1]){
    new_since = v[i];
    break;
      }
    }
 
    for (int i = 0; i < N; i++) {
      if(v[i] < until && until <= v[i+1]){
    new_until = v[i+1];
    break;
      }
    }    
 
    answer = std::make_pair(new_since,new_until);
    return answer; 
  }
 
 
  int CrestFunctions::getTagSize(const std::string& tagname){
    int res = 0; 
    try{
      res = m_crestCl->getSize(tagname);
    } catch (std::exception & e){
      std::cerr<<__FILE__<<":"<<__LINE__<< ": " << e.what() << " Cannot get the tag size for " << tagname << std::endl;
    }
    return res;
  }
 
  std::pair<uint64_t,uint64_t>
  CrestFunctions::getIovInterval(const std::string&  tag, const uint64_t since, const uint64_t until){
    std::vector<uint64_t> v = getIovGroups(tag);
    v.push_back(std::numeric_limits<uint64_t>::max()); // added "infinity" as the last item
    return getSinceUntilPair(std::move(v), since, until);
  }
 
 
  std::vector<IovHashPair>
  CrestFunctions::getIovsForTag(const std::string & tag, uint64_t since, uint64_t until){
    std::vector<IovHashPair> iovHashPairs;
    int iovNumber = getTagSize(tag);
    try{
      IovSetDto dto;
      if (iovNumber <=1000) {
        dto = m_crestCl->selectIovs(tag, 0, -1, 0, 10000, 0, "id.since:ASC");
      } else {
        const auto &[s_time, u_time] = getIovInterval(tag, since, until);
        if (s_time == 0 && u_time == 0){ // data out of range
            return iovHashPairs;
          } else {
          dto = m_crestCl->selectIovs(tag, s_time, u_time, 0, 10000, 0, "id.since:ASC");
          }
      }
      std::vector<IovDto> res = dto.getResources();
      std::map<uint64_t, std::string> hashmap;
      for (const IovDto & item: res) {
    hashmap[item.getSince()] = item.getPayloadHash();
      } 
      for (auto& t : hashmap){
        iovHashPairs.emplace_back(std::to_string(t.first),t.second);
      }
    } catch (std::exception & e){
      std::cerr<<__FILE__<<":"<<__LINE__<< ": "<<e.what()<<" while trying to find the IOVs"<<std::endl;
      return {};
    }
    return iovHashPairs;
  }
}