diff options
author | Devtools Arcadia <arcadia-devtools@yandex-team.ru> | 2022-02-07 18:08:42 +0300 |
---|---|---|
committer | Devtools Arcadia <arcadia-devtools@mous.vla.yp-c.yandex.net> | 2022-02-07 18:08:42 +0300 |
commit | 1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch) | |
tree | e26c9fed0de5d9873cce7e00bc214573dc2195b7 /library/cpp/messagebus/connection.h | |
download | ydb-1110808a9d39d4b808aef724c861a2e1a38d2a69.tar.gz |
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'library/cpp/messagebus/connection.h')
-rw-r--r-- | library/cpp/messagebus/connection.h | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/library/cpp/messagebus/connection.h b/library/cpp/messagebus/connection.h new file mode 100644 index 0000000000..b1df64ddc1 --- /dev/null +++ b/library/cpp/messagebus/connection.h @@ -0,0 +1,61 @@ +#pragma once + +#include "defs.h" +#include "message.h" + +#include <util/generic/ptr.h> + +namespace NBus { + struct TBusClientConnection { + /// if you want to open connection early + virtual void OpenConnection() = 0; + + /// Send message to the destination + /// If addr is set then use it as destination. + /// Takes ownership of addr (see ClearState method). + virtual EMessageStatus SendMessage(TBusMessage* pMes, bool wait = false) = 0; + + virtual EMessageStatus SendMessageOneWay(TBusMessage* pMes, bool wait = false) = 0; + + /// Like SendMessage but cares about message + template <typename T /* <: TBusMessage */> + EMessageStatus SendMessageAutoPtr(const TAutoPtr<T>& mes, bool wait = false) { + EMessageStatus status = SendMessage(mes.Get(), wait); + if (status == MESSAGE_OK) + Y_UNUSED(mes.Release()); + return status; + } + + /// Like SendMessageOneWay but cares about message + template <typename T /* <: TBusMessage */> + EMessageStatus SendMessageOneWayAutoPtr(const TAutoPtr<T>& mes, bool wait = false) { + EMessageStatus status = SendMessageOneWay(mes.Get(), wait); + if (status == MESSAGE_OK) + Y_UNUSED(mes.Release()); + return status; + } + + EMessageStatus SendMessageMove(TBusMessageAutoPtr message, bool wait = false) { + return SendMessageAutoPtr(message, wait); + } + + EMessageStatus SendMessageOneWayMove(TBusMessageAutoPtr message, bool wait = false) { + return SendMessageOneWayAutoPtr(message, wait); + } + + // TODO: implement similar one-way methods + + virtual ~TBusClientConnection() { + } + }; + + namespace NPrivate { + struct TBusClientConnectionPtrOps { + static void Ref(TBusClientConnection*); + static void UnRef(TBusClientConnection*); + }; + } + + using TBusClientConnectionPtr = TIntrusivePtr<TBusClientConnection, NPrivate::TBusClientConnectionPtrOps>; + +} |