Thrift 双向通信实现（c++版）

1. thrift文件
thrift IDL文件源码：

service HelloWorldBidirectionService{
    oneway void SayHello(1:string msg);
}

2. 开发环境设置


3. thrift服务端编写
thrift服务端修改，将HelloWorldBidirectionService_server.skeleton.cpp文件修改为：

// This autogenerated skeleton file illustrates how to build a server.
// You should copy it to another filename to avoid overwriting it.

#include "HelloWorldBidirectionService.h"
#include 
#include 
#include 
#include 
#include 

#include 
#include   
#include 

using namespace ::apache::thrift;
using namespace ::apache::thrift::protocol;
using namespace ::apache::thrift::transport;
using namespace ::apache::thrift::server;

using namespace ::apache::thrift::concurrency;

using boost::shared_ptr;

static boost::shared_ptr g_CurrentTransport = nullptr;

class HelloWorldBidirectionServiceHandler : virtual public HelloWorldBidirectionServiceIf {
public:
    HelloWorldBidirectionServiceHandler() {
        // Your initialization goes here
    }

    void SayHello(const std::string& msg) {
        // Your implementation goes here
        printf("%s\n", msg);

        SayToClient(msg);
    }
    void SayToClient(const std::string& msg)
    {
        shared_ptr protocol(new TBinaryProtocol(g_CurrentTransport));
        HelloWorldBidirectionServiceClient client(protocol);
        //Thread.Sleep(1000);
        client.SayHello(msg);
    }
};

class HelloWorldBidirectionProcessor : public TProcessorFactory
{
public:
    boost::shared_ptr getProcessor(const TConnectionInfo& connInfo)
    {
        g_CurrentTransport = connInfo.transport;

        printf("接收到连接信号\n");

        shared_ptr handler(new HelloWorldBidirectionServiceHandler());
        shared_ptr processor(new HelloWorldBidirectionServiceProcessor(handler));
        return processor;
    }
};
static boost::shared_ptr getProcessorFactory()
{
    shared_ptr processorFactory(new HelloWorldBidirectionProcessor());
    return processorFactory;
}



int main_tt(int argc, char **argv) {
    int port = 9090;
    shared_ptr serverTransport(new TServerSocket(port));
    shared_ptr transportFactory(new TTransportFactory());
    shared_ptr protocolFactory(new TBinaryProtocolFactory());

    shared_ptr threadManager = ThreadManager::newSimpleThreadManager(10);//指定10个线程数  

    shared_ptr threadFactory = shared_ptr(new StdThreadFactory());
    threadManager->threadFactory(threadFactory);
    threadManager->start();

    shared_ptr server(new TThreadPoolServer(getProcessorFactory(), serverTransport, transportFactory, protocolFactory, threadManager));

    /* shared_ptr handler(new HelloWorldBidirectionServiceHandler());
     shared_ptr processor(new HelloWorldBidirectionServiceProcessor(handler));
     shared_ptr serverTransport(new TServerSocket(port));
     shared_ptr transportFactory(new TBufferedTransportFactory());
     shared_ptr protocolFactory(new TBinaryProtocolFactory());

     TSimpleServer server(processor, serverTransport, transportFactory, protocolFactory);*/
    server->serve();
    return 0;
}

4. thrift 客户端编写
thrift 客户端修改，新建文件 HelloWorldBidirectionService_client.cpp，编码：

#include 

#include 
#include 
#include 

#include "HelloWorldBidirectionService.h"

#include 


using namespace std;
using namespace apache::thrift;
using namespace apache::thrift::protocol;
using namespace apache::thrift::transport;

//using namespace tutorial;
using boost::shared_ptr;
//using namespace boost;
using boost::thread;

#define BOOST_DATE_TIME_SOURCE
#define BOOST_THREAD_NO_LIB


class HelloWorldBidirectionFace :public HelloWorldBidirectionServiceIf
{
public:
    void SayHello(const std::string& msg) {
        // Your implementation goes here
        /*printf("%s\n", msg);
        printf("%s\n", &msg);*/

        cout << msg << endl;
        //cout << &msg << endl;
    }
};

static void Run(boost::shared_ptr sock)
{
    boost::shared_ptr handler(new HelloWorldBidirectionFace());
    boost::shared_ptr processor(new HelloWorldBidirectionServiceProcessor(handler));

    //boost::shared_ptr processor(new HelloWorldBidirectionFace());
    try
    {
        boost::shared_ptr inProtocol(new TBinaryProtocol(sock));
        boost::shared_ptr outProtocol(new TBinaryProtocol(sock));
        while (processor->process(inProtocol, outProtocol, "proc"))
        {
            printf("wait next msg\n");
        }
    }
    catch (TException& tx)
    {
        printf("connect close\n");
        cout << "ERROR: " << tx.what() << endl;
    }
}

static void RecFromConsole(boost::shared_ptr client)
{
    string str;
    cout << "input exit stop" << endl;
    cin >> str;
    while (str.compare("exit"))
    {
        client->SayHello(str);

        cin >> str;
    }
}


int main()
{
    boost::shared_ptr socket(new TSocket("localhost", 9090));
    boost::shared_ptr transport(new TBufferedTransport(socket));
    boost::shared_ptr protocol(new TBinaryProtocol(transport));
    boost::shared_ptr client(new HelloWorldBidirectionServiceClient(protocol));

    try {
        transport->open();

        boost::thread mythread(Run, socket);

        boost::thread mythread2(RecFromConsole, client);
        mythread2.join();

        transport->close();
    }
    catch (TException& tx) {
        cout << "ERROR: " << tx.what() << endl;
    }
}

6. 源码地址:
Thrift_HelloWorld_Cpp_Bidirection