library/cpp/messagebus/remote_client_connection.h


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

#pragma once

#include "connection.h"
#include "local_tasks.h"
#include "remote_client_session.h"
#include "remote_connection.h"

#include <util/generic/object_counter.h>

namespace NBus {
    namespace NPrivate {
        class TRemoteClientConnection: public TRemoteConnection, public TBusClientConnection {
            friend class TRemoteConnection;
            friend struct TBusSessionImpl;
            friend class TRemoteClientSession;

        private:
            TObjectCounter<TRemoteClientConnection> ObjectCounter;

            TSyncAckMessages AckMessages;

            TLocalTasks TimeToTimeoutMessages;

            IBusClientHandler* const ClientHandler;

        public:
            TRemoteClientConnection(TRemoteClientSessionPtr session, ui64 id, TNetAddr addr);

            inline TRemoteClientSession* GetSession();

            SOCKET CreateSocket(const TNetAddr& addr);

            void TryConnect() override;

            void HandleEvent(SOCKET socket, void* cookie) override;

            TBusMessage* PopAck(TBusKey id);

            void WriterFillStatus() override;

            void ClearOutgoingQueue(TMessagesPtrs& result, bool reconnect) override;

            void BeforeTryWrite() override;

            void ProcessReplyQueue();

            void MessageSent(TArrayRef<TBusMessagePtrAndHeader> messages) override;

            void TimeoutMessages();

            void ScheduleTimeoutMessages();

            void ReaderProcessMessageUnknownVersion(TArrayRef<const char> dataRef) override;

            EMessageStatus SendMessage(TBusMessage* pMes, bool wait) override;

            EMessageStatus SendMessageOneWay(TBusMessage* pMes, bool wait) override;

            EMessageStatus SendMessageImpl(TBusMessage*, bool wait, bool oneWay);

            void OpenConnection() override;
        };

    }
}