ControlService.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
#include "Framework/ControlService.h"
#include "Framework/DriverClient.h"
#include "Framework/DeviceSpec.h"
#include "Framework/DeviceState.h"
#include "Framework/ServiceRegistry.h"
#include "Framework/RawDeviceService.h"
#include "Framework/Logger.h"
#include "Framework/DataProcessingHelpers.h"
#include <string>
#include <string_view>
#include <regex>
#include <iostream>
 
namespace o2::framework
{
 
ControlService::ControlService(ServiceRegistryRef registry, DeviceState& deviceState)
  : mRegistry{registry},
    mDeviceState{deviceState},
    mDriverClient{registry.get<DriverClient>()}
{
}
 
// This will send an end of stream to all the devices downstream.
void ControlService::endOfStream()
{
  std::scoped_lock lock(mMutex);
  mDeviceState.streaming = StreamingState::EndOfStreaming;
}
 
// All we do is to printout
void ControlService::readyToQuit(QuitRequest what)
{
  std::scoped_lock lock(mMutex);
  if (mOnce == true) {
    return;
  }
  mOnce = true;
  switch (what) {
    case QuitRequest::All:
      mDeviceState.quitRequested = true;
      mDriverClient.tell("CONTROL_ACTION: READY_TO_QUIT_ALL");
      break;
    case QuitRequest::Me:
      mDeviceState.quitRequested = true;
      mDriverClient.tell("CONTROL_ACTION: READY_TO_QUIT_ME");
      break;
  }
}
 
void ControlService::notifyStreamingState(StreamingState state)
{
  std::scoped_lock lock(mMutex);
  switch (state) {
    case StreamingState::Idle:
      mDriverClient.tell("CONTROL_ACTION: NOTIFY_STREAMING_STATE IDLE");
      break;
    case StreamingState::Streaming:
      mDriverClient.tell("CONTROL_ACTION: NOTIFY_STREAMING_STATE STREAMING");
      break;
    case StreamingState::EndOfStreaming:
      mDriverClient.tell("CONTROL_ACTION: NOTIFY_STREAMING_STATE EOS");
      break;
    default:
      throw std::runtime_error("Unknown streaming state");
  }
}
 
void ControlService::push(std::string_view key, std::string_view value, int64_t timestamp)
{
  std::scoped_lock lock(mMutex);
  mDriverClient.tell(fmt::format("CONTROL_ACTION: PUT {} {} {}", key, timestamp, value), true);
}
 
void ControlService::notifyDeviceState(std::string currentState)
{
  std::scoped_lock lock(mMutex);
  mDriverClient.tell(fmt::format("CONTROL_ACTION: NOTIFY_DEVICE_STATE {}", currentState));
  mDriverClient.flushPending(mRegistry);
}
 
} // namespace o2::framework