-
Notifications
You must be signed in to change notification settings - Fork 496
Expand file tree
/
Copy pathControlService.cxx
More file actions
92 lines (84 loc) · 2.81 KB
/
ControlService.cxx
File metadata and controls
92 lines (84 loc) · 2.81 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
// 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