ET9_WebGL_Dev/Cpp/Platform/Rpc/RpcServer.cc

#include <functional>
#include <memory>
#include <boost/asio.hpp>
#include <google/protobuf/service.h>
#include <google/protobuf/descriptor.h>
#include "Base/Marcos.h"
#include "Rpc/Typedef.h"
#include "Rpc/RpcServer.h"
#include "Rpc/RpcSession.h"
#include "Rpc/ResponseHandler.h"
#include "Rpc/MethodInfo.h"

namespace Egametang {

RpcServer::RpcServer(boost::asio::io_service& service, int port):
		ioService(service), acceptor(ioService),
		threadPool(1), sessions(),
		methods()
{
	boost::asio::ip::address address;
	address.from_string("127.0.0.1");
	boost::asio::ip::tcp::endpoint endpoint(address, port);
	acceptor.open(endpoint.protocol());
	acceptor.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
	acceptor.bind(endpoint);
	acceptor.listen();
	auto newSession = std::make_shared<RpcSession>(ioService, *this);
	acceptor.async_accept(newSession->Socket(),
			std::bind(&RpcServer::OnAsyncAccept, this,
					newSession, std::placeholders::_1));
}

RpcServer::~RpcServer()
{
	threadPool.wait();
	acceptor.close();
}

void RpcServer::OnAsyncAccept(RpcSessionPtr session, const boost::system::error_code& err)
{
	if (err)
	{
		return;
	}
	session->Start();
	sessions.insert(session);
	auto newSession = std::make_shared<RpcSession>(ioService, *this);
	acceptor.async_accept(newSession->Socket(),
			std::bind(&RpcServer::OnAsyncAccept, this,
					newSession, std::placeholders::_1));
}

void RpcServer::OnCallMethod(RpcSessionPtr session, ResponseHandlerPtr responseHandler)
{
	// 调度到网络线
	session->Socket().get_io_service().post(
			std::bind(&ResponseHandler::Run, responseHandler));
}

void RpcServer::RunService(
		RpcSessionPtr session, const RpcMetaPtr meta,
		const StringPtr message, MessageHandler messageHandler)
{
	MethodInfoPtr methodInfo = methods[meta->method];

	auto responseHandler =
			std::make_shared<ResponseHandler>(meta, message, methodInfo, messageHandler);

	google::protobuf::Closure* done = google::protobuf::NewCallback(
			this, &RpcServer::OnCallMethod, session, responseHandler);

	threadPool.schedule(
			std::bind(&google::protobuf::Service::CallMethod, methodInfo->GetService(),
					&responseHandler->Method(), (google::protobuf::RpcController*)(nullptr),
					responseHandler->Request(), responseHandler->Response(),
					done));
}

void RpcServer::Register(ProtobufServicePtr service)
{
	std::hash<std::string> stringHash;
	const google::protobuf::ServiceDescriptor* serviceDescriptor = service->GetDescriptor();
	for (int i = 0; i < serviceDescriptor->method_count(); ++i)
	{
		const google::protobuf::MethodDescriptor* methodDescriptor =
				serviceDescriptor->method(i);
		std::size_t methodHash = stringHash(methodDescriptor->full_name());
		auto methodInfo = std::make_shared<MethodInfo>(service, methodDescriptor);
		methods[methodHash] = methodInfo;
	}
}

void RpcServer::Remove(RpcSessionPtr session)
{
	sessions.erase(session);
}

} // namespace Egametang