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;
        };
 
    }
}