File indexing completed on 2024-04-06 12:31:54
0001
0002 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0003 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0004 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0005 #include "FWCore/MessageLogger/interface/JobReport.h"
0006 #include "FWCore/ServiceRegistry/interface/Service.h"
0007 #include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
0008
0009 #include "XrdRequest.h"
0010 #include "XrdStatistics.h"
0011
0012 #include <chrono>
0013
0014 using namespace edm::storage;
0015
0016 namespace xrd_adaptor {
0017 XrdStatistics::~XrdStatistics() {}
0018 }
0019
0020 using namespace XrdAdaptor;
0021
0022 std::atomic<XrdSiteStatisticsInformation *> XrdSiteStatisticsInformation::m_instance;
0023
0024 XrdStatisticsService::XrdStatisticsService(const edm::ParameterSet &iPS, edm::ActivityRegistry &iRegistry) {
0025 XrdSiteStatisticsInformation::createInstance();
0026
0027 if (iPS.getUntrackedParameter<bool>("reportToFJR", false)) {
0028 iRegistry.watchPostEndJob(this, &XrdStatisticsService::postEndJob);
0029 }
0030 }
0031
0032 void XrdStatisticsService::postEndJob() {
0033 edm::Service<edm::JobReport> reportSvc;
0034 if (!reportSvc.isAvailable()) {
0035 return;
0036 }
0037
0038 XrdSiteStatisticsInformation *instance = XrdSiteStatisticsInformation::getInstance();
0039 if (!instance) {
0040 return;
0041 }
0042
0043 std::map<std::string, std::string> props;
0044 for (auto &stats : instance->m_sites) {
0045 stats->recomputeProperties(props);
0046 reportSvc->reportPerformanceForModule(stats->site(), "XrdSiteStatistics", props);
0047 }
0048 }
0049
0050 std::vector<std::pair<std::string, XrdStatisticsService::CondorIOStats>> XrdStatisticsService::condorUpdate() {
0051 std::vector<std::pair<std::string, XrdStatisticsService::CondorIOStats>> result;
0052 XrdSiteStatisticsInformation *instance = XrdSiteStatisticsInformation::getInstance();
0053 if (!instance) {
0054 return result;
0055 }
0056
0057 std::lock_guard<std::mutex> lock(instance->m_mutex);
0058 result.reserve(instance->m_sites.size());
0059 for (auto &stats : instance->m_sites) {
0060 CondorIOStats cs;
0061 std::shared_ptr<XrdSiteStatistics> ss = get_underlying_safe(stats);
0062 if (!ss)
0063 continue;
0064 cs.bytesRead = ss->getTotalBytes();
0065 cs.transferTime = ss->getTotalReadTime();
0066 result.emplace_back(ss->site(), cs);
0067 }
0068 return result;
0069 }
0070
0071 std::shared_ptr<XrdSiteStatistics> XrdSiteStatisticsInformation::getStatisticsForSite(std::string const &site) {
0072 std::lock_guard<std::mutex> lock(m_mutex);
0073 for (auto &stats : m_sites) {
0074 if (stats->site() == site) {
0075 return get_underlying_safe(stats);
0076 }
0077 }
0078 m_sites.emplace_back(new XrdSiteStatistics(site));
0079 return get_underlying_safe(m_sites.back());
0080 }
0081
0082 void XrdSiteStatisticsInformation::createInstance() {
0083 if (!m_instance) {
0084 std::unique_ptr<XrdSiteStatisticsInformation> tmp{new XrdSiteStatisticsInformation()};
0085 XrdSiteStatisticsInformation *expected = nullptr;
0086 if (m_instance.compare_exchange_strong(expected, tmp.get())) {
0087 tmp.release();
0088 }
0089 }
0090 }
0091
0092 XrdSiteStatisticsInformation *XrdSiteStatisticsInformation::getInstance() {
0093 return m_instance.load(std::memory_order_relaxed);
0094 }
0095
0096 void XrdStatisticsService::fillDescriptions(edm::ConfigurationDescriptions &descriptions) {
0097 edm::ParameterSetDescription desc;
0098 desc.setComment("Report Xrootd-related statistics centrally.");
0099 desc.addUntracked<bool>("reportToFJR", true)
0100 ->setComment(
0101 "True: Add per-site Xrootd statistics to the framework job report.\n"
0102 "False: Collect no site-specific statistics.\n");
0103 descriptions.add("XrdStatisticsService", desc);
0104 }
0105
0106 XrdSiteStatistics::XrdSiteStatistics(std::string const &site)
0107 : m_site(site),
0108 m_readvCount(0),
0109 m_chunkCount(0),
0110 m_readvSize(0),
0111 m_readvNS(0.0),
0112 m_readCount(0),
0113 m_readSize(0),
0114 m_readNS(0) {}
0115
0116 std::shared_ptr<XrdReadStatistics> XrdSiteStatistics::startRead(std::shared_ptr<XrdSiteStatistics> parent,
0117 std::shared_ptr<ClientRequest> req) {
0118 std::shared_ptr<XrdReadStatistics> readStats(new XrdReadStatistics(parent, req->getSize(), req->getCount()));
0119 return readStats;
0120 }
0121
0122 static std::string i2str(int input) {
0123 std::ostringstream formatter;
0124 formatter << input;
0125 return formatter.str();
0126 }
0127
0128 static std::string d2str(double input) {
0129 std::ostringstream formatter;
0130 formatter << std::setw(4) << input;
0131 return formatter.str();
0132 }
0133
0134 void XrdSiteStatistics::recomputeProperties(std::map<std::string, std::string> &props) {
0135 props.clear();
0136
0137 props["readv-numOperations"] = i2str(m_readvCount);
0138 props["readv-numChunks"] = i2str(m_chunkCount);
0139 props["readv-totalMegabytes"] = d2str(static_cast<float>(m_readvSize) / (1024.0 * 1024.0));
0140 props["readv-totalMsecs"] = d2str(m_readvNS / 1e6);
0141
0142 props["read-numOperations"] = i2str(m_readCount);
0143 props["read-totalMegabytes"] = d2str(static_cast<float>(m_readSize) / (1024.0 * 1024.0));
0144 props["read-totalMsecs"] = d2str(static_cast<float>(m_readNS) / 1e6);
0145 }
0146
0147 void XrdSiteStatistics::finishRead(XrdReadStatistics const &readStats) {
0148 if (readStats.readCount() > 1) {
0149 m_readvCount++;
0150 m_chunkCount += readStats.readCount();
0151 m_readvSize += readStats.size();
0152 m_readvNS += readStats.elapsedNS();
0153 } else {
0154 m_readCount++;
0155 m_readSize += readStats.size();
0156 m_readNS += readStats.elapsedNS();
0157 }
0158 }
0159
0160 XrdReadStatistics::XrdReadStatistics(std::shared_ptr<XrdSiteStatistics> parent, IOSize size, size_t count)
0161 : m_size(size), m_count(count), m_parent(parent), m_start(std::chrono::steady_clock::now()) {}
0162
0163 uint64_t XrdReadStatistics::elapsedNS() const {
0164 std::chrono::time_point<std::chrono::steady_clock> end = std::chrono::steady_clock::now();
0165 return std::chrono::duration_cast<std::chrono::nanoseconds>(end - m_start).count();
0166 }