From 82a34212908a749964deb419e6baf476a247ed84 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Remko=20Tron=C3=A7on?= <git@el-tramo.be> Date: Sat, 18 Dec 2010 11:16:05 +0100 Subject: Added initial BenchTool. diff --git a/Swiften/Client/ClientXMLTracer.h b/Swiften/Client/ClientXMLTracer.h index f3cfd08..43fdda3 100644 --- a/Swiften/Client/ClientXMLTracer.h +++ b/Swiften/Client/ClientXMLTracer.h @@ -8,12 +8,12 @@ #include <boost/bind.hpp> -#include "Swiften/Client/Client.h" +#include <Swiften/Client/CoreClient.h> namespace Swift { class ClientXMLTracer { public: - ClientXMLTracer(Client* client) { + ClientXMLTracer(CoreClient* client) { client->onDataRead.connect(boost::bind(&ClientXMLTracer::printData, '<', _1)); client->onDataWritten.connect(boost::bind(&ClientXMLTracer::printData, '>', _1)); } diff --git a/Swiften/Examples/BenchTool/.gitignore b/Swiften/Examples/BenchTool/.gitignore new file mode 100644 index 0000000..8220b3b --- /dev/null +++ b/Swiften/Examples/BenchTool/.gitignore @@ -0,0 +1 @@ +BenchTool diff --git a/Swiften/Examples/BenchTool/BenchTool.cpp b/Swiften/Examples/BenchTool/BenchTool.cpp new file mode 100644 index 0000000..a5c0925 --- /dev/null +++ b/Swiften/Examples/BenchTool/BenchTool.cpp @@ -0,0 +1,71 @@ +/* + * Copyright (c) 2010 Remko Tronçon + * Licensed under the GNU General Public License v3. + * See Documentation/Licenses/GPLv3.txt for more information. + */ + +#include <boost/bind.hpp> +#include <boost/thread.hpp> + +#include "Swiften/Client/Client.h" +#include "Swiften/Network/TimerFactory.h" +#include "Swiften/Network/BoostNetworkFactories.h" +#include "Swiften/EventLoop/EventLoop.h" +#include "Swiften/EventLoop/SimpleEventLoop.h" +#include "Swiften/Roster/GetRosterRequest.h" +#include "Swiften/Client/ClientXMLTracer.h" +#include <Swiften/Base/sleep.h> +#include <Swiften/TLS/BlindCertificateTrustChecker.h> + +using namespace Swift; + +SimpleEventLoop eventLoop; +BoostNetworkFactories networkFactories(&eventLoop); +int numberOfConnectedClients = 0; + +void handleConnected() { + numberOfConnectedClients++; + std::cout << "Connected " << numberOfConnectedClients << std::endl; +} + +int main(int, char**) { + int numberOfInstances = 1000; + + char* jid = getenv("SWIFT_BENCHTOOL_JID"); + if (!jid) { + std::cerr << "Please set the SWIFT_BENCHTOOL_JID environment variable" << std::endl; + return -1; + } + char* pass = getenv("SWIFT_BENCHTOOL_PASS"); + if (!pass) { + std::cerr << "Please set the SWIFT_BENCHTOOL_PASS environment variable" << std::endl; + return -1; + } + + BlindCertificateTrustChecker trustChecker; + std::vector<CoreClient*> clients; + for (int i = 0; i < numberOfInstances; ++i) { + CoreClient* client = new Swift::CoreClient(&eventLoop, &networkFactories, JID(jid), String(pass)); + client->setCertificateTrustChecker(&trustChecker); + client->onConnected.connect(&handleConnected); + clients.push_back(client); + } + + for (size_t i = 0; i < clients.size(); ++i) { + clients[i]->connect(); + } + + { + Timer::ref timer = networkFactories.getTimerFactory()->createTimer(30000); + timer->onTick.connect(boost::bind(&SimpleEventLoop::stop, &eventLoop)); + timer->start(); + + eventLoop.run(); + } + + for (size_t i = 0; i < clients.size(); ++i) { + delete clients[i]; + } + + return 0; +} diff --git a/Swiften/Examples/BenchTool/SConscript b/Swiften/Examples/BenchTool/SConscript new file mode 100644 index 0000000..bfd4cea --- /dev/null +++ b/Swiften/Examples/BenchTool/SConscript @@ -0,0 +1,9 @@ +import os + +Import("env") + +myenv = env.Clone() +myenv.UseFlags(myenv["SWIFTEN_FLAGS"]) +myenv.UseFlags(myenv["SWIFTEN_DEP_FLAGS"]) + +myenv.Program("BenchTool", ["BenchTool.cpp"]) diff --git a/Swiften/Examples/SConscript b/Swiften/Examples/SConscript index 7b9b491..9b9a35a 100644 --- a/Swiften/Examples/SConscript +++ b/Swiften/Examples/SConscript @@ -8,4 +8,5 @@ SConscript(dirs = [ "ConnectivityTest", "LinkLocalTool", "ParserTester", + "BenchTool", ]) -- cgit v0.10.2-6-g49f6