Roxanne SkellyBuild fix03326cd7 years agoHistory
143 lines · 5.1 KB · C++
RawDownload
////  ThreadedAssignment.cpp//  libraries/shared/src////  Created by Stephen Birarda on 12/3/2013.//  Copyright 2013 High Fidelity, Inc.////  Distributed under the Apache License, Version 2.0.//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html// #include "ThreadedAssignment.h" #include <QtCore/QCoreApplication>#include <QtCore/QJsonArray>#include <QtCore/QJsonObject>#include <QtCore/QThread>#include <QtCore/QTimer> #include <LogHandler.h> #include "NetworkLogging.h" ThreadedAssignment::ThreadedAssignment(ReceivedMessage& message) :    Assignment(message),    _isFinished(false),    _domainServerTimer(this),    _statsTimer(this){    static const int STATS_TIMEOUT_MS = 1000;    _statsTimer.setInterval(STATS_TIMEOUT_MS); // 1s, Qt::CoarseTimer acceptable    connect(&_statsTimer, &QTimer::timeout, this, &ThreadedAssignment::sendStatsPacket);     connect(&_domainServerTimer, &QTimer::timeout, this, &ThreadedAssignment::checkInWithDomainServerOrExit);    _domainServerTimer.setInterval(DOMAIN_SERVER_CHECK_IN_MSECS); // 1s, Qt::CoarseTimer acceptable     // if the NL tells us we got a DS response, clear our member variable of queued check-ins    auto nodeList = DependencyManager::get<NodeList>();    connect(nodeList.data(), &NodeList::receivedDomainServerList, this, &ThreadedAssignment::clearQueuedCheckIns);    timestamp = p_high_resolution_clock::now();} void ThreadedAssignment::setFinished(bool isFinished) {    if (_isFinished != isFinished) {         _isFinished = isFinished;         if (_isFinished) {             qCDebug(networking) << "ThreadedAssignment::setFinished(true) called - finishing up.";             auto nodeList = DependencyManager::get<NodeList>();             auto& packetReceiver = nodeList->getPacketReceiver();             // we should de-register immediately for any of our packets            packetReceiver.unregisterListener(this);             // we should also tell the packet receiver to drop packets while we're cleaning up            packetReceiver.setShouldDropPackets(true);             // send a disconnect packet to the domain            nodeList->getDomainHandler().disconnect();             // stop our owned timers            _domainServerTimer.stop();            _statsTimer.stop();             // call our virtual aboutToFinish method - this gives the ThreadedAssignment subclass a chance to cleanup            aboutToFinish();             emit finished();        }    }} void ThreadedAssignment::commonInit(const QString& targetName, NodeType_t nodeType) {    // change the logging target name while the assignment is running    LogHandler::getInstance().setTargetName(targetName);     auto nodeList = DependencyManager::get<NodeList>();    nodeList->setOwnerType(nodeType);     // send a domain-server check in immediately and start the timer to fire them every DOMAIN_SERVER_CHECK_IN_MSECS    checkInWithDomainServerOrExit();    _domainServerTimer.start();     // start sending stats packet once we connect to the domain    connect(&nodeList->getDomainHandler(), &DomainHandler::connectedToDomain,            &_statsTimer, static_cast<void (QTimer::*)()>(&QTimer::start));     // stop sending stats if we disconnect    connect(&nodeList->getDomainHandler(), &DomainHandler::disconnectedFromDomain, &_statsTimer, &QTimer::stop);} void ThreadedAssignment::addPacketStatsAndSendStatsPacket(QJsonObject statsObject) {    auto nodeList = DependencyManager::get<NodeList>();     QJsonObject ioStats;    ioStats["inbound_kbps"] = nodeList->getInboundKbps();    ioStats["inbound_pps"] = nodeList->getInboundPPS();    ioStats["outbound_kbps"] = nodeList->getOutboundKbps();    ioStats["outbound_pps"] = nodeList->getOutboundPPS();     statsObject["io_stats"] = ioStats;     QJsonObject assignmentStats;    assignmentStats["numQueuedCheckIns"] = _numQueuedCheckIns;     assignmentStats["globalPostedEventCount"] = nodeList->getGlobalPostedEventCount();     statsObject["assignmentStats"] = assignmentStats;     nodeList->sendStatsToDomainServer(statsObject);} void ThreadedAssignment::sendStatsPacket() {    QJsonObject statsObject;    addPacketStatsAndSendStatsPacket(statsObject);} void ThreadedAssignment::checkInWithDomainServerOrExit() {    // verify that the number of queued check-ins is not >= our max    // the number of queued check-ins is cleared anytime we get a response from the domain-server     timestamp = p_high_resolution_clock::now();     if (_numQueuedCheckIns >= MAX_SILENT_DOMAIN_SERVER_CHECK_INS) {        qCDebug(networking) << "At least" << MAX_SILENT_DOMAIN_SERVER_CHECK_INS << "have been queued without a response from domain-server"            << "Stopping the current assignment";        stop();    } else {        auto nodeList = DependencyManager::get<NodeList>();        QMetaObject::invokeMethod(nodeList.data(), "sendDomainServerCheckIn", Qt::DirectConnection);         // increase the number of queued check ins        _numQueuedCheckIns++;    }} void ThreadedAssignment::domainSettingsRequestFailed() {    qCDebug(networking) << "Failed to retreive settings object from domain-server. Bailing on assignment.";    stop();}

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t