veslibrary/ves_cpplibrary/src/lib/transport/XRpcClientTransport.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

#pragma once

#include "XTransport.h"
#include "thrift/transport/TSocket.h"
#include "thrift/protocol/TCompactProtocol.h"
#include "thrift/transport/TBufferTransports.h"
#include "gen-cpp/rpc_constants.h"
#include "gen-cpp/rpc_types.h"
#include "gen-cpp/XRpcTransport.h"

namespace vagt
{
    namespace transport
    {
        class XRpcClientTransport : public XTransport
        {
        public:
            XRpcClientTransport(const XTransportOption& option);
            virtual ~XRpcClientTransport();

            virtual XErrorCode start() override;
            virtual XErrorCode stop() override;
            virtual XErrorCode post(const std::string& event) override;

            virtual void cancelPost() override;
            virtual void allowPost() override;
            virtual bool shouldCancelPost() override;
        private:
            XErrorCode startRPC();
            XErrorCode stopRPC();

            XTransportOption option_;

            std::shared_ptr<apache::thrift::transport::TTransport> rpcTransport_;
            std::shared_ptr<vagt::transport::rpc::XRpcTransportClient> rpcClient_;
        };
    }
}