You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ktorrent/libktorrent/kademlia/task.cpp

135 lines
3.2 KiB

/***************************************************************************
* Copyright (C) 2005 by Joris Guisson *
* joris.guisson@gmail.com *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program; if not, write to the *
* Free Software Foundation, Inc., *
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. *
***************************************************************************/
#include <kresolver.h>
#include "task.h"
#include "kclosestnodessearch.h"
#include "rpcserver.h"
#include "kbucket.h"
using namespace KNetwork;
namespace dht
{
Task::Task(RPCServer* rpc,Node* node)
: node(node),rpc(rpc),outstanding_reqs(0),task_finished(false),queued(queued)
{
}
Task::~Task()
{
}
void Task::start(const KClosestNodesSearch & kns,bool queued)
{
// fill the todo list
for (KClosestNodesSearch::CItr i = kns.begin(); i != kns.end();i++)
todo.append(i->second);
this->queued = queued;
if (!queued)
update();
}
void Task::start()
{
if (queued)
{
queued = false;
update();
}
}
void Task::onResponse(RPCCall* c, MsgBase* rsp)
{
if (outstanding_reqs > 0)
outstanding_reqs--;
if (!isFinished())
{
callFinished(c,rsp);
if (canDoRequest() && !isFinished())
update();
}
}
void Task::onTimeout(RPCCall* c)
{
if (outstanding_reqs > 0)
outstanding_reqs--;
if (!isFinished())
{
callTimeout(c);
if (canDoRequest() && !isFinished())
update();
}
}
bool Task::rpcCall(MsgBase* req)
{
if (!canDoRequest())
return false;
RPCCall* c = rpc->doCall(req);
c->addListener(this);
outstanding_reqs++;
return true;
}
void Task::done()
{
task_finished = true;
finished(this);
}
void Task::emitDataReady()
{
dataReady(this);
}
void Task::kill()
{
task_finished = true;
finished(this);
}
void Task::addDHTNode(const TQString & ip,bt::Uint16 port)
{
KResolver::resolveAsync(this,TQ_SLOT(onResolverResults(KResolverResults )),
ip,TQString::number(port));
}
void Task::onResolverResults(KResolverResults res)
{
if (res.count() == 0)
return;
todo.append(KBucketEntry(res.front().address(),dht::Key()));
}
}
#include "task.moc"