111 lines
4.1 KiB
C
111 lines
4.1 KiB
C
|
/*
|
|||
|
|
* Copyright (c) 2016 The ZLToolKit project authors. All Rights Reserved.
|
|||
|
|
*
|
|||
|
|
* This file is part of ZLToolKit(https://github.com/ZLMediaKit/ZLToolKit).
|
|||
|
|
*
|
|||
|
|
* Use of this source code is governed by MIT license that can be found in the
|
|||
|
|
* LICENSE file in the root of the source tree. All contributing project authors
|
|||
|
|
* may be found in the AUTHORS file in the root of the source tree.
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
#ifndef TCPSERVER_TCPSERVER_H
|
|||
|
|
#define TCPSERVER_TCPSERVER_H
|
|||
|
|
|
|||
|
|
#include <memory>
|
|||
|
|
#include <functional>
|
|||
|
|
#include <unordered_map>
|
|||
|
|
#include "Server.h"
|
|||
|
|
#include "Session.h"
|
|||
|
|
#include "Poller/Timer.h"
|
|||
|
|
#include "Util/util.h"
|
|||
|
|
|
|||
|
|
namespace toolkit {
|
|||
|
|
|
|||
|
|
//TCP服务器，可配置的；配置通过Session::attachServer方法传递给会话对象
|
|||
|
|
class TcpServer : public Server {
|
|||
|
|
public:
|
|||
|
|
using Ptr = std::shared_ptr<TcpServer>;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 创建tcp服务器，listen fd的accept事件会加入到所有的poller线程中监听
|
|||
|
|
* 在调用TcpServer::start函数时，内部会创建多个子TcpServer对象，
|
|||
|
|
* 这些子TcpServer对象通过Socket对象克隆的方式在多个poller线程中监听同一个listen fd
|
|||
|
|
* 这样这个TCP服务器将会通过抢占式accept的方式把客户端均匀的分布到不同的poller线程
|
|||
|
|
* 通过该方式能实现客户端负载均衡以及提高连接接收速度
|
|||
|
|
*/
|
|||
|
|
explicit TcpServer(const EventPoller::Ptr &poller = nullptr);
|
|||
|
|
~TcpServer() override;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 开始tcp server
|
|||
|
|
* @param port 本机端口，0则随机
|
|||
|
|
* @param host 监听网卡ip
|
|||
|
|
* @param backlog tcp listen backlog
|
|||
|
|
*/
|
|||
|
|
template <typename SessionType>
|
|||
|
|
void start(uint16_t port, const std::string &host = "::", uint32_t backlog = 1024, const std::function<void(std::shared_ptr<SessionType> &)> &cb = nullptr) {
|
|||
|
|
static std::string cls_name = toolkit::demangle(typeid(SessionType).name());
|
|||
|
|
// Session创建器，通过它创建不同类型的服务器
|
|||
|
|
_session_alloc = [cb](const TcpServer::Ptr &server, const Socket::Ptr &sock) {
|
|||
|
|
auto session = std::shared_ptr<SessionType>(new SessionType(sock), [](SessionType *ptr) {
|
|||
|
|
TraceP(static_cast<Session *>(ptr)) << "~" << cls_name;
|
|||
|
|
delete ptr;
|
|||
|
|
});
|
|||
|
|
if (cb) {
|
|||
|
|
cb(session);
|
|||
|
|
}
|
|||
|
|
TraceP(static_cast<Session *>(session.get())) << cls_name;
|
|||
|
|
session->setOnCreateSocket(server->_on_create_socket);
|
|||
|
|
return std::make_shared<SessionHelper>(server, std::move(session), cls_name);
|
|||
|
|
};
|
|||
|
|
start_l(port, host, backlog);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 获取服务器监听端口号, 服务器可以选择监听随机端口
|
|||
|
|
*/
|
|||
|
|
uint16_t getPort();
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 自定义socket构建行为
|
|||
|
|
*/
|
|||
|
|
void setOnCreateSocket(Socket::onCreateSocket cb);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 根据socket对象创建Session对象
|
|||
|
|
* 需要确保在socket归属poller线程执行本函数
|
|||
|
|
*/
|
|||
|
|
Session::Ptr createSession(const Socket::Ptr &socket);
|
|||
|
|
|
|||
|
|
protected:
|
|||
|
|
virtual void cloneFrom(const TcpServer &that);
|
|||
|
|
virtual TcpServer::Ptr onCreatServer(const EventPoller::Ptr &poller);
|
|||
|
|
|
|||
|
|
virtual Session::Ptr onAcceptConnection(const Socket::Ptr &sock);
|
|||
|
|
virtual Socket::Ptr onBeforeAcceptConnection(const EventPoller::Ptr &poller);
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
void onManagerSession();
|
|||
|
|
Socket::Ptr createSocket(const EventPoller::Ptr &poller);
|
|||
|
|
void start_l(uint16_t port, const std::string &host, uint32_t backlog);
|
|||
|
|
Ptr getServer(const EventPoller *) const;
|
|||
|
|
void setupEvent();
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
bool _multi_poller;
|
|||
|
|
bool _is_on_manager = false;
|
|||
|
|
bool _main_server = true;
|
|||
|
|
std::weak_ptr<TcpServer> _parent;
|
|||
|
|
Socket::Ptr _socket;
|
|||
|
|
std::shared_ptr<Timer> _timer;
|
|||
|
|
Socket::onCreateSocket _on_create_socket;
|
|||
|
|
std::unordered_map<SessionHelper *, SessionHelper::Ptr> _session_map;
|
|||
|
|
std::function<SessionHelper::Ptr(const TcpServer::Ptr &server, const Socket::Ptr &)> _session_alloc;
|
|||
|
|
std::unordered_map<const EventPoller *, Ptr> _cloned_server;
|
|||
|
|
//对象个数统计
|
|||
|
|
ObjectStatistic<TcpServer> _statistic;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
} /* namespace toolkit */
|
|||
|
|
#endif /* TCPSERVER_TCPSERVER_H */
|