libcurl: показывает скорость загрузки и выгрузки

Question

У меня есть класс, который реализует file upload и file download, используя libcurl. Код ниже.

структура памяти

    struct MemoryStruct {
    uint8_t* memory;
    size_t size;
  };

функция обратного вызова

size_t handleData(void* contents, size_t size, size_t nmemb, void* stream) {
    size_t realsize = size * nmemb;
    struct MemoryStruct* mem = static_cast<struct MemoryStruct*>(stream);

    uint8_t* ptr = static_cast<uint8_t*>(realloc(mem->memory, mem->size + realsize + 1));
    if (ptr == NULL) {
        return 0;
    }
    mem->memory = ptr;
    memcpy(&(mem->memory[mem->size]), contents, realsize);
    mem->size += realsize;
    mem->memory[mem->size] = 0;
    return realsize;
}

функция загрузки

bool download(const std::string& url, std::string& md5sum) {
        curl_off_t speed_download = 0;
        CURLcode res;
        struct MemoryStruct chunk{};

        chunk.memory = static_cast<uint8_t*>(malloc(1));
        if (nullptr == chunk.memory) {
            m_logger->errorf("malloc failed: Not enough memory");
            return false;
        }
        chunk.size = 0;

        res = curl_easy_setopt(m_curl, CURLOPT_URL, url.c_str());
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt (CURLOPT_URL) failed: %s", curl_easy_strerror(res));
            free(chunk.memory);
            return false;
        }
        res = curl_easy_setopt(m_curl, CURLOPT_WRITEFUNCTION, handleData);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt (CURLOPT_WRITEFUNCTION) failed: %s", curl_easy_strerror(res));
            free(chunk.memory);
            return false;
        }
        res = curl_easy_setopt(m_curl, CURLOPT_WRITEDATA, (void*) &chunk);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt (CURLOPT_WRITEDATA) failed: %s", curl_easy_strerror(res));
            free(chunk.memory);
            return false;
        }

        res = curl_easy_perform(m_curl);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_perform() failed: %s", curl_easy_strerror(res));
            free(chunk.memory);
            return false;
        }

        std::vector<uint8_t> readData(chunk.memory, (chunk.memory + chunk.size));
        md5sum = md5AsHex(readData);

        res = curl_easy_getinfo(m_curl, CURLINFO_SPEED_DOWNLOAD, &speed_download);
        if (CURLE_OK != res) {
            m_logger->errorf("curl_easy_getinfo(CURLINFO_SPEED_DOWNLOAD) failed: %s", curl_easy_strerror(res));
            free(chunk.memory);
            return false;
        }
        m_logger->infof("Average download speed: %" CURL_FORMAT_CURL_OFF_T
                        "megabyte/sec.\n", speed_download / (1024 * 1024));
        free(chunk.memory);
        return true;
    }

функция загрузки

 bool upload(const std::string& filename, const std::string& url) {
    CURLcode res;
    curl_off_t speed_upload = 0;

    std::ifstream in(filename);
    std::string data_contents((std::istreambuf_iterator<char>(in)), std::istreambuf_iterator<char>());

    if (m_curl) {
        res = curl_easy_setopt(m_curl, CURLOPT_URL, url.c_str());
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt(CURLOPT_URL) failed: %s", curl_easy_strerror(res));
            return false;
        }

        res = curl_easy_setopt(m_curl, CURLOPT_POSTFIELDS, data_contents.c_str());
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt(CURLOPT_POSTFIELDS) failed: %s", curl_easy_strerror(res));
            return false;
        }

        res = curl_easy_setopt(m_curl, CURLOPT_POSTFIELDSIZE, (long) strlen(data_contents.c_str()));
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt(CURLOPT_POSTFIELDSIZE) failed: %s", curl_easy_strerror(res));
            return false;
        }

        res = curl_easy_setopt(m_curl, CURLOPT_VERBOSE, 1L);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_setopt(CURLOPT_VERBOSE) failed: %s", curl_easy_strerror(res));
            return false;
        }

        res = curl_easy_perform(m_curl);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_perform() failed: %s", curl_easy_strerror(res));
            return false;
        }

        res = curl_easy_getinfo(m_curl, CURLINFO_SPEED_UPLOAD, &speed_upload);
        if (res != CURLE_OK) {
            m_logger->errorf("curl_easy_getinfo(CURLINFO_SPEED_UPLOAD) failed: %s", curl_easy_strerror(res));
            return false;
        }
        m_logger->infof("Average upload speed: %" CURL_FORMAT_CURL_OFF_T
                        " megabyte/sec.\n", speed_upload / (1024 * 1024));
        return true;
    }
    return false;
}

Я хочу знать текущую скорость загрузки / выгрузки, чтобы я мог отобразить ее на экране. В приведенных выше примерах я получаю upload и download в конце. Я хочу рассчитать скорость загрузки и выгрузки и отобразить что-то вроде индикатора выполнения. Как мне это сделать с помощью libcurl.

Answer 1

Я предлагаю вам проверить 'CURLOPT_PROGRESSFUNCTION' в документации. Вы можете легко реализовать такую ​​функциональность, используя ее.

https://curl.haxx.se/libcurl/c/CURLOPT_PROGRESSFUNCTION.html