code: Сохранение данных из сети в буфер

main
FilippovDY 11 месяцев назад
Родитель 2b2d225f47
Сommit 382e7bee90

@ -4,6 +4,8 @@
#include "text.h" #include "text.h"
#include "svg.h" #include "svg.h"
#include <curl/curl.h> #include <curl/curl.h>
#include <sstream>
#include <string>
using namespace std; using namespace std;
@ -41,34 +43,46 @@ input_data(istream& stream, bool prompt)
in.max_count = 0; in.max_count = 0;
return in; return in;
} }
size_t write_data(void* items, size_t item_size, size_t item_count, void* ctx)
{
size_t data_size = item_size * item_count;
int main(int argc, char* argv[]) stringstream* buffer = reinterpret_cast<stringstream*>(ctx);
auto text = reinterpret_cast<const char*>(items);
buffer->write(text, data_size);
return data_size;
}
Input download(const string& address)
{ {
if (argc > 1) stringstream buffer;
curl_global_init(CURL_GLOBAL_ALL);
CURL* curl = curl_easy_init();
if(curl)
{ {
CURL* curl = curl_easy_init(); CURLcode res;
if(curl) curl_easy_setopt(curl, CURLOPT_URL, address.c_str());
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_data);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &buffer);
res = curl_easy_perform(curl);
if (res != CURLE_OK)
{ {
CURLcode res; curl_easy_strerror(res);
curl_easy_setopt(curl, CURLOPT_URL, argv[1]); exit(1);
res = curl_easy_perform(curl);
if (res != CURLE_OK)
{
curl_easy_strerror(res);
exit(1);
}
curl_easy_cleanup(curl);
} }
return 0; curl_easy_cleanup(curl);
} }
curl_global_init(CURL_GLOBAL_ALL); return input_data(buffer, false);
bool prompt = true; }
auto in = input_data(cin, prompt); int main(int argc, char* argv[]){
auto bins = make_histogram(in.numbers, in.bin_count, in.number_count, in.max_count);
vector<double> test; Input in;
show_histogram_svg(bins, in.max_count, in.bin_count, test); if (argc > 1) {
return 0; in = download(argv[1]);
} else {
in = input_data(cin, true);
}
auto bins = make_histogram(in.numbers, in.bin_count, in.number_count, in.max_count);
vector<double> test;
show_histogram_svg(bins, in.max_count, in.bin_count, test);
return 0;
} }

Загрузка…
Отмена
Сохранить