diff --git a/main.cpp b/main.cpp index 8d02ba5..7ef1894 100644 --- a/main.cpp +++ b/main.cpp @@ -1,4 +1,5 @@ #include "TCPServer.h" +#include #include std::atomic shouldStop = false; @@ -13,10 +14,9 @@ void signalHandler( int signum ) { int main(int argc, char* argv[]) { signal(SIGINT, signalHandler); - int port = 8080; - if (argc >= 2) { - port = std::stoi(argv[1]); - } + CLParser clParser(argc, argv); + + int port = clParser.getOption("port", 8080); TCPServer server(port);