23#include <seastar/core/format.hh>
24#include <seastar/core/function_traits.hh>
25#include <seastar/core/shared_ptr.hh>
26#include <seastar/core/sstring.hh>
27#include <seastar/core/when_all.hh>
28#include <seastar/util/assert.hh>
29#include <seastar/util/is_smart_ptr.hh>
30#include <seastar/core/simple-stream.hh>
31#include <seastar/net/packet-data-source.hh>
33#include <boost/type.hpp>
39enum class exception_type : uint32_t {
65template <
typename Ret,
typename... In>
68 using arg_types = std::tuple<In...>;
75template <
typename Ret,
typename... In>
78 using arg_types = std::tuple<In...>;
84template <
typename Ret,
typename... In>
87 using arg_types = std::tuple<In...>;
94template <
typename Ret,
typename... In>
97 using arg_types = std::tuple<In...>;
103template <
typename Ret,
typename... In>
105 using ret_type = Ret;
106 using arg_types = std::tuple<In...>;
113template <
typename Ret,
typename... In>
115 using ret_type = Ret;
116 using arg_types = std::tuple<In...>;
125 using cleaned_type = T;
128template <
typename... T>
137 using cleaned_type = void;
149template <
typename... In>
153 return std::move(args);
156template <
typename... In>
158std::tuple<std::reference_wrapper<client_info>, In...>
159maybe_add_client_info(do_want_client_info, client_info& ci, std::tuple<In...>&& args) {
160 return std::tuple_cat(std::make_tuple(
std::ref(ci)), std::move(args));
163template <
typename... In>
166maybe_add_time_point(dont_want_time_point, opt_time_point&, std::tuple<In...>&& args) {
167 return std::move(args);
170template <
typename... In>
172std::tuple<opt_time_point, In...>
173maybe_add_time_point(do_want_time_point, opt_time_point& otp, std::tuple<In...>&& args) {
174 return std::tuple_cat(std::make_tuple(otp), std::move(args));
177inline sstring serialize_connection_id(
const connection_id&
id) {
178 sstring p = uninitialized_string(
sizeof(
id));
180 write_le(c,
id.
id());
184inline connection_id deserialize_connection_id(
const sstring& s) {
185 using id_type =
decltype(connection_id{0}.id());
187 auto id = read_le<id_type>(p);
188 return connection_id{
id};
191template <
bool IsSmartPtr>
196 template <
typename Serializer,
typename Output,
typename T>
197 static inline void serialize(Serializer& serializer, Output& out,
const T& t) {
198 return write(serializer, out, t);
204 template <
typename Serializer,
typename Output,
typename T>
205 static inline void serialize(Serializer& serializer, Output& out,
const T& t) {
206 return write(serializer, out, *t);
210template <
typename Serializer,
typename Output,
typename... T>
211inline void do_marshall(Serializer& serializer, Output& out,
const T&... args);
213template <
typename Serializer,
typename Output>
216 static void doit(Serializer& serializer, Output& out,
const T& arg) {
218 serialize_helper_type::serialize(serializer, out, arg);
222 static void doit(Serializer& serializer, Output& out,
const std::reference_wrapper<const T>& arg) {
226 static void put_connection_id(
const connection_id& cid, Output& out) {
227 sstring
id = serialize_connection_id(cid);
228 out.write(
id.c_str(),
id.size());
231 static void doit(Serializer&, Output& out,
const sink<T...>& arg) {
232 put_connection_id(arg.get_id(), out);
236 static void doit(Serializer&, Output& out,
const source<T...>& arg) {
237 put_connection_id(arg.get_id(), out);
241 static void doit(Serializer& serializer, Output& out,
const tuple<T...>& arg) {
242 auto do_do_marshall = [&serializer, &out] (
const auto&... args) {
243 do_marshall(serializer, out, args...);
249 std::apply(do_do_marshall,
static_cast<const std::tuple<T...
>&>(arg));
254template <
typename Serializer,
typename Output,
typename... T>
255inline void do_marshall(Serializer& serializer, Output& out,
const T&... args) {
260static inline memory_output_stream<snd_buf::iterator> make_serializer_stream(snd_buf& output) {
261 auto* b = std::get_if<temporary_buffer<char>>(&output.bufs);
263 return memory_output_stream<snd_buf::iterator>(memory_output_stream<snd_buf::iterator>::simple(b->get_write(), b->size()));
265 auto& ar = std::get<std::vector<temporary_buffer<char>>>(output.bufs);
266 return memory_output_stream<snd_buf::iterator>(memory_output_stream<snd_buf::iterator>::fragmented(ar.begin(), output.size));
270template <
typename Serializer,
typename... T>
271inline snd_buf marshall(Serializer& serializer,
size_t head_space,
const T&... args) {
272 measuring_output_stream measure;
273 do_marshall(serializer, measure, args...);
274 snd_buf ret(measure.size() + head_space);
275 auto out = make_serializer_stream(ret);
276 out.skip(head_space);
277 do_marshall(serializer, out, args...);
281template <
typename Serializer,
typename Input,
typename... T>
282std::tuple<T...> do_unmarshall(connection& c, Input& in);
291template <
typename T,
typename... Args>
292requires requires (Args... args, type<T> t) { read(std::forward<Args>(args)..., t); }
294read_via_type_marker(Args&&... args) {
295 return read(std::forward<Args>(args)..., type<T>());
298template <
typename T,
typename... Args>
299requires requires (Args... args, boost::type<T> t) { read(std::forward<Args>(args)..., t); }
300[[deprecated(
"Use rpc::type<> instead of boost::type<>")]]
302read_via_type_marker(Args&&... args) {
303 return read(std::forward<Args>(args)..., boost::type<T>());
306template<
typename Serializer,
typename Input>
310 return read_via_type_marker<T>(c.serializer<Serializer>(), in);
316 return optional<T>(read_via_type_marker<
typename remove_optional<T>::type>(c.serializer<Serializer>(), in));
330 return deserialize_connection_id(
id);
334 return sink<T...>(make_shared<
sink_impl<Serializer, T...>>(c.get_stream(get_connection_id(in))));
339 return source<T...>(make_shared<
source_impl<Serializer, T...>>(c.get_stream(get_connection_id(in))));
344 return do_unmarshall<Serializer, Input, T...>(c, in);
349template <
typename... T>
354 using type = std::tuple<>;
357template <
typename T0,
typename... T>
359 using type = std::tuple<
362 std::is_default_constructible_v<T>,
369template <
typename... T>
373template <
typename... Tout,
typename... Tin>
375unwrap_optional_if_needed(std::tuple<Tin...>&& tuple_in) {
376 using tuple_in_t = std::tuple<Tin...>;
377 using tuple_out_t = std::tuple<Tout...>;
378 return std::invoke([&] <
size_t... Idx> (std::index_sequence<Idx...>) {
381 if constexpr (std::same_as<std::tuple_element_t<Idx, tuple_in_t>, std::tuple_element_t<Idx, tuple_out_t>>) {
382 return std::move(std::get<Idx>(tuple_in));
384 return std::move(std::get<Idx>(tuple_in).value());
387 }, std::make_index_sequence<
sizeof...(Tout)>());
390template <
typename Serializer,
typename Input,
typename... T>
391inline std::tuple<T...> do_unmarshall(connection& c, Input& in) {
404 using ret_type = std::tuple<T...>;
405 using temporary_type = default_constructible_tuple_except_first_t<T...>;
406 return std::invoke([&] <
size_t... Idx> (std::index_sequence<Idx...>) {
407 auto tmp = temporary_type(
408 std::invoke([&] () -> std::tuple_element_t<Idx, temporary_type> {
409 if constexpr (Idx == 0) {
411 return unmarshal_one<Serializer, Input>::template helper<std::tuple_element_t<Idx, ret_type>>::doit(c, in);
419 (void)(..., ((Idx == 0
421 : ((std::get<Idx>(tmp) = unmarshal_one<Serializer, Input>::template helper<std::tuple_element_t<Idx, ret_type>>::doit(c, in), 0)))));
422 if constexpr (std::same_as<ret_type, temporary_type>) {
426 return unwrap_optional_if_needed<T...>(std::move(tmp));
428 }, std::index_sequence_for<T...>());
431template <
typename Serializer,
typename... T>
432inline std::tuple<T...> unmarshall(connection& c, rcv_buf input) {
433 auto in = make_deserializer_stream(input);
434 return do_unmarshall<Serializer,
decltype(in), T...>(c, in);
437inline std::exception_ptr unmarshal_exception(rcv_buf& d) {
438 std::exception_ptr ex;
439 auto data = make_deserializer_stream(d);
442 data.read(
reinterpret_cast<char*
>(&v32), 4);
443 exception_type ex_type = exception_type(le_to_cpu(v32));
444 data.read(
reinterpret_cast<char*
>(&v32), 4);
445 uint32_t ex_len = le_to_cpu(v32);
448 case exception_type::USER: {
449 std::string s(ex_len,
'\0');
450 data.read(&*s.begin(), ex_len);
451 ex = std::make_exception_ptr(remote_verb_error(std::move(s)));
454 case exception_type::UNKNOWN_VERB: {
456 data.read(
reinterpret_cast<char*
>(&v64), 8);
457 ex = std::make_exception_ptr(unknown_verb_error(le_to_cpu(v64)));
461 ex = std::make_exception_ptr(unknown_exception_error());
467template <
typename Payload,
typename... T>
471 template<
typename... V>
472 void set_value(V&&... v) {
474 p.
set_value(internal::untuple(std::forward<V>(v))...);
483template<
typename Serializer,
typename T>
486 this->set_value(unmarshall<Serializer, T>(dst, std::move(input)));
490template<
typename Serializer,
typename... T>
493 this->set_value(unmarshall<Serializer, T...>(dst, std::move(input)));
497template<
typename Serializer>
504template<
typename Serializer>
507template <
typename Serializer,
typename Ret,
typename... InArgs>
508inline auto wait_for_reply(
wait_type, std::optional<rpc_clock_type::time_point> timeout, rpc_clock_type::time_point start,
cancellable* cancel,
rpc::client& dst, id_type msg_id,
511 auto lambda = [] (reply_type& r,
rpc::client& dst, id_type msg_id,
rcv_buf data)
mutable {
513 dst.get_stats_internal().replied++;
514 return r.get_reply(dst, std::move(data));
516 dst.get_stats_internal().exception_received++;
518 r.p.set_exception(unmarshal_exception(data));
521 using handler_type =
typename rpc::client::template reply_handler<reply_type,
decltype(lambda)>;
522 auto r = std::make_unique<handler_type>(std::move(lambda));
524 auto fut = r->reply.p.get_future();
525 dst.wait_for_reply(msg_id, std::move(r), timeout, cancel);
529template<
typename Serializer,
typename... InArgs>
530inline auto wait_for_reply(no_wait_type, std::optional<rpc_clock_type::time_point>, rpc_clock_type::time_point start, cancellable*, rpc::client&, id_type,
531 signature<no_wait_type (InArgs...)>) {
532 return make_ready_future<>();
535template<
typename Serializer,
typename... InArgs>
536inline auto wait_for_reply(no_wait_type, std::optional<rpc_clock_type::time_point>, rpc_clock_type::time_point, cancellable*, rpc::client&, id_type,
537 signature<future<no_wait_type> (InArgs...)>) {
538 return make_ready_future<>();
544inline rpc_clock_type::time_point
545relative_timeout_to_absolute(rpc_clock_type::duration relative) {
547 return now + std::min(relative, rpc_clock_type::time_point::max() -
now);
551static constexpr size_t request_frame_headroom = 28;
557template<
typename Serializer,
typename MsgType,
typename Ret,
typename... InArgs>
558auto send_helper(MsgType xt, signature<Ret (InArgs...)> xsig) {
561 signature<Ret (InArgs...)> sig;
562 auto send(rpc::client& dst, std::optional<rpc_clock_type::time_point> timeout, cancellable* cancel,
const InArgs&... args) {
564 using cleaned_ret_type =
typename wait_signature<Ret>::cleaned_type;
565 return futurize<cleaned_ret_type>::make_exception_future(closed_error());
570 auto msg_id = dst.next_message_id();
571 snd_buf data = marshall(dst.template serializer<Serializer>(), request_frame_headroom, args...);
574 using wait = wait_signature_t<Ret>;
575 return when_all(dst.request(uint64_t(t), msg_id, std::move(data), timeout, cancel), wait_for_reply<Serializer>(wait(), timeout, start, cancel, dst, msg_id, sig)).then([] (
auto r) {
576 std::get<0>(r).ignore_ready_future();
577 return std::move(std::get<1>(r));
580 auto operator()(rpc::client& dst,
const InArgs&... args) {
581 return send(dst, {},
nullptr, args...);
583 auto operator()(rpc::client& dst, rpc_clock_type::time_point timeout,
const InArgs&... args) {
584 return send(dst, timeout,
nullptr, args...);
586 auto operator()(rpc::client& dst, rpc_clock_type::duration timeout,
const InArgs&... args) {
587 return send(dst, relative_timeout_to_absolute(timeout),
nullptr, args...);
589 auto operator()(rpc::client& dst, cancellable& cancel,
const InArgs&... args) {
590 return send(dst, {}, &cancel, args...);
594 return shelper{xt, xsig};
598static constexpr size_t response_frame_headroom = 16;
600template<
typename Serializer,
typename RetTypes>
601inline future<> reply(wait_type, future<RetTypes>&& ret, int64_t msg_id, shared_ptr<server::connection> client,
602 std::optional<rpc_clock_type::time_point> timeout, std::optional<rpc_clock_type::duration> handler_duration) {
603 if (!client->error()) {
606 if constexpr (std::is_void_v<RetTypes>) {
608 data = std::invoke(marshall<Serializer>,
std::ref(client->template serializer<Serializer>()), response_frame_headroom);
610 data = std::invoke(marshall<Serializer, const RetTypes&>,
std::ref(client->template serializer<Serializer>()), response_frame_headroom, std::move(ret.get()));
612 }
catch (std::exception& ex) {
613 uint32_t len = std::strlen(ex.what());
614 data = snd_buf(response_frame_headroom + 2 *
sizeof(uint32_t) + len);
615 auto os = make_serializer_stream(data);
616 os.skip(response_frame_headroom);
617 uint32_t v32 = cpu_to_le(uint32_t(exception_type::USER));
618 os.write(
reinterpret_cast<char*
>(&v32),
sizeof(v32));
619 v32 = cpu_to_le(len);
620 os.write(
reinterpret_cast<char*
>(&v32),
sizeof(v32));
621 os.write(ex.what(), len);
625 return client->respond(msg_id, std::move(data), timeout, handler_duration);
627 ret.ignore_ready_future();
628 return make_ready_future<>();
633template<
typename Serializer>
634inline future<> reply(no_wait_type, future<no_wait_type>&& r, int64_t msgid, shared_ptr<server::connection> client,
635 std::optional<rpc_clock_type::time_point>, std::optional<rpc_clock_type::duration>) {
638 }
catch (std::exception& ex) {
639 client->get_logger()(client->info(), msgid, to_sstring(
"exception \"") + ex.what() +
"\" in no_wait handler ignored");
641 return make_ready_future<>();
644template<
typename Ret,
typename... InArgs,
typename WantClientInfo,
typename WantTimePoint,
typename Func,
typename ArgsTuple>
645inline futurize_t<Ret> apply(Func& func, client_info& info, opt_time_point time_point, WantClientInfo wci, WantTimePoint wtp, signature<Ret (InArgs...)>, ArgsTuple&& args) {
646 using futurator = futurize<Ret>;
647 return futurator::apply(func, maybe_add_client_info(wci, info, maybe_add_time_point(wtp, time_point, std::forward<ArgsTuple>(args))));
652auto lref_to_cref(T&& x) {
657auto lref_to_cref(T& x) {
663template <
typename Serializer,
typename Func,
typename Ret,
typename... InArgs,
typename WantClientInfo,
typename WantTimePoint>
664auto recv_helper(signature<Ret (InArgs...)> sig, Func&& func, WantClientInfo, WantTimePoint) {
665 using signature =
decltype(sig);
666 using wait_style = wait_signature_t<Ret>;
667 return [func = lref_to_cref(std::forward<Func>(func))](shared_ptr<server::connection> client,
668 std::optional<rpc_clock_type::time_point> timeout,
671 gate::holder guard)
mutable {
672 auto memory_consumed = client->estimate_request_size(data.size);
673 if (memory_consumed > client->max_request_size()) {
674 auto err =
format(
"request size {:d} large than memory limit {:d}", memory_consumed, client->max_request_size());
675 client->get_logger()(client->peer_address(), err);
677 (void)try_with_gate(client->get_server().reply_gate(), [client, timeout, msg_id, err = std::move(err)] {
678 return reply<Serializer>(wait_style(), futurize<Ret>::make_exception_future(std::runtime_error(err.c_str())), msg_id, client, timeout, std::nullopt).handle_exception([client, msg_id] (std::exception_ptr eptr) {
679 client->get_logger()(client->info(), msg_id,
format(
"got exception while processing an oversized message: {}", eptr));
681 }).handle_exception_type([] (gate_closed_exception&) {});
685 auto f = client->wait_for_resources(memory_consumed, timeout).then([client, timeout, msg_id, data = std::move(data), &func, g = std::move(guard)] (
auto permit)
mutable {
687 (void)try_with_gate(client->get_server().reply_gate(), [client, timeout, msg_id, data = std::move(data), permit = std::move(permit), &func] ()
mutable {
689 auto args = unmarshall<Serializer, InArgs...>(*client, std::move(data));
690 auto start = rpc_clock_type::now();
691 return apply(func, client->info(), timeout, WantClientInfo(), WantTimePoint(), signature(), std::move(args)).then_wrapped([client, timeout, msg_id, permit = std::move(permit), start] (futurize_t<Ret> ret) mutable {
692 return reply<Serializer>(wait_style(), std::move(ret), msg_id, client, timeout, rpc_clock_type::now() - start).handle_exception([permit = std::move(permit), client, msg_id] (std::exception_ptr eptr) {
693 client->get_logger()(client->info(), msg_id, format(
"got exception while processing a message: {}", eptr));
697 client->get_logger()(client->info(), msg_id,
format(
"caught exception while processing a message: {}", std::current_exception()));
700 }).handle_exception_type([g = std::move(g)] (gate_closed_exception&) {});
704 f = f.handle_exception_type([] (semaphore_timed_out&) { });
712template<
typename Func>
713auto make_copyable_function(Func&& func, std::enable_if_t<!std::is_copy_constructible_v<std::decay_t<Func>>,
void*> =
nullptr) {
714 auto p = make_lw_shared<typename std::decay_t<Func>>(std::forward<Func>(func));
715 return [p] (
auto&&... args) {
return (*p)( std::forward<decltype(args)>(args)... ); };
718template<
typename Func>
719auto make_copyable_function(Func&& func, std::enable_if_t<std::is_copy_constructible_v<std::decay_t<Func>>,
void*> =
nullptr) {
720 return std::forward<Func>(func);
730template<
typename Ret,
typename... In>
732 template<
typename T,
bool IsSmartPtr>
733 struct drop_smart_ptr_impl;
735 struct drop_smart_ptr_impl<T, true> {
736 using type =
typename T::element_type;
739 struct drop_smart_ptr_impl<T, false> {
743 using drop_smart_ptr = drop_smart_ptr_impl<T, is_smart_ptr<T>::value>;
746 using return_type =
typename drop_smart_ptr<Ret>::type;
748 using type = return_type(
typename remove_optional<In>::type...);
751template<
typename Serializer,
typename MsgType>
752template<
typename Ret,
typename... In>
755 return send_helper<Serializer>(t, sig_type());
758template<
typename Serializer,
typename MsgType>
759template<
typename Func>
764template<
typename Serializer,
typename MsgType>
765template<
typename Func>
768 using clean_sig_type =
typename sig_type::clean;
769 using want_client_info =
typename sig_type::want_client_info;
770 using want_time_point =
typename sig_type::want_time_point;
771 auto recv = recv_helper<Serializer>(clean_sig_type(), std::forward<Func>(func),
772 want_client_info(), want_time_point());
773 register_receiver(t,
rpc_handler{sg, make_copyable_function(std::move(recv)), {}});
774 return make_client(clean_sig_type(), t);
777template<
typename Serializer,
typename MsgType>
778template<
typename Func>
783template<
typename Serializer,
typename MsgType>
785 auto it = _handlers.find(t);
786 if (it != _handlers.end()) {
787 return it->second.use_gate.close().finally([
this, t] {
791 return make_ready_future<>();
794template<
typename Serializer,
typename MsgType>
796 auto it = _handlers.find(msg_id);
797 if (it == _handlers.end()) {
800 return !it->second.use_gate.is_closed();
803template<
typename Serializer,
typename MsgType>
804std::optional<protocol_base::handler_with_holder> protocol<Serializer, MsgType>::get_handler(uint64_t msg_id) {
805 const auto it = _handlers.find(MsgType(msg_id));
806 if (it != _handlers.end()) {
808 return handler_with_holder{it->second, it->second.use_gate.hold()};
809 }
catch (gate_closed_exception&) {
816template<
typename T> T make_shard_local_buffer_copy(foreign_ptr<std::unique_ptr<T>> org);
818template<
typename Serializer,
typename... Out>
819future<> sink_impl<Serializer, Out...>::operator()(
const Out&... args) {
822 snd_buf data = marshall(this->_con->get()->template serializer<Serializer>(), 4, args...);
823 static_assert(snd_buf::chunk_size >= 4,
"send buffer chunk size is too small");
824 auto p = data.front().get_write();
825 write_le<uint32_t>(p, data.size - 4);
828 auto size = std::min(
size_t(data.size), max_stream_buffers_memory);
829 const auto seq_num = _next_seq_num++;
830 return get_units(this->_sem, size).then([
this, data = make_foreign(std::make_unique<snd_buf>(std::move(data))), seq_num] (semaphore_units<> su)
mutable {
836 (void)
smp::submit_to(this->_con->get_owner_shard(), [
this, data = std::move(data), seq_num] ()
mutable {
837 connection* con = this->_con->get();
841 if(con->sink_closed()) {
845 auto& last_seq_num = _remote_state.last_seq_num;
846 auto& out_of_order_bufs = _remote_state.out_of_order_bufs;
848 auto local_data = make_shard_local_buffer_copy(std::move(data));
849 const auto seq_num_diff = seq_num - last_seq_num;
850 if (seq_num_diff > 1) {
851 auto [it, _] = out_of_order_bufs.emplace(seq_num, deferred_snd_buf{
promise<>{}, std::move(local_data)});
852 return it->second.pr.get_future();
855 last_seq_num = seq_num;
856 auto ret_fut = con->send(std::move(local_data), {},
nullptr);
857 while (!out_of_order_bufs.empty() && out_of_order_bufs.begin()->first == (last_seq_num + 1)) {
858 auto it = out_of_order_bufs.begin();
859 last_seq_num = it->first;
860 auto fut = con->send(std::move(it->second.data), {},
nullptr);
861 fut.forward_to(std::move(it->second.pr));
862 out_of_order_bufs.erase(it);
865 }).then_wrapped([su = std::move(su),
this] (future<> f) {
866 if (f.failed() && !this->_ex) {
867 this->_ex = f.get_exception();
869 f.ignore_ready_future();
872 return make_ready_future<>();
876template<
typename Serializer,
typename... Out>
877future<> sink_impl<Serializer, Out...>::flush() {
879 return with_semaphore(this->_sem, max_stream_buffers_memory, [
this] {
887template<
typename Serializer,
typename... Out>
888future<> sink_impl<Serializer, Out...>::close() {
889 return with_semaphore(this->_sem, max_stream_buffers_memory, [
this] {
891 connection* con = this->_con->get();
892 if (con->sink_closed()) {
893 return make_exception_future(stream_closed());
895 future<> f = make_ready_future<>();
896 if (!con->error() && !this->_ex) {
897 snd_buf data = marshall(con->template serializer<Serializer>(), 4);
898 static_assert(snd_buf::chunk_size >= 4,
"send buffer chunk size is too small");
899 auto p = data.front().get_write();
900 write_le<uint32_t>(p, -1U);
901 f = con->send(std::move(data), {},
nullptr);
905 return f.finally([con] {
return con->close_sink(); });
910template<
typename Serializer,
typename... Out>
911sink_impl<Serializer, Out...>::~sink_impl() {
914 SEASTAR_ASSERT(this->_con->get()->sink_closed());
917template<
typename Serializer,
typename... In>
918future<std::optional<std::tuple<In...>>> source_impl<Serializer, In...>::operator()() {
919 auto process_one_buffer = [
this] {
920 foreign_ptr<std::unique_ptr<rcv_buf>> buf = std::move(this->_bufs.front());
921 this->_bufs.pop_front();
922 return std::apply([] (In&&... args) {
923 auto ret = std::make_optional(std::make_tuple(std::move(args)...));
925 }, unmarshall<Serializer, In...>(*this->_con->get(), make_shard_local_buffer_copy(std::move(buf))));
928 if (!this->_bufs.empty()) {
929 return process_one_buffer();
933 return smp::submit_to(this->_con->get_owner_shard(), [
this] () -> future<> {
934 connection* con = this->_con->get();
935 if (con->_source_closed) {
936 return make_exception_future<>(stream_closed());
938 return con->stream_receive(this->_bufs).then_wrapped([
this, con] (future<>&& f) {
940 return con->close_source().then_wrapped([ex = f.get_exception()] (future<> f){
941 f.ignore_ready_future();
942 return make_exception_future<>(ex);
945 if (this->_bufs.empty()) {
946 return con->close_source().then_wrapped([] (future<> f) {
947 f.ignore_ready_future();
948 return make_ready_future<>();
951 return make_ready_future<>();
953 }).then([
this, process_one_buffer] () {
954 if (this->_bufs.empty()) {
957 return process_one_buffer();
962template<
typename... Out>
963connection_id sink<Out...>::get_id()
const {
964 return _impl->_con->get()->get_connection_id();
967template<
typename... In>
968connection_id source<In...>::get_id()
const {
969 return _impl->_con->get()->get_connection_id();
972template<
typename... In>
973template<
typename Serializer,
typename... Out>
974sink<Out...> source<In...>::make_sink() {
975 return sink<Out...>(make_shared<sink_impl<Serializer, Out...>>(_impl->_con));
984struct hash<
seastar::rpc::streaming_domain_type> {
987 boost::hash_combine(h, std::hash<uint64_t>{}(domain._id));
A representation of a possibly not-yet-computed value.
Definition: future.hh:1197
static time_point now() noexcept
Definition: lowres_clock.hh:74
promise - allows a future value to be made available at a later time.
Definition: future.hh:912
void set_value(A &&... a) noexcept
Sets the promises value.
Definition: future.hh:968
Definition: reference_wrapper.hh:43
Definition: rpc_impl.hh:731
Definition: rpc_types.hh:140
Definition: rpc_types.hh:67
Definition: rpc_types.hh:194
Definition: rpc_types.hh:189
Definition: rpc_types.hh:318
Definition: rpc_types.hh:357
Definition: rpc_types.hh:398
Identifies function calls that are accounted as a group.
Definition: scheduling.hh:293
static futurize_t< std::invoke_result_t< Func > > submit_to(unsigned t, smp_submit_to_options options, Func &&func) noexcept
Definition: smp.hh:354
future< T > make_ready_future(A &&... value) noexcept
Creates a future in an available, value state.
Definition: future.hh:1869
future< T > make_exception_future(std::exception_ptr &&value) noexcept
Creates a future in an available, failed state.
Definition: future.hh:1875
auto when_all(FutOrFuncs &&... fut_or_funcs) noexcept
Definition: when_all.hh:252
future now()
Returns a ready future.
Definition: later.hh:35
reference_wrapper< T > ref(T &object) noexcept
Wraps reference in a reference_wrapper.
Definition: reference_wrapper.hh:62
std::future< T > submit_to(instance &instance, unsigned shard, Func func)
Definition: alien.hh:204
Seastar API namespace.
Definition: abort_on_ebadf.hh:26
sstring format(fmt::format_string< A... > fmt, A &&... a)
Definition: format.hh:42
Definition: function_traits.hh:74
Definition: rpc_types.hh:204
Definition: rpc_types.hh:98
Definition: rpc_impl.hh:350
Definition: rpc_impl.hh:57
Definition: rpc_impl.hh:61
Definition: rpc_impl.hh:58
Definition: rpc_impl.hh:62
Definition: rpc_impl.hh:215
Definition: rpc_impl.hh:214
Definition: rpc_types.hh:180
Definition: rpc_types.hh:240
Definition: rpc_impl.hh:468
Definition: rpc_impl.hh:484
Definition: rpc_impl.hh:45
Definition: rpc_impl.hh:192
Definition: rpc_impl.hh:66
Definition: rpc_impl.hh:308
Definition: rpc_impl.hh:307
Definition: rpc_impl.hh:123
Definition: rpc_impl.hh:54