int ObClientManager::do_post_request(const ObServer& server, const int32_t pcode, const int32_t version, const int64_t session_id, const int64_t timeout, const ObDataBuffer& in_buffer, tbnet::IPacketHandler* handler, void* args) const { int rc = OB_SUCCESS; ObPacket* packet = new (std::nothrow) ObPacket(); if (NULL == packet) { rc = OB_ALLOCATE_MEMORY_FAILED; } else if (OB_SUCCESS != error_) { packet->free(); rc = error_; TBSYS_LOG(ERROR, "prev_error=%d", error_); } else { packet->set_packet_code(pcode); packet->setChannelId(0); packet->set_source_timeout(timeout); packet->set_session_id(session_id); packet->set_api_version(version); packet->set_data(in_buffer); if (timeout > max_request_timeout_) { max_request_timeout_ = timeout; connmgr_->setDefaultQueueTimeout(0, static_cast<int32_t>(max_request_timeout_ / 1000)); } rc = packet->serialize(); if (OB_SUCCESS != rc) { TBSYS_LOG(WARN, "packet serialize error"); packet->free(); packet = NULL; } else { rc = do_post_packet(server, packet, handler, args); } } return rc; }
int ObBaseServer::send_response(const int32_t pcode, const int32_t version, const ObDataBuffer& buffer, tbnet::Connection* connection, const int32_t channel_id) { int rc = OB_SUCCESS; if (connection == NULL) { rc = OB_ERROR; TBSYS_LOG(WARN, "connection is NULL"); } ObPacket* packet = new(std::nothrow) ObPacket(); if (packet == NULL) { rc = OB_ALLOCATE_MEMORY_FAILED; TBSYS_LOG(ERROR, "create packet failed"); } else { packet->set_packet_code(pcode); packet->setChannelId(channel_id); packet->set_api_version(version); packet->set_data(buffer); } if (rc == OB_SUCCESS) { rc = packet->serialize(); if (rc != OB_SUCCESS) TBSYS_LOG(WARN, "packet serialize error, error: %d", rc); } if (rc == OB_SUCCESS) { if (!connection->postPacket(packet)) { uint64_t peer_id = connection->getPeerId(); TBSYS_LOG(WARN, "send packet to [%s] failed", tbsys::CNetUtil::addrToString(peer_id).c_str()); rc = OB_ERROR; } } if (rc != OB_SUCCESS) { if (NULL != packet) { packet->free(); packet = NULL; } } return rc; }
int ObClientManager::do_send_request( const ObServer& server, const int32_t pcode, const int32_t version, const int64_t timeout, ObDataBuffer& in_buffer, ObPacket* &response) const { int rc = OB_SUCCESS; ObPacket* packet = new (std::nothrow) ObPacket(); if (NULL == packet) { rc = OB_ALLOCATE_MEMORY_FAILED; } else { packet->set_packet_code(pcode); packet->setChannelId(0); packet->set_api_version(version); packet->set_data(in_buffer); packet->set_source_timeout(timeout); } if (OB_SUCCESS == rc) { rc = packet->serialize(); if (OB_SUCCESS != rc) TBSYS_LOG(WARN, "packet serialize error"); } // serialize failed if (OB_SUCCESS != rc && NULL != packet) { packet->free(); } if (OB_SUCCESS == rc) { rc = do_send_packet(server, packet, timeout, response); } return rc; }
int ObClientManager::get_next(const ObServer& server, const int64_t session_id, const int64_t timeout, ObDataBuffer& in_buffer, ObDataBuffer& out_buffer) const { int rc = OB_SUCCESS; ObPacket* response = NULL; //rc = send_request(server, pcode, version, timeout, in_buffer, response); ObPacket* packet = new (std::nothrow) ObPacket(); if (NULL == packet) { rc = OB_ALLOCATE_MEMORY_FAILED; } else { packet->set_packet_code(OB_SESSION_NEXT_REQUEST); packet->setChannelId(0); packet->set_api_version(0); packet->set_data(in_buffer); packet->set_source_timeout(timeout); packet->set_session_id(session_id); //TODO } if (OB_SUCCESS == rc) { rc = packet->serialize(); if (OB_SUCCESS != rc) TBSYS_LOG(WARN, "packet serialize error, code=%d", packet->get_packet_code()); } // serialize failed if (OB_SUCCESS != rc && NULL != packet) { packet->free(); } if (OB_SUCCESS == rc) { rc = do_send_packet(server, packet, timeout, response); } // deserialize response packet to out_buffer if (OB_SUCCESS == rc && NULL != response) { // copy response's inner_buffer to out_buffer. int64_t data_length = response->get_data_length(); ObDataBuffer* response_buffer = response->get_buffer(); if (out_buffer.get_remain() < data_length) { TBSYS_LOG(ERROR, "insufficient memory in out_buffer, remain:%ld, length=%ld", out_buffer.get_remain(), data_length); rc = OB_ERROR; } else { memcpy(out_buffer.get_data() + out_buffer.get_position(), response_buffer->get_data() + response_buffer->get_position(), data_length); out_buffer.get_position() += data_length; } } return rc; }