123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732 |
- #ifndef BOOST_MYSQL_CONNECTION_POOL_HPP
- #define BOOST_MYSQL_CONNECTION_POOL_HPP
- #include <boost/mysql/any_connection.hpp>
- #include <boost/mysql/diagnostics.hpp>
- #include <boost/mysql/error_code.hpp>
- #include <boost/mysql/pool_params.hpp>
- #include <boost/mysql/detail/access.hpp>
- #include <boost/mysql/detail/config.hpp>
- #include <boost/mysql/detail/connection_pool_fwd.hpp>
- #include <boost/asio/any_completion_handler.hpp>
- #include <boost/asio/any_io_executor.hpp>
- #include <boost/asio/async_result.hpp>
- #include <chrono>
- #include <memory>
- #include <utility>
- namespace boost {
- namespace mysql {
- class pooled_connection
- {
- #ifndef BOOST_MYSQL_DOXYGEN
- friend struct detail::access;
- friend class detail::basic_pool_impl<detail::io_traits, pooled_connection>;
- #endif
- detail::connection_node* impl_{nullptr};
- std::shared_ptr<detail::pool_impl> pool_impl_;
- pooled_connection(detail::connection_node& node, std::shared_ptr<detail::pool_impl> pool_impl) noexcept
- : impl_(&node), pool_impl_(std::move(pool_impl))
- {
- }
- public:
-
- pooled_connection() noexcept = default;
-
- pooled_connection(pooled_connection&& other) noexcept
- : impl_(other.impl_), pool_impl_(std::move(other.pool_impl_))
- {
- other.impl_ = nullptr;
- }
-
- pooled_connection& operator=(pooled_connection&& other) noexcept
- {
- if (impl_)
- detail::return_connection(std::move(pool_impl_), *impl_, true);
- impl_ = other.impl_;
- other.impl_ = nullptr;
- pool_impl_ = std::move(other.pool_impl_);
- return *this;
- }
- #ifndef BOOST_MYSQL_DOXYGEN
- pooled_connection(const pooled_connection&) = delete;
- pooled_connection& operator=(const pooled_connection&) = delete;
- #endif
-
- ~pooled_connection()
- {
- if (impl_)
- detail::return_connection(std::move(pool_impl_), *impl_, true);
- }
-
- bool valid() const noexcept { return impl_ != nullptr; }
-
- any_connection& get() noexcept { return detail::get_connection(*impl_); }
-
- const any_connection& get() const noexcept { return detail::get_connection(*impl_); }
-
- any_connection* operator->() noexcept { return &get(); }
-
- const any_connection* operator->() const noexcept { return &get(); }
-
- void return_without_reset() noexcept
- {
- BOOST_ASSERT(valid());
- detail::return_connection(std::move(pool_impl_), *impl_, false);
- impl_ = nullptr;
- }
- };
- class connection_pool
- {
- std::shared_ptr<detail::pool_impl> impl_;
- #ifndef BOOST_MYSQL_DOXYGEN
- friend struct detail::access;
- #endif
- static constexpr std::chrono::steady_clock::duration get_default_timeout() noexcept
- {
- return std::chrono::seconds(30);
- }
- struct initiate_run
- {
- template <class Handler>
- void operator()(Handler&& h, std::shared_ptr<detail::pool_impl> self)
- {
- async_run_erased(std::move(self), std::forward<Handler>(h));
- }
- };
- BOOST_MYSQL_DECL
- static void async_run_erased(
- std::shared_ptr<detail::pool_impl> pool,
- asio::any_completion_handler<void(error_code)> handler
- );
- struct initiate_get_connection
- {
- template <class Handler>
- void operator()(
- Handler&& h,
- std::shared_ptr<detail::pool_impl> self,
- std::chrono::steady_clock::duration timeout,
- diagnostics* diag
- )
- {
- async_get_connection_erased(std::move(self), timeout, diag, std::forward<Handler>(h));
- }
- };
- BOOST_MYSQL_DECL
- static void async_get_connection_erased(
- std::shared_ptr<detail::pool_impl> pool,
- std::chrono::steady_clock::duration timeout,
- diagnostics* diag,
- asio::any_completion_handler<void(error_code, pooled_connection)> handler
- );
- template <class CompletionToken>
- auto async_get_connection_impl(
- std::chrono::steady_clock::duration timeout,
- diagnostics* diag,
- CompletionToken&& token
- )
- -> decltype(asio::async_initiate<CompletionToken, void(error_code, pooled_connection)>(
- initiate_get_connection{},
- token,
- impl_,
- timeout,
- diag
- ))
- {
- BOOST_ASSERT(valid());
- return asio::async_initiate<CompletionToken, void(error_code, pooled_connection)>(
- initiate_get_connection{},
- token,
- impl_,
- timeout,
- diag
- );
- }
- BOOST_MYSQL_DECL
- connection_pool(pool_executor_params&& ex_params, pool_params&& params, int);
- public:
-
- connection_pool(pool_executor_params ex_params, pool_params params)
- : connection_pool(std::move(ex_params), std::move(params), 0)
- {
- }
-
- connection_pool(asio::any_io_executor ex, pool_params params)
- : connection_pool(pool_executor_params{ex, ex}, std::move(params), 0)
- {
- }
-
- template <
- class ExecutionContext
- #ifndef BOOST_MYSQL_DOXYGEN
- ,
- class = typename std::enable_if<std::is_convertible<
- decltype(std::declval<ExecutionContext&>().get_executor()),
- asio::any_io_executor>::value>::type
- #endif
- >
- connection_pool(ExecutionContext& ctx, pool_params params)
- : connection_pool({ctx.get_executor(), ctx.get_executor()}, std::move(params), 0)
- {
- }
- #ifndef BOOST_MYSQL_DOXYGEN
- connection_pool(const connection_pool&) = delete;
- connection_pool& operator=(const connection_pool&) = delete;
- #endif
-
- connection_pool(connection_pool&& other) = default;
-
- connection_pool& operator=(connection_pool&& other) = default;
-
- ~connection_pool() = default;
-
- bool valid() const noexcept { return impl_.get() != nullptr; }
-
- using executor_type = asio::any_io_executor;
-
- BOOST_MYSQL_DECL
- executor_type get_executor() noexcept;
-
- template <BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code)) CompletionToken>
- auto async_run(CompletionToken&& token) BOOST_MYSQL_RETURN_TYPE(
- decltype(asio::async_initiate<CompletionToken, void(error_code)>(initiate_run{}, token, impl_))
- )
- {
- BOOST_ASSERT(valid());
- return asio::async_initiate<CompletionToken, void(error_code)>(initiate_run{}, token, impl_);
- }
-
- template <
- BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code, ::boost::mysql::pooled_connection))
- CompletionToken>
- auto async_get_connection(CompletionToken&& token) BOOST_MYSQL_RETURN_TYPE(
- decltype(async_get_connection_impl({}, nullptr, std::forward<CompletionToken>(token)))
- )
- {
- return async_get_connection_impl(
- get_default_timeout(),
- nullptr,
- std::forward<CompletionToken>(token)
- );
- }
-
- template <
- BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code, ::boost::mysql::pooled_connection))
- CompletionToken>
- auto async_get_connection(diagnostics& diag, CompletionToken&& token) BOOST_MYSQL_RETURN_TYPE(
- decltype(async_get_connection_impl({}, nullptr, std::forward<CompletionToken>(token)))
- )
- {
- return async_get_connection_impl(get_default_timeout(), &diag, std::forward<CompletionToken>(token));
- }
-
- template <
- BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code, ::boost::mysql::pooled_connection))
- CompletionToken>
- auto async_get_connection(std::chrono::steady_clock::duration timeout, CompletionToken&& token)
- BOOST_MYSQL_RETURN_TYPE(
- decltype(async_get_connection_impl({}, nullptr, std::forward<CompletionToken>(token)))
- )
- {
- return async_get_connection_impl(timeout, nullptr, std::forward<CompletionToken>(token));
- }
-
- template <
- BOOST_ASIO_COMPLETION_TOKEN_FOR(void(::boost::mysql::error_code, ::boost::mysql::pooled_connection))
- CompletionToken>
- auto async_get_connection(
- std::chrono::steady_clock::duration timeout,
- diagnostics& diag,
- CompletionToken&& token
- )
- BOOST_MYSQL_RETURN_TYPE(
- decltype(async_get_connection_impl({}, nullptr, std::forward<CompletionToken>(token)))
- )
- {
- return async_get_connection_impl(timeout, &diag, std::forward<CompletionToken>(token));
- }
-
- BOOST_MYSQL_DECL
- void cancel();
- };
- }
- }
- #ifdef BOOST_MYSQL_HEADER_ONLY
- #include <boost/mysql/impl/connection_pool.ipp>
- #endif
- #endif
|