forked from rsocket/rsocket-cpp
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathChannelResponder.cpp
More file actions
102 lines (84 loc) · 2.26 KB
/
ChannelResponder.cpp
File metadata and controls
102 lines (84 loc) · 2.26 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
93
94
95
96
97
98
99
100
101
102
// Copyright 2004-present Facebook. All Rights Reserved.
#include "rsocket/statemachine/ChannelResponder.h"
namespace rsocket {
using namespace yarpl;
using namespace yarpl::flowable;
void ChannelResponder::onSubscribe(
Reference<Subscription> subscription) noexcept {
publisherSubscribe(std::move(subscription));
}
void ChannelResponder::onNext(Payload response) noexcept {
checkPublisherOnNext();
if (!publisherClosed()) {
writePayload(std::move(response), false);
}
}
void ChannelResponder::onComplete() noexcept {
if (!publisherClosed()) {
publisherComplete();
completeStream();
tryCompleteChannel();
}
}
void ChannelResponder::onError(folly::exception_wrapper ex) noexcept {
if (!publisherClosed()) {
publisherComplete();
applicationError(ex.get_exception()->what());
tryCompleteChannel();
}
}
void ChannelResponder::tryCompleteChannel() {
if (publisherClosed() && consumerClosed()) {
closeStream(StreamCompletionSignal::COMPLETE);
}
}
void ChannelResponder::request(int64_t n) noexcept {
ConsumerBase::generateRequest(n);
}
void ChannelResponder::cancel() noexcept {
cancelConsumer();
cancelStream();
tryCompleteChannel();
}
void ChannelResponder::endStream(StreamCompletionSignal signal) {
terminatePublisher();
ConsumerBase::endStream(signal);
}
// TODO: remove this unused function
void ChannelResponder::processInitialFrame(Frame_REQUEST_CHANNEL&& frame) {
onNextPayloadFrame(
frame.requestN_,
std::move(frame.payload_),
frame.header_.flagsComplete(),
true);
}
void ChannelResponder::handlePayload(
Payload&& payload,
bool complete,
bool flagsNext) {
onNextPayloadFrame(0, std::move(payload), complete, flagsNext);
}
void ChannelResponder::onNextPayloadFrame(
uint32_t requestN,
Payload&& payload,
bool complete,
bool next) {
processRequestN(requestN);
processPayload(std::move(payload), next);
if (complete) {
completeConsumer();
tryCompleteChannel();
}
}
void ChannelResponder::handleCancel() {
publisherComplete();
tryCompleteChannel();
}
void ChannelResponder::handleRequestN(uint32_t n) {
processRequestN(n);
}
void ChannelResponder::handleError(folly::exception_wrapper ex) {
errorConsumer(std::move(ex));
tryCompleteChannel();
}
}