source: sample/testdht/DHTTest.cpp @ 6760

#include "DHTTest.h"
#include "ariba/utility/configuration/Configuration.h"

using ariba::utility::Configuration;
using namespace ariba;

namespace ariba {
namespace application {
namespace dhttest {

// logging
use_logging_cpp( DHTTest );

// the service that the dhttest wants to use
ServiceID DHTTest::DHTTEST_SERVICEID = ServiceID( 111 );

// construction
DHTTest::DHTTest() {
        Timer::setInterval( 10000 );
}

// destruction
DHTTest::~DHTTest() {
}

// implementation of the startup interface
void DHTTest::startup() {

        logging_info( "starting up DHTTest service ... " );

        // create ariba module
        logging_debug( "creating ariba underlay module ... " );
        ariba = new AribaModule();

        Name spovnetName("dhttest");
        Name nodeName = Name::UNSPECIFIED;
        this->name = string("<dhttest>");

        // get settings from configuration object
        if( Configuration::haveConfig() ){
                Configuration& config = Configuration::instance();

                // get node name
                if (config.exists("node.name"))
                        nodeName = config.read<string> ("node.name");

                // configure ariba module
                if (config.exists("ariba.endpoints"))
                        ariba->setProperty("endpoints", config.read<string>("ariba.endpoints"));
                if (config.exists("ariba.bootstrap.hints"))
                        ariba->setProperty("bootstrap.hints", config.read<string>("ariba.bootstrap.hints"));
                if (config.exists("dhttest.name"))
                        name = config.read<string>("dhttest.name");

                // configure test parameteres
                if (config.exists("dhttest.key"))
                        key = config.read<string>("dhttest.key");
                if (config.exists("dhttest.data"))
                        data = config.read<string>("dhttest.data");

        } // if( Configuration::haveConfig() )

        // start ariba module
        ariba->start();

        // create node and join
        node = new Node( *ariba, nodeName );

        // bind communication and node listener
        node->bind( this );                              /*NodeListener*/
        //node->bind( this, DHTTest::DHTTEST_SERVICEID); /*CommunicationListener*/

        // start node module
        node->start();

        // when initiating, you can define the overlay type, default is Chord [CHORD_OVERLAY]
        SpoVNetProperties params;
        //params.setBaseOverlayType( SpoVNetProperties::ONE_HOP_OVERLAY ); // alternative: OneHop

        // initiate the spovnet
        logging_info("initiating spovnet");
        node->initiate(spovnetName, params);

        // join the spovnet
        logging_info("joining spovnet");
        node->join(spovnetName);

        // dht test started up...
        logging_info( "dhttest starting up with"
                        << " [spovnetid " << node->getSpoVNetId().toString() << "]"
                        << " and [nodeid " << node->getNodeId().toString() << "]" );
}

// implementation of the startup interface
void DHTTest::shutdown() {

        logging_info( "dhttest service starting shutdown sequence ..." );

        // stop timer
        Timer::stop();

        // leave spovnet
        node->leave();

        // unbind communication and node listener
        node->unbind( this );                               /*NodeListener*/
        //node->unbind( this, DHTTest::DHTTEST_SERVICEID ); /*CommunicationListener*/

        // stop the ariba module
        ariba->stop();

        // delete node and ariba module
        delete node;
        delete ariba;

        // now we are completely shut down
        logging_info( "dhttest service shut down" );
}

// timer event
void DHTTest::eventFunction() {

        switch(counter) {
        case 1:
                logging_info( "Putting '" << key << "'->'" << data << "' in the dht.");

                node->put(stod(key), stod(data), 3600);
                break;
        default:
                logging_info( "Trying to get '" << key << "' from the DHT. This is try number " << counter-1 <<".");

                node->get(stod(key), DHTTEST_SERVICEID);
                break;
        }

        counter++;

}

void DHTTest::onJoinCompleted( const SpoVNetID& vid ) {
        logging_info( "dhttest node join completed, spovnetid=" << vid.toString() );

        // key is set -> this node will be testing
        if(!key.empty()) {
                counter = 0;

                // start the timer to ping every second
                Timer::start();
        }

}

void DHTTest::onJoinFailed( const SpoVNetID& vid ) {
        logging_error("dhttest node join failed, spovnetid=" << vid.toString() );
}

void DHTTest::onLeaveCompleted( const SpoVNetID& vid ){
        logging_info("dhttest node leave completed, spovnetid=" << vid.toString() );
}

void DHTTest::onLeaveFailed( const SpoVNetID& vid ){
        logging_error("dhttest node leave failed, spovnetid=" << vid.toString() );
}


bool DHTTest::onLinkRequest(const NodeID& remote, const DataMessage& msg) {
        logging_info( "node " << remote.toString() << " wants to build up a link with us ... allowing" );
        return true;
}

void DHTTest::onLinkUp(const LinkID& lnk, const NodeID& remote){
        logging_info( "received link-up event for link " << lnk.toString()
                        << " and node " << remote.toString() );
}

void DHTTest::onLinkDown(const LinkID& lnk, const NodeID& remote){
        logging_info( "received link-down event for link " << lnk.toString()
                        << " and node " << remote.toString() );
}

void DHTTest::onLinkChanged(const LinkID& lnk, const NodeID& remote){
        logging_info( "link-changed event for link " << lnk.toString()
                        << " and node " << remote.toString() );
}

void DHTTest::onLinkFail(const LinkID& lnk, const NodeID& remote){
        logging_info( "received link-failed event for link " << lnk.toString()
                        << " and node " << remote.toString() );
}

void DHTTest::onKeyValue( const Data& key, const vector<Data>& value ) {
        logging_info( "----> Received DHT answer for '" << dtos(key) << "'.");
        // TODO: implement
}

Data DHTTest::stod(string s) {
        return Data((uint8_t*)s.data(), s.length()*8).clone();
}

string DHTTest::dtos(Data d) {
        return string((char*)d.getBuffer(), d.getLength()/8);
}

}}} // namespace ariba, application, dhttest