summaryrefslogtreecommitdiffstats
path: root/veslibrary/ves_cpplibrary/src/lib/transport/XRpcServertTransport.h
blob: 14eab07b5959350641817e776528718a81189910 (plain)
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
#pragma once

#include "XTransport.h"
#include "thrift/server/TServer.h"
#include "gen-cpp/rpc_constants.h"
#include "gen-cpp/rpc_types.h"
#include "gen-cpp/XRpcTransport.h"
#include <future>

namespace vagt
{
    namespace transport
    {
        class XRpcServerTransport : public XTransport
        {
        public:
            XRpcServerTransport(std::shared_ptr<XTransport> transport, const XTransportOption& option);
            virtual ~XRpcServerTransport();

            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();
            void worker();
            std::future<void> future_;

            std::shared_ptr<XTransport> transport_;
            XTransportOption option_;

            std::shared_ptr<apache::thrift::server::TServer> rpcServer_;
        };
    }
}