game2006/server/robotserver/httpproxy.cc

#include "precompile.h"

#include <a8/sysutils.h>
#include <a8/mutable_xobject.h>
#include <a8/awaiter.h>
#include <a8/promise.h>

#include <f8/udplog.h>
#include <f8/jsonhttprequest.h>

#include "httpproxy.h"
#include "app.h"
#include "handlermgr.h"

#include "f8/httpclientpool.h"

struct HttpProxyRequest
{
    std::string req_id;
    f8::HttpProxyCb cb;
    std::string url;
    a8::XObject url_params;
    long long add_tick = 0;
};

class HttpProxyPromise : public a8::Promise
{
 public:
    HttpProxyPromise(const char* url, a8::XObject url_params, bool* ret, std::shared_ptr<a8::XObject>* rsp)
    {
        url_ = url;
        url_params_ = url_params;
        ret_ = ret;
        rsp_ = rsp;
    }

protected:

    virtual void DoAwait() override
    {
        f8::HttpClientPool::Instance()->HttpGet
            (
             [this, _self = shared_from_this()]
             (bool ok, a8::XObject* rsp_obj, f8::HttpContext* ctx)
             {
                 *ret_ = ok;
                 if (ok) {
                     **rsp_ = *rsp_obj;
                 }
                 DoDone();
             },
             url_.c_str(),
             url_params_,
             rand() % MAX_SYS_HTTP_NUM
             );
    }

 private:
    std::string url_;
    a8::XObject url_params_;
    bool* ret_ = nullptr;
    std::shared_ptr<a8::XObject>* rsp_ = nullptr;
};

static void _ProxyCallback(std::shared_ptr<f8::JsonHttpRequest> request)
{
    #ifdef MYDEBUG
    #if 0
    f8::UdpLog::Instance()->Debug("ProxyCallbBack request:%s",
                                  {
                                      request->GetParams()->ToJsonStr()
                                  });
    #endif
    #endif
    std::string seq_id = request->GetParams()->Get("seq_id");
    std::shared_ptr<HttpProxyRequest> req = HttpProxy::Instance()->GetRequest(seq_id);
    if (req) {
        a8::XObject data;

        data.ReadFromJsonString(request->GetParams()->Get("data").GetString());
        if (data.GetType() == a8::XOT_SIMPLE) {
            data.ReadFromJsonString("{}");
        }
        f8::HttpContext ctx;
        if (request->GetParams()->HasKey("errcode") &&
            request->GetParams()->Get("errcode").GetInt() == 0) {
            req->cb(true, &data, &ctx);
        } else {
            req->cb(false, request->GetParams().get(), &ctx);
        }
        HttpProxy::Instance()->DestoryRequest(req);
    }
}

void HttpProxy::Init()
{
    request_prefix_ = "robot2006_" + a8::XValue(a8::GetMilliSecond()).GetString() + "_";
    HandlerMgr::Instance()->RegisterGMMsgHandler("Proxy@callback", _ProxyCallback);
}

void HttpProxy::UnInit()
{

}

std::string HttpProxy::HttpGet(
                               f8::HttpProxyCb cb,
                               const char* url,
                               a8::XObject url_params
                               )
{
    std::shared_ptr<HttpProxyRequest> request = std::make_shared<HttpProxyRequest>();
    request->req_id = CreateRequestId();
    request->cb = cb;
    request->url = url;
    request->url_params = url_params;
    request->add_tick = a8::XGetTickCount();
    if (request_hash_.find(request->req_id) != request_hash_.end()) {
        abort();
    }
    request_hash_[request->req_id] = request;


    auto proxy_url_params = a8::MutableXObject::CreateObject();
    proxy_url_params->SetVal("seq_id", request->req_id);
    proxy_url_params->SetVal("target_url", std::string(url));
    proxy_url_params->SetVal("params", url_params.ToJsonStr());
    proxy_url_params->SetVal("cb_url",
                             a8::Format("http://%s:%d/webapp/index.php?c=Proxy&a=callback",
                                        {
                                         //"192.168.100.39",
                                         "127.0.0.1",
                                                3333
                                                //JsonDataMgr::Instance()->ip,
                                                //JsonDataMgr::Instance()->listen_port
                                                }));
    //std::string proxy_url = "http://192.168.100.21:8321/webapp/index.php?c=Proxy&a=get";
    std::string proxy_url = "http://127.0.0.1:8321/webapp/index.php?c=Proxy&a=get";
    //JsonDataMgr::Instance()->GetHttpProxyUrl(proxy_url);
    f8::HttpClientPool::Instance()->HttpGet
         (
          [request] (bool ok, a8::XObject* rsp_obj, f8::HttpContext* ctx)
         {
             long long cost_time = a8::XGetTickCount() - request->add_tick;
             if (ok) {
#ifdef MYDEBUG
                 #if 0
                 f8::UdpLog::Instance()->Debug("ProxyHttpGet ok cost_time:%d url:%s params:%s",
                                               {
                                                cost_time,
                                                request->url,
                                                request->url_params.ToJsonStr(),
                                               });
                 #endif
#endif
             } else {
                 f8::UdpLog::Instance()->Warning("ProxyHttpGet error cost_time:%d url:%s params:%s response:%s",
                                                 {
                                                  cost_time,
                                                  request->url,
                                                  request->url_params.ToJsonStr(),
                                                  ctx->response
                                                 });
                 request->cb(false, rsp_obj, ctx);
                 HttpProxy::Instance()->DestoryRequest(request);
             }
         },
         proxy_url.c_str(),
         *proxy_url_params,
         rand() % MAX_SYS_HTTP_NUM
         );
    return request->req_id;
}

std::string HttpProxy::CreateRequestId()
{
    return request_prefix_ + f8::App::Instance()->NewGlobalUuid();
}

std::shared_ptr<HttpProxyRequest> HttpProxy::GetRequest(const std::string& req_id)
{
    auto itr = request_hash_.find(req_id);
    return itr != request_hash_.end() ? itr->second : nullptr;
}

void HttpProxy::DestoryRequest(std::shared_ptr<HttpProxyRequest> request)
{
    request_hash_.erase(request->req_id);
}

std::shared_ptr<a8::Awaiter> HttpProxy::CoHttpGet(const char* url,
                                                  a8::XObject url_params,
                                                  bool* ret,
                                                  std::shared_ptr<a8::XObject>* rsp)
{
    *ret = false;
    *rsp = std::make_shared<a8::XObject>();
    return std::make_shared<HttpProxyPromise>(url, url_params, ret, rsp);
}