/** rpc **/ bool CallbackService::rpc(PACKET& packet, Object* req_param, RpcInfo* rpc_info){ if(packet.to == static_cast<uint64_t>(m_id)){ WARN("service %s(%lld) who %lld fail to rpc to %lld, can't request self", name(), (long long)m_id, (long long)packet.who, (long long)packet.to); } if(set_rpc(rpc_info) > 0){ packet.sn =static_cast<uint64_t>(rpc_info->getId()); if(DispatcherManager::RequestByObject(this, packet, req_param)){ return true; } else{ del_rpc(rpc_info->getId()); WARN("service %s(%lld) who %lld fail to rpc to %lld, service not ready", name(), (long long)m_id, (long long)packet.who, (long long)packet.to); } } return false; }
/** rpc **/ Object* CoroutineService::rpc(const int64_t who, const int64_t to, const int64_t cmd, const int64_t res_proto_grp_id, Object* req_param){ if(to == getId()){ ERROR("service %s(%lld) can't request self", name(), (long long)m_id); return 0; } // prepare Coroutine* cr =Coroutine::Running(); if(!cr){ WARN("service %s(%lld) %lld fail to rpc to %lld, in main thread", name(), (long long)m_id, (long long)who, (long long)to); return 0; } ASSERT(cr->canYield()); const int64_t cr_id =cr->getId(); // set rpc info CoroutineRpcInfo* rpc_info =SafeNew<CoroutineRpcInfo>(res_proto_grp_id, cr_id); if(!set_rpc(rpc_info)){ return 0; } const int64_t rpc_id =rpc_info->getId(); // prepare packet PACKET packet; packet.from =getId(); packet.to =to; packet.who =who; packet.sn =rpc_id; packet.command =cmd; packet.option =0; // request if(!DispatcherManager::RequestByObject(this, packet, req_param)){ WARN("service %s(%lld) %lld fail to rpc to %lld, service not ready", name(), (long long)m_id, (long long)who, (long long)to); return 0; } rpc_info->set(rpc_id, this); ENSURE(cr->yield(0, rpc_id)); // process respond if(Command* respond =dynamic_cast< Command* >(cr->getResumeParam())){ // prepare const PACKET res_packet =respond->getPacket(); if(res_packet.sn != static_cast<uint64_t>(rpc_id)){ WARN("service %s(%lld) %lld fail to rpc to %lld, rpc id mismatch", name(), (long long)m_id, (long long)who, (long long)to); return 0; } // body is object pointer if(res_packet.option & OPT_BODY_IS_OBJECT_POINTER){ return respond; } // body is protocol if(res_proto_grp_id>0){ Bytes* res_bs =respond->getBody(); const int64_t res_cmd =static_cast<int64_t>(res_packet.command); const int64_t group_id =res_proto_grp_id; const int64_t protocol_id =res_cmd; ProtocolBase* res_proto = ProtocolManager::CreateProtocol(group_id, protocol_id); if(!res_proto){ ERROR("service %s(%lld) %lld fail to rpc to %lld, create protocol group %lld id %lld error", name(), (long long)m_id, (long long)who, (long long)to, (long long)group_id, (long long)protocol_id); return 0; } if(!res_proto->fromBytes(res_bs)){ ERROR("service %s(%lld) %lld fail to rpc to %lld, unmarshal protocol name %s group %lld id %lld error", name(), (long long)m_id, (long long)who, (long long)to, res_proto->name(), (long long)group_id, (long long)protocol_id); return 0; } respond->setRequest(res_proto); } // body is not special return respond; } else{ return cr->getResumeParam(); } }