Method StartClient

src/plugin/net/src/AFCWebSocketClient.cpp:34–126

}

bool AFCWebSocketClient::StartClient(
    AFHeadLength len, const int target_busid, const std::string& ip, const int port, bool ip_v6)
{
    using namespace brynet::net;
    using namespace brynet::net::http;

    dst_bus_id_ = target_busid;

    auto shared_this = shared_from_this();
    auto OnEnterCallback = [shared_this, len, ip](const HttpSession::Ptr& httpSession) {
        brynet::net::http::HttpRequest request;
        request.setMethod(HttpRequest::HTTP_METHOD::HTTP_METHOD_GET);
        request.setUrl("/ws");
        request.addHeadValue("Host", ip);
        request.addHeadValue("Upgrade", "websocket");
        request.addHeadValue("Connection", "Upgrade");
        request.addHeadValue("Sec-WebSocket-Key", "dGhlIHNhbXBsZSBub25jZQ==");
        request.addHeadValue("Sec-WebSocket-Version", "13");

        std::string requestStr = request.getResult();
        httpSession->send(requestStr.c_str(), requestStr.size());

        httpSession->setWSConnected([shared_this, &len, &ip](const HttpSession::Ptr& httpSession, const HTTPParser&) {
            // now session_id
            //int64_t cur_session_id = this->trust_session_id_++;
            auto cur_session_id = shared_this->uid_generator_->GetUID(shared_this->dst_bus_id_);
            // create net event
            AFNetEvent* net_connect_event = AFNetEvent::AllocEvent();
            net_connect_event->SetId(cur_session_id);
            net_connect_event->SetType(AFNetEventType::CONNECTED);
            net_connect_event->SetBusId(shared_this->dst_bus_id_);
            net_connect_event->SetIP(ip);

            // set session ud
            httpSession->setUD(cur_session_id);

            // create session and net event
            {
                AFScopeWLock guard(shared_this->rw_lock_);

                AFHttpSessionPtr session_ptr = ARK_NEW AFHttpSession(len, cur_session_id, httpSession);
                shared_this->client_session_ptr_.reset(session_ptr);
                session_ptr->AddNetEvent(net_connect_event);
            }
        });

        httpSession->setWSCallback([shared_this](const HttpSession::Ptr& httpSession,
                                       WebSocketFormat::WebSocketFrameType opcode, const std::string& payload) {
            const auto ud = cast<int64_t>(httpSession->getUD());
            int64_t session_id = *ud;

            // Scope lock
            {
                AFScopeRLock xGuard(shared_this->rw_lock_);
                shared_this->client_session_ptr_->AddBuffer(payload.c_str(), payload.size());
                shared_this->client_session_ptr_->ParseBufferToMsg();
            }
        });