C++ Download类代码示例

bool DownloadManagerImpl::removeAndUpdateDownload(int downloadId, DownloadManager::DownloadStatus status)
{
    Download *download = getDownloadByDownloadId(downloadId);
    if (!download) {
        qDebug() << "Could not find downloadId = " << downloadId;

        // check download
        checkDownloadQueue();

        return false;
    }

    // Update status
    download->setDownloadStatus(status);
    if (!m_downloadDAO->updateDownload(download))
        qDebug() << __PRETTY_FUNCTION__ << " Could not update information for download with id = " << downloadId;

    // Remove the download from the downloading list
    removeDownloadById(downloadId);

    qDebug() << "Remove the download with downloadId = " << downloadId;

    // check download
    checkDownloadQueue();

    return true;
}

void DownloadManager::checkDownloads(UserConnection* aConn) {
    dcassert(aConn->getDownload() == NULL);

    QueueItem::Priority prio = QueueManager::getInstance()->hasDownload(aConn->getUser());
    if(!startDownload(prio)) {
        removeConnection(aConn);
        return;
    }

    Download* d = QueueManager::getInstance()->getDownload(*aConn, aConn->isSet(UserConnection::FLAG_SUPPORTS_TTHL));

    if(!d) {
        Lock l(cs);
        aConn->setState(UserConnection::STATE_IDLE);
        idlers.push_back(aConn);
        return;
    }

    aConn->setState(UserConnection::STATE_SND);

    if(aConn->isSet(UserConnection::FLAG_SUPPORTS_XML_BZLIST) && d->getType() == Transfer::TYPE_FULL_LIST) {
        d->setFlag(Download::FLAG_XML_BZ_LIST);
    }

    {
        Lock l(cs);
        downloads.push_back(d);
    }
    fire(DownloadManagerListener::Requesting(), d);

    dcdebug("Requesting " I64_FMT "/" I64_FMT "\n", static_cast<long long int>(d->getStartPos()), static_cast<long long int>(d->getSize()));

    aConn->send(d->getCommand(aConn->isSet(UserConnection::FLAG_SUPPORTS_ZLIB_GET)));
}

void DownloadManagerImpl::release()
{
    m_mutexLocker.lock();
    // Free downloads in the downloading list
    foreach(Download *download, m_downloadingList) {
        if (download == 0)
            continue;
        download->stop();
        download->deleteLater();
        download = 0;
    }
    m_downloadingList.clear();

    for (int i = 0; i < m_downloadQueue.count(); i++) {
        Download *download = m_downloadQueue.at(i);
        if (download == 0)
            continue;
        download->stop();
        download->deleteLater();
        download = 0;
    }
    m_downloadQueue.clear();
    m_mutexLocker.unlock();

    if (m_dbConnection) {
        m_dbConnection->close();
        delete m_dbConnection;
        m_dbConnection = 0;
    }

    if (m_downloadDAO) {
        delete m_downloadDAO;
        m_downloadDAO = 0;
    }
}

Download *DownloadManagerImpl::getDownloadByContactId(int contactId)
{
    QMutexLocker locker(&m_mutexLocker);
    for (int i = 0; i < m_downloadQueue.count(); i++) {
        Download *download = m_downloadQueue.at(i);
        if (download == 0)
            continue;
        Contact *contact = download->getContact();
        if (contact == 0)
            continue;
        if (contact->getId() == contactId)
            return download;
    }

    foreach(Download *download, m_downloadingList) {
        if (download == 0)
            continue;
        Contact *contact = download->getContact();
        if (contact == 0)
            continue;
        if (contact->getId() == contactId)
            return download;
    }

    return 0;
}

FileDownloader::Download& FileDownloader::DownloadFile(std::string url) {
    Download* download = new FileDownloader::Download();
    download->container = this;
    download->url = url;
    activeDownloads.push_back(download);
    
    MemoryStruct mem;
    mem.memory = 0;
    mem.size = 0;
    
    CURL* curl = (CURL*)this->curl;
    
    curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, &mem);
    CURLcode res = curl_easy_perform(curl);
    
    DownloadedFile downloadedFile;
    downloadedFile.url = download->url;
    downloadedFile.data = mem.memory;
    downloadedFile.size = (int)mem.size;
    download->Completed(downloadedFile);
    download->container->activeDownloads.erase(std::find(download->container->activeDownloads.begin(), download->container->activeDownloads.end(), download));
    delete download;
    DownloadComplete(downloadedFile);
    return *download;
}

int DownloadManagerImpl::getCurrentRemainTimeByContact(int contactId)
{
    Download *download = getDownloadByContactId(contactId);
    if (download)
        return download->getCurrentRemainTime();

    return -1;
}

void didCreateDestinationCallback(CFURLDownloadRef, CFURLRef, const void* clientInfo)
{
    // The concept of the ".download bundle" is internal to the Download, so we try to hide its
    // existence by reporting the final destination was created, when in reality the bundle was created.

    Download* download = downloadFromClientInfo(clientInfo);
    download->didCreateDestination(download->destination());
}

OSStatus SecureDownloadFinished (SecureDownloadRef downloadRef)
{
	API_BEGIN
	Required (downloadRef);
	Download* d = (Download*) downloadRef;
	d->Finalize ();
	API_END
}

void DownloadManager::cancelDownload(DownloadID downloadID)
{
    Download* download = m_downloads.get(downloadID);
    if (!download)
        return;

    download->cancel();
}

OSStatus SecureDownloadUpdateWithData (SecureDownloadRef downloadRef, CFDataRef data)
{
	API_BEGIN
	Required (downloadRef);
	Required (data);
	Download* d = (Download*) downloadRef;
	d->UpdateWithData (data);
	API_END
}

int64_t DownloadManager::getRunningAverage() {
    Lock l(cs);
    int64_t avg = 0;
    for(DownloadList::iterator i = downloads.begin(); i != downloads.end(); ++i) {
        Download* d = *i;
        avg += d->getAverageSpeed();
    }
    return avg;
}

OSStatus SecureDownloadCopyCreationDate (SecureDownloadRef downloadRef, CFDateRef* date)
{
	API_BEGIN

	Required (downloadRef);
	Download* d = (Download*) downloadRef;
	Required (date) = d->CopyDate ();

	API_END
}

OSStatus SecureDownloadCopyName (SecureDownloadRef downloadRef, CFStringRef* name)
{
	API_BEGIN

	Required (downloadRef);
	Download* d = (Download*) downloadRef;
	Required (name) = d->CopyName ();

	API_END
}

OSStatus SecureDownloadCopyURLs (SecureDownloadRef downloadRef, CFArrayRef* urls)
{
	API_BEGIN

	Required (downloadRef);
	Download* d = (Download*) downloadRef;
	Required (urls) = d->CopyURLs ();

	API_END
}

OSStatus SecureDownloadGetDownloadSize (SecureDownloadRef downloadRef, SInt64* size)
{
	API_BEGIN

	Required (downloadRef);
	Download* d = (Download*) downloadRef;
	Required (size) = d->GetDownloadSize ();

	API_END
}