2018-03-13 10:36:20 +08:00
|
|
|
/***************************************************************************
|
|
|
|
|
*
|
|
|
|
|
* Project _____ __ ____ _ _
|
|
|
|
|
* ( _ ) /__\ (_ _)_| |_ _| |_
|
|
|
|
|
* )(_)( /(__)\ )( (_ _)(_ _)
|
|
|
|
|
* (_____)(__)(__)(__) |_| |_|
|
|
|
|
|
*
|
|
|
|
|
*
|
|
|
|
|
* Copyright 2018-present, Leonid Stryzhevskyi, <lganzzzo@gmail.com>
|
|
|
|
|
*
|
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
|
*
|
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
*
|
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
|
* limitations under the License.
|
|
|
|
|
*
|
|
|
|
|
***************************************************************************/
|
|
|
|
|
|
|
|
|
|
#include "HttpRequestExecutor.hpp"
|
|
|
|
|
|
2018-03-29 12:50:20 +08:00
|
|
|
#include "oatpp/web/protocol/http/outgoing/Request.hpp"
|
|
|
|
|
#include "oatpp/web/protocol/http/outgoing/BufferBody.hpp"
|
2018-03-13 10:36:20 +08:00
|
|
|
|
2018-03-29 12:50:20 +08:00
|
|
|
#include "oatpp/network/Connection.hpp"
|
|
|
|
|
#include "oatpp/core/data/stream/ChunkedBuffer.hpp"
|
|
|
|
|
#include "oatpp/core/data/stream/StreamBufferedProxy.hpp"
|
2018-03-13 10:36:20 +08:00
|
|
|
|
|
|
|
|
#include <stdlib.h>
|
|
|
|
|
#include <sys/types.h>
|
|
|
|
|
#include <sys/socket.h>
|
|
|
|
|
#include <netdb.h>
|
|
|
|
|
#include <netinet/in.h>
|
|
|
|
|
#include <unistd.h>
|
|
|
|
|
|
|
|
|
|
namespace oatpp { namespace web { namespace client {
|
|
|
|
|
|
|
|
|
|
std::shared_ptr<HttpRequestExecutor::Response>
|
2018-06-26 07:30:01 +08:00
|
|
|
HttpRequestExecutor::execute(const String& method,
|
|
|
|
|
const String& path,
|
2018-03-13 10:36:20 +08:00
|
|
|
const std::shared_ptr<Headers>& headers,
|
|
|
|
|
const std::shared_ptr<Body>& body) {
|
|
|
|
|
|
|
|
|
|
//auto stream = oatpp::data::stream::ChunkedBuffer::createShared();
|
|
|
|
|
//stream << "/" << path;
|
|
|
|
|
//ENV::log("HTTP_EXECUTOR", "Execute: '%s'", (const char*) stream->toStringAsString()->getData());
|
|
|
|
|
|
|
|
|
|
auto connection = m_connectionProvider->getConnection();
|
|
|
|
|
|
|
|
|
|
if(!connection){
|
|
|
|
|
return nullptr; // TODO ERROR HERE
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto request = oatpp::web::protocol::http::outgoing::Request::createShared(method, path, headers, body);
|
2018-06-23 20:00:25 +08:00
|
|
|
request->headers->putIfNotExists(oatpp::web::protocol::http::Header::HOST, m_connectionProvider->getHost());
|
2018-03-13 10:36:20 +08:00
|
|
|
|
|
|
|
|
auto ioBuffer = oatpp::data::buffer::IOBuffer::createShared();
|
|
|
|
|
|
|
|
|
|
auto upStream = oatpp::data::stream::OutputStreamBufferedProxy::createShared(connection, ioBuffer);
|
|
|
|
|
request->send(upStream);
|
|
|
|
|
upStream->flush();
|
|
|
|
|
|
|
|
|
|
//auto upStream = oatpp::data::stream::ChunkedBuffer::createShared();
|
|
|
|
|
//request->send(upStream);
|
|
|
|
|
//ENV::log("request", "request:'%s'\n", (const char*) upStream->toStringAsString()->getData());
|
|
|
|
|
//upStream->flushToStream(connection);
|
|
|
|
|
|
|
|
|
|
auto readCount = connection->read(ioBuffer->getData(), ioBuffer->getSize());
|
|
|
|
|
|
|
|
|
|
//((p_char8) ioBuffer->getData())[readCount] = 0;
|
|
|
|
|
//ENV::log("asd", "response='%s'", (const char*) ioBuffer->getData());
|
|
|
|
|
|
|
|
|
|
if(readCount > 0) {
|
|
|
|
|
|
|
|
|
|
oatpp::parser::ParsingCaret caret((p_char8) ioBuffer->getData(), ioBuffer->getSize());
|
|
|
|
|
auto line = protocol::http::Protocol::parseResponseStartingLine(caret);
|
|
|
|
|
if(!line){
|
|
|
|
|
return nullptr; // TODO ERROR HERE
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
oatpp::web::protocol::http::Status error;
|
|
|
|
|
auto headers = protocol::http::Protocol::parseHeaders(caret, error);
|
|
|
|
|
|
|
|
|
|
if(error.code != 0){
|
|
|
|
|
return nullptr; // TODO ERROR HERE
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto bodyStream = oatpp::data::stream::InputStreamBufferedProxy::createShared(connection,
|
2018-04-18 00:21:13 +08:00
|
|
|
ioBuffer,
|
|
|
|
|
caret.getPosition(),
|
|
|
|
|
(v_int32) readCount);
|
2018-03-13 10:36:20 +08:00
|
|
|
|
|
|
|
|
return Response::createShared(line->statusCode, line->description, headers, bodyStream);
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nullptr;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
2018-04-18 00:21:13 +08:00
|
|
|
oatpp::async::Action HttpRequestExecutor::executeAsync(oatpp::async::AbstractCoroutine* parentCoroutine,
|
|
|
|
|
AsyncCallback callback,
|
2018-06-26 07:30:01 +08:00
|
|
|
const String& method,
|
|
|
|
|
const String& path,
|
2018-04-18 00:21:13 +08:00
|
|
|
const std::shared_ptr<Headers>& headers,
|
|
|
|
|
const std::shared_ptr<Body>& body) {
|
|
|
|
|
|
|
|
|
|
class ExecutorCoroutine : public oatpp::async::CoroutineWithResult<ExecutorCoroutine, std::shared_ptr<HttpRequestExecutor::Response>> {
|
|
|
|
|
private:
|
2018-04-18 06:33:19 +08:00
|
|
|
std::shared_ptr<oatpp::network::ClientConnectionProvider> m_connectionProvider;
|
2018-06-26 07:30:01 +08:00
|
|
|
String m_method;
|
|
|
|
|
String m_path;
|
2018-04-18 00:21:13 +08:00
|
|
|
std::shared_ptr<Headers> m_headers;
|
|
|
|
|
std::shared_ptr<Body> m_body;
|
|
|
|
|
private:
|
|
|
|
|
std::shared_ptr<oatpp::data::stream::IOStream> m_connection;
|
2018-04-18 06:33:19 +08:00
|
|
|
std::shared_ptr<oatpp::data::buffer::IOBuffer> m_ioBuffer;
|
|
|
|
|
void* m_bufferPointer;
|
|
|
|
|
os::io::Library::v_size m_bufferBytesLeftToRead;
|
2018-04-18 00:21:13 +08:00
|
|
|
public:
|
|
|
|
|
|
2018-04-18 06:33:19 +08:00
|
|
|
ExecutorCoroutine(const std::shared_ptr<oatpp::network::ClientConnectionProvider>& connectionProvider,
|
2018-06-26 07:30:01 +08:00
|
|
|
const String& method,
|
|
|
|
|
const String& path,
|
2018-04-18 00:21:13 +08:00
|
|
|
const std::shared_ptr<Headers>& headers,
|
|
|
|
|
const std::shared_ptr<Body>& body)
|
2018-04-18 06:33:19 +08:00
|
|
|
: m_connectionProvider(connectionProvider)
|
|
|
|
|
, m_method(method)
|
2018-04-18 00:21:13 +08:00
|
|
|
, m_path(path)
|
|
|
|
|
, m_headers(headers)
|
|
|
|
|
, m_body(body)
|
|
|
|
|
{}
|
|
|
|
|
|
|
|
|
|
Action act() override {
|
2018-04-18 06:33:19 +08:00
|
|
|
oatpp::network::ClientConnectionProvider::AsyncCallback callback =
|
|
|
|
|
static_cast<oatpp::network::ClientConnectionProvider::AsyncCallback>(&ExecutorCoroutine::onConnectionReady);
|
|
|
|
|
return m_connectionProvider->getConnectionAsync(this, callback);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Action onConnectionReady(const std::shared_ptr<oatpp::data::stream::IOStream>& connection) {
|
|
|
|
|
m_connection = connection;
|
|
|
|
|
auto request = oatpp::web::protocol::http::outgoing::Request::createShared(m_method, m_path, m_headers, m_body);
|
|
|
|
|
request->headers->put(oatpp::web::protocol::http::Header::HOST, m_connectionProvider->getHost());
|
|
|
|
|
m_ioBuffer = oatpp::data::buffer::IOBuffer::createShared();
|
|
|
|
|
auto upStream = oatpp::data::stream::OutputStreamBufferedProxy::createShared(connection, m_ioBuffer);
|
|
|
|
|
m_bufferPointer = m_ioBuffer->getData();
|
|
|
|
|
m_bufferBytesLeftToRead = m_ioBuffer->getSize();
|
|
|
|
|
return request->sendAsync(this, upStream->flushAsync(this, yieldTo(&ExecutorCoroutine::readResponse)), upStream);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Action readResponse() {
|
|
|
|
|
return oatpp::data::stream::IOStream::
|
|
|
|
|
readSomeDataAsyncInline(m_connection.get(), m_bufferPointer, m_bufferBytesLeftToRead, yieldTo(&ExecutorCoroutine::parseResponse));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Action parseResponse() {
|
|
|
|
|
|
|
|
|
|
os::io::Library::v_size readCount = m_ioBuffer->getSize() - m_bufferBytesLeftToRead;
|
|
|
|
|
|
|
|
|
|
if(readCount > 0) {
|
|
|
|
|
|
|
|
|
|
oatpp::parser::ParsingCaret caret((p_char8) m_ioBuffer->getData(), m_ioBuffer->getSize());
|
|
|
|
|
auto line = protocol::http::Protocol::parseResponseStartingLine(caret);
|
|
|
|
|
if(!line){
|
|
|
|
|
return error("Invalid starting line");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
oatpp::web::protocol::http::Status err;
|
|
|
|
|
auto headers = protocol::http::Protocol::parseHeaders(caret, err);
|
|
|
|
|
|
|
|
|
|
if(err.code != 0){
|
|
|
|
|
return error("can't parse headers");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto bodyStream = oatpp::data::stream::InputStreamBufferedProxy::createShared(m_connection,
|
|
|
|
|
m_ioBuffer,
|
|
|
|
|
caret.getPosition(),
|
|
|
|
|
(v_int32) readCount);
|
|
|
|
|
|
|
|
|
|
return _return(Response::createShared(line->statusCode, line->description, headers, bodyStream));
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return error("Read zero bytes from Response");
|
|
|
|
|
|
2018-04-18 00:21:13 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
};
|
|
|
|
|
|
2018-04-18 06:33:19 +08:00
|
|
|
return parentCoroutine->startCoroutineForResult<ExecutorCoroutine>(callback, m_connectionProvider, method, path, headers, body);
|
2018-04-18 00:21:13 +08:00
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
2018-03-13 10:36:20 +08:00
|
|
|
}}}
|
