Train_Identify/nvidia_ascend_engine/common_engine/DataUploadEngine/ResultToHttpSrvEngine.cpp

#include "ResultToHttpSrvEngine.h"
#include "myutils.h"


ResultToHttpSrvEngine::ResultToHttpSrvEngine() {}
ResultToHttpSrvEngine::~ResultToHttpSrvEngine() {}

APP_ERROR ResultToHttpSrvEngine::Init()
{
    strPort0_ = engineName_ + "_" + std::to_string(engineId_) + "_0";
    strUsername_ = MyYaml::GetIns()->GetStringValue("username");
    strPassword_ = MyYaml::GetIns()->GetStringValue("password");
    strURL_ = MyYaml::GetIns()->GetStringValue("gc_http_url");
    strGetTokenURL_ = MyYaml::GetIns()->GetStringValue("gc_gettoken_url");
    strImageSrv_ = MyYaml::GetIns()->GetPathValue("gc_image_srv");
    strPoundNo_ = MyYaml::GetIns()->GetStringValue("atlas_poundno");
    strFailSavePath_ = MyYaml::GetIns()->GetPathValue("gc_result_path") + "httpfailcontent.csv";
    strFailSaveBakPath_ = MyYaml::GetIns()->GetPathValue("gc_result_path") + "httpfailcontent_bak.csv";
    LogInfo << "ResultToHttpSrvEngine Init ok";
    return APP_ERR_OK;
}

APP_ERROR ResultToHttpSrvEngine::DeInit()
{
    LogDebug << "curl_easy_cleanup";
    curl_easy_cleanup(pCurl_);

    /*  这个处理移动到main中， 防止多线程调用
    LogDebug << "内存清理";
    curl_global_cleanup();
    */
    LogInfo << "ResultToHttpSrvEngine DeInit ok";
    return APP_ERR_OK;
}

/**
* libcurl回调函数
* inParam : void *pBuffer   回调内容地址
          : size_t size     回调单个数据大小
          : size_t nmemb    回调数据个数               
* outParam: std::string &strResp    返回内容
* return  : 回调数据大小
*/
size_t ResultToHttpSrvEngine::WriteCallBack(void *pBuffer, size_t size, size_t nmemb, std::string &strResp)
{
    size_t sizes = size * nmemb;
    std::string strTemp((char*)pBuffer, sizes);
    strResp += strTemp;
    return sizes;
}

/**
* 调用http接口获取token
* inParam : 
* outParam: std::string &strBladeAuth   返回的token信息                      
* return  : true:成功; false:失败
*/
bool ResultToHttpSrvEngine::GetToken(std::string &strBladeAuth)
{
    //1. 获得curl操作符
    if (nullptr == pCurl_)
    {
        LogDebug<<"pCurl_ is null, invoke curl_easy_init";
        pCurl_ = curl_easy_init();
        if (nullptr == pCurl_)
        {
            LogError << "curl_easy_init failed !";
            return false;
        }
    }

    //2. 设置head, 和表单
    //设置head信息
    struct curl_slist *pHeaderList = nullptr;
    pHeaderList = curl_slist_append(pHeaderList, "Authorization:Basic Y2xpZW50X2VudGVycHJpc2U6Y2xpZW50X2VudGVycHJpc2Vfc2VjcmV0");

    //设置表单信息
    curl_mime *pMultipart = curl_mime_init(pCurl_);
    curl_mimepart *pPart = curl_mime_addpart(pMultipart);
    curl_mime_name(pPart, "username");
    curl_mime_data(pPart, strUsername_.c_str(), CURL_ZERO_TERMINATED);
    pPart = curl_mime_addpart(pMultipart);
    curl_mime_name(pPart, "password");
    curl_mime_data(pPart, strPassword_.c_str(), CURL_ZERO_TERMINATED);
    pPart = curl_mime_addpart(pMultipart);
    curl_mime_name(pPart, "tenantId");
    curl_mime_data(pPart, "000000", CURL_ZERO_TERMINATED);
    pPart = curl_mime_addpart(pMultipart);
    curl_mime_name(pPart, "grant_type");
    curl_mime_data(pPart, "password", CURL_ZERO_TERMINATED);

    curl_easy_setopt(pCurl_, CURLOPT_CONNECTTIMEOUT, 1);              //连接超时(1s连接不上服务器返回超时)
    curl_easy_setopt(pCurl_, CURLOPT_URL, strGetTokenURL_.c_str());   //设置url
    curl_easy_setopt(pCurl_, CURLOPT_HTTPHEADER, pHeaderList);        //设置报文头
    curl_easy_setopt(pCurl_, CURLOPT_MIMEPOST, pMultipart);           //设置表单
    //curl_easy_setopt(pCurl_, CURLOPT_POSTFIELDS, strBody.c_str());    //设置post内容
    //curl_easy_setopt(pCurl_, CURLOPT_POST, 1);                        //设置操作为POST(为非0表示post)
    curl_easy_setopt(pCurl_, CURLOPT_WRITEFUNCTION, WriteCallBack);   //设置回调函数
    std::string strResponse;
    curl_easy_setopt(pCurl_, CURLOPT_WRITEDATA, &strResponse);        //设置回调参数

    //3. 执行http请求
    CURLcode res = curl_easy_perform(pCurl_);
    curl_sl