109 lines
3.3 KiB
C++
109 lines
3.3 KiB
C++
|
//
|
||
|
|
// Copyright (c) 2019-2024 Ruben Perez Hidalgo (rubenperez038 at gmail dot com)
|
||
|
|
//
|
||
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
//
|
||
|
|
|
||
|
|
#ifndef BOOST_MYSQL_IMPL_INTERNAL_SANSIO_START_EXECUTION_HPP
|
||
|
|
#define BOOST_MYSQL_IMPL_INTERNAL_SANSIO_START_EXECUTION_HPP
|
||
|
|
|
||
|
|
#include <boost/mysql/diagnostics.hpp>
|
||
|
|
#include <boost/mysql/error_code.hpp>
|
||
|
|
|
||
|
|
#include <boost/mysql/detail/algo_params.hpp>
|
||
|
|
#include <boost/mysql/detail/any_execution_request.hpp>
|
||
|
|
#include <boost/mysql/detail/execution_processor/execution_processor.hpp>
|
||
|
|
#include <boost/mysql/detail/next_action.hpp>
|
||
|
|
|
||
|
|
#include <boost/mysql/impl/internal/coroutine.hpp>
|
||
|
|
#include <boost/mysql/impl/internal/protocol/serialization.hpp>
|
||
|
|
#include <boost/mysql/impl/internal/sansio/connection_state_data.hpp>
|
||
|
|
#include <boost/mysql/impl/internal/sansio/read_resultset_head.hpp>
|
||
|
|
|
||
|
|
namespace boost {
|
||
|
|
namespace mysql {
|
||
|
|
namespace detail {
|
||
|
|
|
||
|
|
inline error_code check_client_errors(const any_execution_request& req)
|
||
|
|
{
|
||
|
|
if (req.is_query)
|
||
|
|
return error_code();
|
||
|
|
return req.data.stmt.stmt.num_params() == req.data.stmt.params.size() ? error_code()
|
||
|
|
: client_errc::wrong_num_params;
|
||
|
|
}
|
||
|
|
|
||
|
|
inline resultset_encoding get_encoding(const any_execution_request& req)
|
||
|
|
{
|
||
|
|
return req.is_query ? resultset_encoding::text : resultset_encoding::binary;
|
||
|
|
}
|
||
|
|
|
||
|
|
class start_execution_algo
|
||
|
|
{
|
||
|
|
int resume_point_{0};
|
||
|
|
read_resultset_head_algo read_head_st_;
|
||
|
|
any_execution_request req_;
|
||
|
|
|
||
|
|
std::uint8_t& seqnum() { return processor().sequence_number(); }
|
||
|
|
execution_processor& processor() { return read_head_st_.processor(); }
|
||
|
|
diagnostics& diag() { return read_head_st_.diag(); }
|
||
|
|
|
||
|
|
next_action compose_request(connection_state_data& st)
|
||
|
|
{
|
||
|
|
if (req_.is_query)
|
||
|
|
{
|
||
|
|
return st.write(query_command{req_.data.query}, seqnum());
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
return st.write(execute_stmt_command{req_.data.stmt.stmt.id(), req_.data.stmt.params}, seqnum());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public:
|
||
|
|
start_execution_algo(start_execution_algo_params params) noexcept
|
||
|
|
: read_head_st_(read_resultset_head_algo_params{params.diag, params.proc}), req_(params.req)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
next_action resume(connection_state_data& st, error_code ec)
|
||
|
|
{
|
||
|
|
next_action act;
|
||
|
|
|
||
|
|
switch (resume_point_)
|
||
|
|
{
|
||
|
|
case 0:
|
||
|
|
|
||
|
|
// Clear diagnostics
|
||
|
|
diag().clear();
|
||
|
|
|
||
|
|
// Check for errors
|
||
|
|
ec = check_client_errors(req_);
|
||
|
|
if (ec)
|
||
|
|
return ec;
|
||
|
|
|
||
|
|
// Reset the processor
|
||
|
|
processor().reset(get_encoding(req_), st.meta_mode);
|
||
|
|
|
||
|
|
// Send the execution request
|
||
|
|
BOOST_MYSQL_YIELD(resume_point_, 1, compose_request(st))
|
||
|
|
|
||
|
|
if (ec)
|
||
|
|
return ec;
|
||
|
|
|
||
|
|
// Read the first resultset's head and return its result
|
||
|
|
while (!(act = read_head_st_.resume(st, ec)).is_done())
|
||
|
|
BOOST_MYSQL_YIELD(resume_point_, 2, act)
|
||
|
|
return act;
|
||
|
|
}
|
||
|
|
|
||
|
|
return next_action();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace detail
|
||
|
|
} // namespace mysql
|
||
|
|
} // namespace boost
|
||
|
|
|
||
|
|
#endif /* INCLUDE_MYSQL_IMPL_NETWORK_ALGORITHMS_READ_RESULTSET_HEAD_HPP_ */
|