97 lines
3.8 KiB
C++
Executable File
97 lines
3.8 KiB
C++
Executable File
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
// A thrift server to receive EchoRequest and send back EchoResponse.
|
|
|
|
#include <gflags/gflags.h>
|
|
|
|
#include <butil/logging.h>
|
|
|
|
#include "gen-cpp/EchoService.h"
|
|
|
|
#include <thrift/protocol/TBinaryProtocol.h>
|
|
#include <thrift/server/TSimpleServer.h>
|
|
#include <thrift/transport/TServerSocket.h>
|
|
#include <thrift/transport/TTransportUtils.h>
|
|
#include <thrift/server/TNonblockingServer.h>
|
|
#include <thrift/concurrency/PosixThreadFactory.h>
|
|
|
|
// _THRIFT_STDCXX_H_ is defined by thrift/stdcxx.h which was added since thrift 0.11.0
|
|
#include <thrift/TProcessor.h> // to include stdcxx.h if present
|
|
#ifndef THRIFT_STDCXX
|
|
#if defined(_THRIFT_STDCXX_H_)
|
|
# define THRIFT_STDCXX apache::thrift::stdcxx
|
|
#include <thrift/transport/TNonblockingServerSocket.h>
|
|
#else
|
|
# define THRIFT_STDCXX boost
|
|
# include <boost/make_shared.hpp>
|
|
#endif
|
|
#endif
|
|
|
|
DEFINE_int32(port, 8019, "Port of server");
|
|
|
|
class EchoServiceHandler : virtual public example::EchoServiceIf {
|
|
public:
|
|
EchoServiceHandler() {}
|
|
|
|
void Echo(example::EchoResponse& res, const example::EchoRequest& req) {
|
|
// Process request, just attach a simple string.
|
|
res.data = req.data + " world";
|
|
return;
|
|
}
|
|
|
|
};
|
|
|
|
int main(int argc, char *argv[]) {
|
|
// Parse gflags. We recommend you to use gflags as well.
|
|
google::ParseCommandLineFlags(&argc, &argv, true);
|
|
|
|
THRIFT_STDCXX::shared_ptr<EchoServiceHandler> handler(new EchoServiceHandler());
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::concurrency::PosixThreadFactory> thread_factory(
|
|
new apache::thrift::concurrency::PosixThreadFactory(
|
|
apache::thrift::concurrency::PosixThreadFactory::ROUND_ROBIN,
|
|
apache::thrift::concurrency::PosixThreadFactory::NORMAL, 1, false));
|
|
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::server::TProcessor> processor(
|
|
new example::EchoServiceProcessor(handler));
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::protocol::TProtocolFactory> protocol_factory(
|
|
new apache::thrift::protocol::TBinaryProtocolFactory());
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::transport::TTransportFactory> transport_factory(
|
|
new apache::thrift::transport::TBufferedTransportFactory());
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::concurrency::ThreadManager> thread_mgr(
|
|
apache::thrift::concurrency::ThreadManager::newSimpleThreadManager(2));
|
|
|
|
thread_mgr->threadFactory(thread_factory);
|
|
|
|
thread_mgr->start();
|
|
|
|
#if defined(_THRIFT_STDCXX_H_)
|
|
THRIFT_STDCXX::shared_ptr<apache::thrift::transport::TNonblockingServerSocket> server_transport =
|
|
THRIFT_STDCXX::make_shared<apache::thrift::transport::TNonblockingServerSocket>(FLAGS_port);
|
|
|
|
apache::thrift::server::TNonblockingServer server(processor,
|
|
transport_factory, transport_factory, protocol_factory,
|
|
protocol_factory, server_transport);
|
|
#else
|
|
apache::thrift::server::TNonblockingServer server(processor,
|
|
transport_factory, transport_factory, protocol_factory,
|
|
protocol_factory, FLAGS_port);
|
|
#endif
|
|
server.serve();
|
|
return 0;
|
|
}
|