UsageService.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidKernel/UsageService.h"
#include "MantidJson/Json.h"
#include "MantidKernel/ChecksumHelper.h"
#include "MantidKernel/ConfigService.h"
#include "MantidKernel/DateAndTime.h"
#include "MantidKernel/Exception.h"
#include "MantidKernel/InternetHelper.h"
#include "MantidKernel/Logger.h"
#include "MantidKernel/MantidVersion.h"
 
#include <Poco/ActiveResult.h>
#include <Poco/String.h>
#include <algorithm>
#include <boost/algorithm/string/join.hpp>
 
#include <json/json.h>
 
namespace {
constexpr auto SEPARATOR = "->";
}
 
namespace Mantid::Kernel {
 
Kernel::Logger g_log("UsageServiceImpl");
 
//----------------------------------------------------------------------------------------------
FeatureUsage::FeatureUsage(const FeatureType &type, std::string name, const bool internal, std::string application)
    : type(type), name(std::move(name)), internal(internal), application(std::move(application)) {}
 
// Better brute force.
bool FeatureUsage::operator<(const FeatureUsage &r) const {
  if (type < r.type)
    return true;
  if (type > r.type)
    return false;
  // Otherwise type are equal
  if (name < r.name)
    return true;
  if (name > r.name)
    return false;
  // Otherwise name are equal
  if (static_cast<int>(internal) < static_cast<int>(r.internal))
    return true;
  if (static_cast<int>(internal) > static_cast<int>(r.internal))
    return false;
  // Otherwise all are equal
  return false;
}
 
std::string FeatureUsage::featureTypeToString() const {
 
  switch (type) {
  case FeatureType::Algorithm:
    return "Algorithm";
  case FeatureType::Feature:
    return "Feature";
  case FeatureType::Interface:
    return "Interface";
  case FeatureType::Function:
    return "Function";
  }
  return "Unknown";
}
 
::Json::Value FeatureUsage::asJson() const {
  ::Json::Value retVal;
 
  retVal["type"] = featureTypeToString();
  retVal["name"] = name;
  retVal["internal"] = internal;
  retVal["application"] = application;
 
  return retVal;
}
 
//----------------------------------------------------------------------------------------------
UsageServiceImpl::UsageServiceImpl()
    : m_timer(), m_timerTicks(0), m_timerTicksTarget(0), m_FeatureQueue(), m_FeatureQueueSizeThreshold(50),
      m_isEnabled(false), m_mutex(), m_application("python"), m_startTime(Types::Core::DateAndTime::getCurrentTime()),
      m_startupActiveMethod(this, &UsageServiceImpl::sendStartupAsyncImpl),
      m_featureActiveMethod(this, &UsageServiceImpl::sendFeatureAsyncImpl) {
  setInterval(60);
  auto url = Mantid::Kernel::ConfigService::Instance().getValue<std::string>("usagereports.rooturl");
  if (!url.is_initialized()) {
    g_log.debug() << "Failed to load usage report url\n";
  } else {
    m_url = url.get();
    g_log.debug() << "Root usage reporting url is " << m_url << "\n";
  };
}
 
void UsageServiceImpl::setApplicationName(const std::string &name) { m_application = name; }
 
std::string UsageServiceImpl::getApplicationName() const { return m_application; }
 
void UsageServiceImpl::setInterval(const uint32_t seconds) {
  // set the ticks target to by 24 hours / interval
  m_timerTicksTarget = 24 * 60 * 60 / seconds;
 
  m_timer.setStartInterval((seconds * 1000));
  m_timer.setPeriodicInterval((seconds * 1000));
}
 
void UsageServiceImpl::registerStartup() {
  if (isEnabled()) {
    sendStartupReport();
  }
}
 
void UsageServiceImpl::registerFeatureUsage(const FeatureType &type, const std::vector<std::string> &name,
                                            const bool internal) {
  if (isEnabled()) {
    std::lock_guard<std::mutex> _lock(m_mutex);
 
    using boost::algorithm::join;
    m_FeatureQueue.push(FeatureUsage(type, join(name, SEPARATOR), internal, getApplicationName()));
  }
}
 
void UsageServiceImpl::registerFeatureUsage(const FeatureType &type, const std::string &name, const bool internal) {
  if (isEnabled()) {
    std::lock_guard<std::mutex> _lock(m_mutex);
    m_FeatureQueue.push(FeatureUsage(type, name, internal, getApplicationName()));
  }
}
 
void UsageServiceImpl::registerFeatureUsage(const FeatureType &type, std::initializer_list<std::string> name,
                                            const bool internal) {
 
  registerFeatureUsage(type, std::vector<std::string>(name), internal);
}
 
bool UsageServiceImpl::isEnabled() const { return m_isEnabled; }
 
void UsageServiceImpl::setEnabled(const bool enabled) {
  if (m_isEnabled != enabled) {
    if (enabled) {
      m_timer.start(Poco::TimerCallback<UsageServiceImpl>(*this, &UsageServiceImpl::timerCallback));
    } else {
      m_timer.stop();
    }
  }
  m_isEnabled = enabled;
}
 
void UsageServiceImpl::clear() {
  std::queue<FeatureUsage> empty;
  std::swap(m_FeatureQueue, empty);
}
 
void UsageServiceImpl::flush() {
  if (isEnabled()) {
    sendFeatureUsageReport(true);
  }
}
 
Types::Core::time_duration UsageServiceImpl::getUpTime() {
  return Types::Core::DateAndTime::getCurrentTime() - m_startTime;
}
 
void UsageServiceImpl::shutdown() {
  try {
    // stop the timer
    setEnabled(false);
    // send any remaining feature usage records
    sendFeatureUsageReport(true);
  } catch (std::exception &ex) {
    g_log.error() << "Error during the shutdown of the UsageService. " << ex.what();
  }
}
 
void UsageServiceImpl::sendStartupReport() {
  try {
    std::string message = this->generateStartupMessage();
    // send the report
    Poco::ActiveResult<InternetHelper::HTTPStatus> result = m_startupActiveMethod(message);
  } catch (std::exception &ex) {
    g_log.debug() << "Send startup usage failure. " << ex.what() << '\n';
  }
}
 
void UsageServiceImpl::sendFeatureUsageReport(const bool synchronous = false) {
  try {
    std::string message = this->generateFeatureUsageMessage();
    if (!message.empty()) {
      if (synchronous) {
        sendFeatureAsyncImpl(message);
      } else {
        Poco::ActiveResult<InternetHelper::HTTPStatus> result = m_featureActiveMethod(message);
      }
    }
 
  } catch (std::exception &ex) {
    g_log.debug() << "sendFeatureUsageReport failure. " << ex.what() << '\n';
  }
}
 
void UsageServiceImpl::timerCallback(Poco::Timer & /*unused*/) {
  m_timerTicks++;
  if (m_timerTicks > m_timerTicksTarget) {
    // send startup report
    sendStartupReport();
    m_timerTicks = 0;
  }
 
  // Check bufferlength
  if (m_FeatureQueue.size() > m_FeatureQueueSizeThreshold) {
    sendFeatureUsageReport();
  }
}
 
::Json::Value UsageServiceImpl::generateFeatureHeader() {
  ::Json::Value header;
 
  // mantid version and sha1
  header["mantidVersion"] = MantidVersion::versionShort();
 
  return header;
}
 
std::string UsageServiceImpl::generateStartupMessage() {
  ::Json::Value message;
 
  // username
  message["uid"] = Kernel::ChecksumHelper::md5FromString(ConfigService::Instance().getUsername());
  // hostname
  message["host"] = Kernel::ChecksumHelper::md5FromString(ConfigService::Instance().getComputerName());
 
  // os name, version, and architecture
  message["osName"] = ConfigService::Instance().getOSName();
  message["osArch"] = ConfigService::Instance().getOSArchitecture();
  message["osVersion"] = ConfigService::Instance().getOSVersion();
  message["osReadable"] = ConfigService::Instance().getOSVersionReadable();
 
  // legacy interface requires paraview version DON'T REMOVE
  message["ParaView"] = 0;
 
  // mantid version and sha1
  message["mantidVersion"] = MantidVersion::version();
  message["mantidSha1"] = MantidVersion::revisionFull();
 
  // mantid version and sha1
  message["dateTime"] = m_startTime.toISO8601String();
 
  message["application"] = m_application;
 
  return Mantid::JsonHelpers::jsonToString(message);
}
 
std::string UsageServiceImpl::generateFeatureUsageMessage() {
 
  std::map<FeatureUsage, int> featureCountMap;
 
  if (!m_FeatureQueue.empty()) {
    // lock around emptying of the Q so any further threads have to wait
    std::lock_guard<std::mutex> _lock(m_mutex);
    // generate a map containing the counts of identical feature usage records
    while (!m_FeatureQueue.empty()) {
      auto featureUsage = m_FeatureQueue.front();
      m_FeatureQueue.pop();
      if (featureCountMap.find(featureUsage) == featureCountMap.end()) {
        featureCountMap[featureUsage] = 1;
      } else {
        featureCountMap[featureUsage]++;
      }
    }
  }
 
  if (!featureCountMap.empty()) {
    ::Json::Value features;
    auto message = this->generateFeatureHeader();
    for (auto const &featureItem : featureCountMap) {
      ::Json::Value thisFeature = featureItem.first.asJson();
      thisFeature["count"] = featureItem.second;
      features.append(thisFeature);
    }
    if (!features.empty()) {
      message["features"] = features;
      return Mantid::JsonHelpers::jsonToString(message);
    }
  }
  return "";
}
 
//--------------------------------------------------------------------------------------------
InternetHelper::HTTPStatus UsageServiceImpl::sendStartupAsyncImpl(const std::string &message) {
  return this->sendReport(message, m_url + "/api/usage");
}
 
InternetHelper::HTTPStatus UsageServiceImpl::sendFeatureAsyncImpl(const std::string &message) {
  return this->sendReport(message, m_url + "/api/feature");
}
 
InternetHelper::HTTPStatus UsageServiceImpl::sendReport(const std::string &message, const std::string &url) {
  InternetHelper::HTTPStatus status{InternetHelper::HTTPStatus::BAD_REQUEST};
  try {
    Kernel::InternetHelper helper;
    std::stringstream responseStream;
    helper.setTimeout(20);
    helper.setBody(message);
    status = helper.sendRequest(url, responseStream);
  } catch (Mantid::Kernel::Exception::InternetError &e) {
    status = static_cast<InternetHelper::HTTPStatus>(e.errorCode());
    g_log.information() << "Call to \"" << url << "\" responded with " << static_cast<int>(status) << "\n"
                        << e.what() << "\n";
  }
 
  return status;
}
 
} // namespace Mantid::Kernel