Add is_busy()

taocpp · Dec 15, 2024 · 35774de · 35774de
1 parent b726879
commit 35774de
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/include/tao/pq/connection.hpp b/include/tao/pq/connection.hpp
@@ -150,6 +150,8 @@ namespace tao::pq
          return transaction_status() == transaction_status::idle;
       }
 
+      [[nodiscard]] auto is_busy() const noexcept -> bool;
+
       [[nodiscard]] auto direct() -> std::shared_ptr< pq::transaction >;
 
       [[nodiscard]] auto transaction() -> std::shared_ptr< pq::transaction >;

diff --git a/src/lib/pq/connection.cpp b/src/lib/pq/connection.cpp
@@ -277,7 +277,7 @@ namespace tao::pq
    auto connection::get_result( const std::chrono::steady_clock::time_point end ) -> std::unique_ptr< PGresult, decltype( &PQclear ) >
    {
       bool wait_for_write = true;
-      while( PQisBusy( m_pgconn.get() ) != 0 ) {
+      while( is_busy() ) {
          if( wait_for_write ) {
             switch( PQflush( m_pgconn.get() ) ) {
                case 0:
@@ -289,7 +289,8 @@ namespace tao::pq
                   break;
 
                default:
-                  throw std::runtime_error( std::format( "PQflush() failed: {}", error_message() ) );  // LCOV_EXCL_STOP
+                  throw std::runtime_error( std::format( "PQflush() failed: {}", error_message() ) );
+                  // LCOV_EXCL_STOP
             }
          }
          connection::wait( wait_for_write, end );
@@ -502,6 +503,11 @@ namespace tao::pq
       }
    }
 
+   auto connection::is_busy() const noexcept -> bool
+   {
+      return PQisBusy( m_pgconn.get() ) != 0;
+   }
+
    auto connection::direct() -> std::shared_ptr< pq::transaction >
    {
       return std::make_shared< internal::autocommit_transaction >( shared_from_this() );