235 lines
8.4 KiB
C++
235 lines
8.4 KiB
C++
/**
|
|
* Copyright (c) 2017 Baidu.com, Inc. All Rights Reserved
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
|
|
* an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
* specific language governing permissions and limitations under the License.
|
|
*
|
|
* @author baidu aip
|
|
*/
|
|
#ifndef __AIP_HTTP_H__
|
|
#define __AIP_HTTP_H__
|
|
|
|
#include "curl/curl.h"
|
|
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <map>
|
|
#include <json/json.h>
|
|
|
|
namespace aip {
|
|
|
|
inline size_t onWriteData(void * buffer, size_t size, size_t nmemb, void * userp)
|
|
{
|
|
std::string * str = dynamic_cast<std::string *>((std::string *)userp);
|
|
str->append((char *)buffer, size * nmemb);
|
|
return nmemb;
|
|
}
|
|
|
|
class HttpClient
|
|
{
|
|
private:
|
|
bool debug = false;
|
|
int connect_timeout = 10000;
|
|
int socket_timeout = 10000;
|
|
|
|
void makeUrlencodedForm(std::map<std::string, std::string> const & params, std::string * content) const
|
|
{
|
|
content->clear();
|
|
std::map<std::string, std::string>::const_iterator it;
|
|
for(it=params.begin(); it!=params.end(); it++)
|
|
{
|
|
char * key = curl_escape(it->first.c_str(), (int) it->first.size());
|
|
char * value = curl_escape(it->second.c_str(),(int) it->second.size());
|
|
*content += key;
|
|
*content += '=';
|
|
*content += value;
|
|
*content += '&';
|
|
curl_free(key);
|
|
curl_free(value);
|
|
}
|
|
}
|
|
|
|
void appendUrlParams(std::map<std::string, std::string> const & params, std::string* url) const
|
|
{
|
|
if(params.empty()) {
|
|
return;
|
|
}
|
|
std::string content;
|
|
this->makeUrlencodedForm(params, &content);
|
|
bool url_has_param = false;
|
|
for (const auto& ch : *url) {
|
|
if (ch == '?') {
|
|
url_has_param = true;
|
|
break;
|
|
}
|
|
}
|
|
if (url_has_param) {
|
|
url->append("&");
|
|
} else {
|
|
url->append("?");
|
|
}
|
|
url->append(content);
|
|
}
|
|
|
|
void appendHeaders(std::map<std::string, std::string> const & headers, curl_slist ** slist) const
|
|
{
|
|
std::ostringstream ostr;
|
|
std::map<std::string, std::string>::const_iterator it;
|
|
for(it=headers.begin(); it!=headers.end(); it++)
|
|
{
|
|
ostr << it->first << ":" << it->second;
|
|
*slist = curl_slist_append(*slist, ostr.str().c_str());
|
|
ostr.str("");
|
|
}
|
|
}
|
|
|
|
public:
|
|
HttpClient() = default;
|
|
|
|
HttpClient(const HttpClient &) = delete;
|
|
HttpClient & operator=(const HttpClient &) = delete;
|
|
|
|
void setConnectTimeout(int connect_timeout)
|
|
{
|
|
this->connect_timeout = connect_timeout;
|
|
}
|
|
|
|
void setSocketTimeout(int socket_timeout)
|
|
{
|
|
this->socket_timeout = socket_timeout;
|
|
}
|
|
|
|
void setDebug(bool debug)
|
|
{
|
|
this->debug = debug;
|
|
}
|
|
|
|
int get(
|
|
std::string url,
|
|
std::map<std::string, std::string> const * params,
|
|
std::map<std::string, std::string> const * headers,
|
|
std::string * response) const
|
|
{
|
|
CURL * curl = curl_easy_init();
|
|
struct curl_slist * slist = NULL;
|
|
if (headers) {
|
|
this->appendHeaders(*headers, &slist);
|
|
}
|
|
if (params) {
|
|
this->appendUrlParams(*params, &url);
|
|
}
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
|
|
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, slist);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, onWriteData);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *) response);
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, true);
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT_MS, this->connect_timeout);
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, this->socket_timeout);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, false);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, false);
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, this->debug);
|
|
|
|
int status_code = curl_easy_perform(curl);
|
|
|
|
curl_easy_cleanup(curl);
|
|
curl_slist_free_all(slist);
|
|
|
|
return status_code;
|
|
}
|
|
|
|
int post(
|
|
std::string url,
|
|
std::map<std::string, std::string> const * params,
|
|
const std::string & body,
|
|
std::map<std::string, std::string> const * headers,
|
|
std::string * response) const
|
|
{
|
|
struct curl_slist * slist = NULL;
|
|
CURL * curl = curl_easy_init();
|
|
if (headers) {
|
|
this->appendHeaders(*headers, &slist);
|
|
}
|
|
if (params) {
|
|
this->appendUrlParams(*params, &url);
|
|
}
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL, url.c_str());
|
|
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, slist);
|
|
curl_easy_setopt(curl, CURLOPT_POST, true);
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, body.c_str());
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, body.size());
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, onWriteData);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, (void *) response);
|
|
curl_easy_setopt(curl, CURLOPT_NOSIGNAL, true);
|
|
curl_easy_setopt(curl, CURLOPT_CONNECTTIMEOUT_MS, this->connect_timeout);
|
|
curl_easy_setopt(curl, CURLOPT_TIMEOUT_MS, this->socket_timeout);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, false);
|
|
curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, false);
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, this->debug);
|
|
|
|
int status_code = curl_easy_perform(curl);
|
|
|
|
curl_easy_cleanup(curl);
|
|
curl_slist_free_all(slist);
|
|
|
|
return status_code;
|
|
}
|
|
|
|
int post(
|
|
std::string url,
|
|
std::map<std::string, std::string> const * params,
|
|
std::map<std::string, std::string> const & data,
|
|
std::map<std::string, std::string> const * headers,
|
|
std::string * response) const
|
|
{
|
|
|
|
std::string body;
|
|
this->makeUrlencodedForm(data, &body);
|
|
return this->post(std::move(url), params, body, headers, response);
|
|
}
|
|
|
|
int post(
|
|
std::string url,
|
|
std::map<std::string, std::string> const * params,
|
|
Json::Value const & data,
|
|
std::map<std::string, std::string> const * headers,
|
|
std::string * response) const
|
|
{
|
|
std::string body;
|
|
Json::StreamWriterBuilder swb;
|
|
std::unique_ptr<Json::StreamWriter> writer(swb.newStreamWriter());
|
|
std::ostringstream os;
|
|
writer->write(data, &os);
|
|
body = os.str();
|
|
std::map<std::string, std::string> temp_headers;
|
|
if (headers) {
|
|
std::map<std::string, std::string> temp_headers(*headers);
|
|
}
|
|
|
|
temp_headers["Content-Type"] = "application/json";
|
|
return this->post(url.c_str(), params, body, &temp_headers, response);
|
|
}
|
|
|
|
|
|
int post(
|
|
std::string url,
|
|
std::map<std::string, std::string> const * params,
|
|
std::map<std::string, std::string> const * headers,
|
|
std::string * response) const
|
|
{
|
|
const static std::string EMPTY_STRING;
|
|
return this->post(std::move(url), params, EMPTY_STRING, headers, response);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|