Relintai/rcpp_framework
1
0
Fork 0
mirror of https://github.com/Relintai/rcpp_framework.git synced 2024-11-14 04:57:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

on_connection().

Browse Source
This commit is contained in:
Relintai 2021-07-06 15:11:03 +02:00
parent eab0802df2
commit 3b4a0ae80f
2 changed files with 31 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
modules/drogon/web_application.cpp
View File

@ -13,6 +13,7 @@
#include <stdlib.h>
#include <trantor/utils/AsyncFileLogger.h>
#include <trantor/net/TcpConnection.h>
#include <fcntl.h>
#include <sys/stat.h>
@ -276,8 +277,6 @@ void DWebApplication::run() {
std::bind(&DWebApplication::on_connection, this, std::placeholders::_1),
_idle_connection_timeout, _ssl_cert_path, _ssl_key_path, _ssl_conf_cmds, _thread_num, _sync_advices, _pre_sending_advices);
assert(ioLoops.size() == _thread_num);
for (size_t i = 0; i < _thread_num; ++i) {
@ -591,6 +590,8 @@ bool DWebApplication::reuse_port() const {
void DWebApplication::on_async_request(const HttpRequestImplPtr &req, std::function<void(const HttpResponsePtr &)> &&callback) {
LOG_INFO << "on_async_request";
LOG_TRACE << "new request:" << req->peerAddr().toIpPort() << "->"
<< req->localAddr().toIpPort();
LOG_TRACE << "Headers " << req->methodString() << " " << req->path();
@ -634,6 +635,8 @@ void DWebApplication::on_async_request(const HttpRequestImplPtr &req, std::funct
}
void DWebApplication::on_new_websock_request(const HttpRequestImplPtr &req, std::function<void(const HttpResponsePtr &)> &&callback, const WebSocketConnectionImplPtr &wsConnPtr) {
LOG_INFO << "on_new_websock_request";
/*
findSessionForRequest(req);
// Route to controller
@ -664,56 +667,60 @@ void DWebApplication::on_new_websock_request(const HttpRequestImplPtr &req, std:
}*/
}
void DWebApplication::on_connection(const trantor::TcpConnectionPtr &conn) {
LOG_INFO << "on_connection";
/*
static std::mutex mtx;
LOG_TRACE << "connect!!!" << maxConnectionNum_
<< " num=" << connectionNum_.load();
LOG_TRACE << "connect!!!" << _max_connection_num
<< " num=" << _connection_num.load();
if (conn->connected()) {
if (connectionNum_.fetch_add(1, std::memory_order_relaxed) >=
maxConnectionNum_) {
if (_connection_num.fetch_add(1, std::memory_order_relaxed) >= _max_connection_num) {
LOG_ERROR << "too much connections!force close!";
conn->forceClose();
return;
} else if (maxConnectionNumPerIP_ > 0) {
} else if (_max_connection_num_per_ip > 0) {
{
std::lock_guard<std::mutex> lock(mtx);
auto iter = connectionsNumMap_.find(conn->peerAddr().toIp());
if (iter == connectionsNumMap_.end()) {
connectionsNumMap_[conn->peerAddr().toIp()] = 1;
} else if (iter->second++ > maxConnectionNumPerIP_) {
conn->get_loop()->queueInLoop(
[conn]() { conn->forceClose(); });
auto iter = _connections_num_map.find(conn->peerAddr().toIp());
if (iter == _connections_num_map.end()) {
_connections_num_map[conn->peerAddr().toIp()] = 1;
} else if (iter->second++ > _max_connection_num_per_ip) {
conn->getLoop()->queueInLoop([conn]() { conn->forceClose(); });
return;
}
}
}
for (auto &advice : newConnectionAdvices_) {
for (auto &advice : _new_connection_advices) {
if (!advice(conn->peerAddr(), conn->localAddr())) {
conn->forceClose();
return;
}
}
} else {
if (!conn->hasContext()) {
// If the connection is connected to the SSL port and then
// disconnected before the SSL handshake.
return;
}
connectionNum_.fetch_sub(1, std::memory_order_relaxed);
if (maxConnectionNumPerIP_ > 0) {
_connection_num.fetch_sub(1, std::memory_order_relaxed);
if (_max_connection_num_per_ip > 0) {
std::lock_guard<std::mutex> lock(mtx);
auto iter = connectionsNumMap_.find(conn->peerAddr().toIp());
if (iter != connectionsNumMap_.end()) {
auto iter = _connections_num_map.find(conn->peerAddr().toIp());
if (iter != _connections_num_map.end()) {
--iter->second;
if (iter->second <= 0) {
connectionsNumMap_.erase(iter);
_connections_num_map.erase(iter);
}
}
}
}
*/
}
void DWebApplication::find_session_for_request(const HttpRequestImplPtr &req) {

1
modules/drogon/web_application.h
View File

@ -232,6 +232,7 @@ protected:
trantor::EventLoop *_loop;
std::vector<std::function<void()> > _beginning_advices;
std::vector<std::function<bool(const trantor::InetAddress &, const trantor::InetAddress &)> > _new_connection_advices;
std::vector<std::function<HttpResponsePtr(const HttpRequestPtr &)> > _sync_advices;
std::vector<std::function<void(const HttpRequestPtr &, const HttpResponsePtr &)> > _pre_sending_advices;
};