123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201 |
- #ifndef BOOST_MYSQL_IMPL_INTERNAL_CONNECTION_POOL_SANSIO_CONNECTION_NODE_HPP
- #define BOOST_MYSQL_IMPL_INTERNAL_CONNECTION_POOL_SANSIO_CONNECTION_NODE_HPP
- #include <boost/mysql/diagnostics.hpp>
- #include <boost/mysql/error_code.hpp>
- #include <boost/assert.hpp>
- namespace boost {
- namespace mysql {
- namespace detail {
- enum class connection_status
- {
-
-
- initial,
-
- connect_in_progress,
-
- sleep_connect_failed_in_progress,
-
- reset_in_progress,
-
- ping_in_progress,
-
- idle,
-
- in_use,
-
- terminated,
- };
- enum class next_connection_action
- {
-
- none,
-
- connect,
-
- sleep_connect_failed,
-
- idle_wait,
-
- reset,
-
- ping,
- };
- enum class collection_state
- {
-
- none,
-
- needs_collect,
-
- needs_collect_with_reset
- };
- template <class Derived>
- class sansio_connection_node
- {
- connection_status status_;
- inline bool is_pending(connection_status status) noexcept
- {
- return status != connection_status::initial && status != connection_status::idle &&
- status != connection_status::in_use && status != connection_status::terminated;
- }
- inline static next_connection_action status_to_action(connection_status status) noexcept
- {
- switch (status)
- {
- case connection_status::connect_in_progress: return next_connection_action::connect;
- case connection_status::sleep_connect_failed_in_progress:
- return next_connection_action::sleep_connect_failed;
- case connection_status::ping_in_progress: return next_connection_action::ping;
- case connection_status::reset_in_progress: return next_connection_action::reset;
- case connection_status::idle:
- case connection_status::in_use: return next_connection_action::idle_wait;
- default: return next_connection_action::none;
- }
- }
- next_connection_action set_status(connection_status new_status)
- {
- auto& derived = static_cast<Derived&>(*this);
-
- if (new_status == connection_status::idle && status_ != connection_status::idle)
- derived.entering_idle();
- else if (new_status != connection_status::idle && status_ == connection_status::idle)
- derived.exiting_idle();
-
- if (!is_pending(status_) && is_pending(new_status))
- derived.entering_pending();
- else if (is_pending(status_) && !is_pending(new_status))
- derived.exiting_pending();
-
- status_ = new_status;
- return status_to_action(new_status);
- }
- public:
- sansio_connection_node(connection_status initial_status = connection_status::initial) noexcept
- : status_(initial_status)
- {
- }
- void mark_as_in_use() noexcept
- {
- BOOST_ASSERT(status_ == connection_status::idle);
- set_status(connection_status::in_use);
- }
- void cancel() { set_status(connection_status::terminated); }
- next_connection_action resume(error_code ec, collection_state col_st)
- {
- switch (status_)
- {
- case connection_status::initial: return set_status(connection_status::connect_in_progress);
- case connection_status::connect_in_progress:
- return ec ? set_status(connection_status::sleep_connect_failed_in_progress)
- : set_status(connection_status::idle);
- case connection_status::sleep_connect_failed_in_progress:
- return set_status(connection_status::connect_in_progress);
- case connection_status::idle:
-
-
- return set_status(connection_status::ping_in_progress);
- case connection_status::in_use:
-
-
-
- if (col_st == collection_state::needs_collect)
- {
-
- return set_status(connection_status::idle);
- }
- else if (col_st == collection_state::needs_collect_with_reset)
- {
- return set_status(connection_status::reset_in_progress);
- }
- else
- {
-
-
- return next_connection_action::idle_wait;
- }
- case connection_status::ping_in_progress:
- case connection_status::reset_in_progress:
-
- return ec ? set_status(connection_status::connect_in_progress)
- : set_status(connection_status::idle);
- case connection_status::terminated:
- default: return next_connection_action::none;
- }
- }
-
- connection_status status() const noexcept { return status_; }
- };
- }
- }
- }
- #endif
|