Swiftdiff options
| author | Remko Tronçon <git@el-tramo.be> | 2011-03-17 19:37:31 (GMT) |
|---|---|---|
| committer | Remko Tronçon <git@el-tramo.be> | 2011-03-17 21:58:42 (GMT) |
| commit | 9cd05402468a7df3588ee1b1a8e1bf118eb72302 (patch) | |
| tree | e6de1c5446224bd5996450e392bacc92771e6a69 /Sluift | |
| parent | 27b05c969f191bcce130c8bc171cca94d6b50b64 (diff) | |
| download | swift-contrib-9cd05402468a7df3588ee1b1a8e1bf118eb72302.zip swift-contrib-9cd05402468a7df3588ee1b1a8e1bf118eb72302.tar.bz2 | |
Sluift: Add timeout to get and set.
Diffstat (limited to 'Sluift')
| -rw-r--r-- | Sluift/sluift.cpp | 70 |
1 files changed, 56 insertions, 14 deletions
diff --git a/Sluift/sluift.cpp b/Sluift/sluift.cpp index 458c44d..abfde24 100644 --- a/Sluift/sluift.cpp +++ b/Sluift/sluift.cpp @@ -94,15 +94,23 @@ class SluiftClient { client->sendPresence(boost::make_shared<Presence>(status)); } - std::string sendQuery(const JID& jid, IQ::Type type, const std::string& data) { + boost::optional<std::string> sendQuery(const JID& jid, IQ::Type type, const std::string& data, int timeout) { rawRequestResponse.reset(); RawRequest::ref request = RawRequest::create(type, jid, data, client->getIQRouter()); - request->onResponse.connect(boost::bind(&SluiftClient::handleRawRequestResponse, this, _1)); + boost::signals::scoped_connection c = request->onResponse.connect(boost::bind(&SluiftClient::handleRawRequestResponse, this, _1)); request->send(); - while (!rawRequestResponse) { + + Watchdog watchdog(timeout, networkFactories.getTimerFactory()); + while (!watchdog.getTimedOut() && !rawRequestResponse) { eventLoop.runUntilEvents(); } - return *rawRequestResponse; + + if (watchdog.getTimedOut()) { + return boost::optional<std::string>(); + } + else { + return *rawRequestResponse; + } } void disconnect() { @@ -285,21 +293,31 @@ static int sluift_client_get_contacts(lua_State *L) { static int sluift_client_get_version(lua_State *L) { try { SluiftClient* client = getClient(L); + int timeout = -1; + if (lua_type(L, 3) != LUA_TNONE) { + timeout = luaL_checknumber(L, 3); + } ResponseSink<SoftwareVersion> sink; GetSoftwareVersionRequest::ref request = GetSoftwareVersionRequest::create(std::string(luaL_checkstring(L, 2)), client->getClient()->getIQRouter()); request->onResponse.connect(boost::ref(sink)); request->send(); - while (!sink.hasResponse()) { + + Watchdog watchdog(timeout, networkFactories.getTimerFactory()); + while (!watchdog.getTimedOut() && !sink.hasResponse()) { eventLoop.runUntilEvents(); } - if (ErrorPayload::ref error = sink.getResponseError()) { + ErrorPayload::ref error = sink.getResponseError(); + if (error || watchdog.getTimedOut()) { lua_pushnil(L); - // TODO - if (error->getCondition() == ErrorPayload::RemoteServerNotFound) { + if (watchdog.getTimedOut()) { + lua_pushstring(L, "Timeout"); + } + else if (error->getCondition() == ErrorPayload::RemoteServerNotFound) { lua_pushstring(L, "Remote server not found"); } + // TODO else { lua_pushstring(L, "Error"); } @@ -353,15 +371,27 @@ static int sluift_client_get(lua_State *L) { SluiftClient* client = getClient(L); JID jid; std::string data; - if (lua_type(L, 3) != LUA_TNONE) { + int timeout = -1; + if (lua_type(L, 3) == LUA_TSTRING) { jid = JID(std::string(luaL_checkstring(L, 2))); data = std::string(luaL_checkstring(L, 3)); + if (lua_type(L, 4) != LUA_TNONE) { + timeout = luaL_checknumber(L, 4); + } } else { data = std::string(luaL_checkstring(L, 2)); + if (lua_type(L, 3) != LUA_TNONE) { + timeout = luaL_checknumber(L, 3); + } + } + boost::optional<std::string> result = client->sendQuery(jid, IQ::Get, data, timeout); + if (result) { + lua_pushstring(L, result->c_str()); + } + else { + lua_pushnil(L); } - std::string result = client->sendQuery(jid, IQ::Get, data); - lua_pushstring(L, result.c_str()); return 1; } catch (const SluiftException& e) { @@ -374,15 +404,27 @@ static int sluift_client_set(lua_State *L) { SluiftClient* client = getClient(L); JID jid; std::string data; - if (lua_type(L, 3) != LUA_TNONE) { + int timeout = -1; + if (lua_type(L, 3) == LUA_TSTRING) { jid = JID(std::string(luaL_checkstring(L, 2))); data = std::string(luaL_checkstring(L, 3)); + if (lua_type(L, 4) != LUA_TNONE) { + timeout = luaL_checknumber(L, 4); + } } else { data = std::string(luaL_checkstring(L, 2)); + if (lua_type(L, 3) != LUA_TNONE) { + timeout = luaL_checknumber(L, 3); + } + } + boost::optional<std::string> result = client->sendQuery(jid, IQ::Set, data, timeout); + if (result) { + lua_pushstring(L, result->c_str()); + } + else { + lua_pushnil(L); } - std::string result = client->sendQuery(jid, IQ::Set, data); - lua_pushstring(L, result.c_str()); return 1; } catch (const SluiftException& e) { |