Rooba
/
drogon
mirror of https://github.com/drogonframework/drogon.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Check whether the port is in use

This commit is contained in:
an-tao 2019-04-27 14:45:06 +08:00
parent 0b96089a0a
commit 87c51d1b53
1 changed files with 43 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
lib/src/HttpAppFrameworkImpl.cc
View File

@ -41,6 +41,7 @@
#include <utility> #include <utility>
#include <unistd.h> #include <unistd.h>
#include <fcntl.h> #include <fcntl.h>
#include <sys/file.h>
using namespace drogon; using namespace drogon;
using namespace std::placeholders; using namespace std::placeholders;
@ -51,7 +52,27 @@ drogon::InitBeforeMainFunction drogon::HttpAppFrameworkImpl::_initFirst([]() {
LOG_TRACE << "Initialize the main event loop in the main thread"; LOG_TRACE << "Initialize the main event loop in the main thread";
}); });
}); });
namespace drogon
{
class DrogonFileLocker : public trantor::NonCopyable
{
public:
DrogonFileLocker()
{
_fd = open("/tmp/drogon.lock", O_TRUNC | O_CREAT, 0755);
flock(_fd, LOCK_EX);
}
~DrogonFileLocker()
{
close(_fd);
}
private:
int _fd = 0;
};
} // namespace drogon
static void godaemon(void) static void godaemon(void)
{ {
printf("Initializing daemon mode\n"); printf("Initializing daemon mode\n");
@ -297,9 +318,27 @@ void HttpAppFrameworkImpl::run()
{ {
auto ip = std::get<0>(listener); auto ip = std::get<0>(listener);
bool isIpv6 = ip.find(":") == std::string::npos ? false : true; bool isIpv6 = ip.find(":") == std::string::npos ? false : true;
auto serverPtr = std::make_shared<HttpServer>(loopThreadPtr->getLoop(), std::shared_ptr<HttpServer> serverPtr;
InetAddress(ip, std::get<1>(listener), isIpv6), if (i == 0)
"drogon"); {
DrogonFileLocker lock;
// Check whether the port is in use.
TcpServer server(getLoop(),
InetAddress(ip, std::get<1>(listener), isIpv6),
"drogonPortTest",
true,
false);
serverPtr = std::make_shared<HttpServer>(loopThreadPtr->getLoop(),
InetAddress(ip, std::get<1>(listener), isIpv6),
"drogon");
}
else
{
serverPtr = std::make_shared<HttpServer>(loopThreadPtr->getLoop(),
InetAddress(ip, std::get<1>(listener), isIpv6),
"drogon");
}
if (std::get<2>(listener)) if (std::get<2>(listener))
{ {
#ifdef USE_OPENSSL #ifdef USE_OPENSSL
@ -566,7 +605,7 @@ void HttpAppFrameworkImpl::onNewWebsockRequest(const HttpRequestImplPtr &req,
std::vector<std::tuple<std::string, HttpMethod, std::string>> HttpAppFrameworkImpl::getHandlersInfo() const std::vector<std::tuple<std::string, HttpMethod, std::string>> HttpAppFrameworkImpl::getHandlersInfo() const
{ {
auto ret = _httpSimpleCtrlsRouter.getHandlersInfo(); auto ret = _httpSimpleCtrlsRouter.getHandlersInfo();
auto v=_httpCtrlsRouter.getHandlersInfo(); auto v = _httpCtrlsRouter.getHandlersInfo();
ret.insert(ret.end(), v.begin(), v.end()); ret.insert(ret.end(), v.begin(), v.end());
v = _websockCtrlsRouter.getHandlersInfo(); v = _websockCtrlsRouter.getHandlersInfo();
ret.insert(ret.end(), v.begin(), v.end()); ret.insert(ret.end(), v.begin(), v.end());