diff --git a/include/tao/pq/pipeline.hpp b/include/tao/pq/pipeline.hpp index 98f1bc9..f4eae9e 100644 --- a/include/tao/pq/pipeline.hpp +++ b/include/tao/pq/pipeline.hpp @@ -5,6 +5,7 @@ #ifndef TAO_PQ_PIPELINE_HPP #define TAO_PQ_PIPELINE_HPP +#include #include #include @@ -40,7 +41,7 @@ namespace tao::pq void operator=( pipeline&& ) = delete; void sync(); - void consume_sync(); + void consume_sync( const std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now() ); void finish(); }; diff --git a/include/tao/pq/transaction_base.hpp b/include/tao/pq/transaction_base.hpp index 98fa707..353e7aa 100644 --- a/include/tao/pq/transaction_base.hpp +++ b/include/tao/pq/transaction_base.hpp @@ -136,8 +136,6 @@ namespace tao::pq } [[nodiscard]] auto get_result( const std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now() ) -> result; - - // TODO: move this to the pipeline_transaction class void consume_pipeline_sync( const std::chrono::steady_clock::time_point start = std::chrono::steady_clock::now() ); }; diff --git a/src/lib/pq/pipeline.cpp b/src/lib/pq/pipeline.cpp index fc6c9cc..896936c 100644 --- a/src/lib/pq/pipeline.cpp +++ b/src/lib/pq/pipeline.cpp @@ -4,6 +4,8 @@ #include +#include + #include namespace tao::pq @@ -21,9 +23,9 @@ namespace tao::pq connection()->pipeline_sync(); } - void pipeline::consume_sync() + void pipeline::consume_sync( const std::chrono::steady_clock::time_point start ) { - current_transaction()->consume_pipeline_sync(); + current_transaction()->consume_pipeline_sync( start ); } void pipeline::finish()