bahman/scylla
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

streaming: Introduce send_failed_complete_message

Browse Source 
Currently, send_complete_message is not used. We will use it shortly in
case the local session is failed. Send a complete message with failed
flag to notify peer node that the session is failed so that peer can
close the session. This can speed up the closing of failed session.

Also rename it to send_failed_complete_message.
This commit is contained in:
Asias He
2017-07-18 15:31:31 +08:00
parent f21cb75cdb
commit ca5248cd58
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
streaming/stream_session.cc
View File

@@ -335,7 +335,7 @@ void stream_session::transfer_task_completed(UUID cf_id) {
maybe_completed();
}
void stream_session::send_complete_message() {
void stream_session::send_failed_complete_message() {
if (!_complete_sent) {
_complete_sent = true;
} else {
@@ -345,7 +345,8 @@ void stream_session::send_complete_message() {
auto plan_id = this->plan_id();
sslog.debug("[Stream #{}] SEND COMPLETE_MESSAGE to {}", plan_id, id);
auto session = shared_from_this();
this->ms().send_complete_message(id, plan_id, this->dst_cpu_id).then([session, id, plan_id] {
bool failed = true;
this->ms().send_complete_message(id, plan_id, this->dst_cpu_id, failed).then([session, id, plan_id] {
sslog.debug("[Stream #{}] GOT COMPLETE_MESSAGE Reply from {}", plan_id, id.addr);
}).handle_exception([session, id, plan_id] (auto ep) {
sslog.warn("[Stream #{}] COMPLETE_MESSAGE for {} has failed: {}", plan_id, id.addr, ep);

2
streaming/stream_session.hh
View File

@@ -329,7 +329,7 @@ public:
void receive_task_completed(UUID cf_id);
void transfer_task_completed(UUID cf_id);
private:
void send_complete_message();
void send_failed_complete_message();
bool maybe_completed();
void prepare_receiving(stream_summary& summary);
void start_streaming_files();