lidavidm commented on code in PR #14524:
URL: https://github.com/apache/arrow/pull/14524#discussion_r1018417571
##########
cpp/src/arrow/util/async_util.h:
##########
@@ -145,206 +188,216 @@ class ARROW_EXPORT AsyncTaskScheduler {
/// acquired and the caller can proceed. If a future is returned then the
caller
/// should wait for the future to complete first. When the returned
future completes
/// the permits have NOT been acquired and the caller must call Acquire
again
+ ///
+ /// \param amt the number of permits to acquire
virtual std::optional<Future<>> TryAcquire(int amt) = 0;
/// Release amt permits
///
/// This will possibly complete waiting futures and should probably not be
/// called while holding locks.
+ ///
+ /// \param amt the number of permits to release
virtual void Release(int amt) = 0;
/// The size of the largest task that can run
///
/// Incoming tasks will have their cost latched to this value to ensure
- /// they can still run (although they will generally be the only thing
allowed to
+ /// they can still run (although they will be the only thing allowed to
/// run at that time).
virtual int Capacity() = 0;
+
+ /// Pause the throttle
+ ///
+ /// Any tasks that have been submitted already will continue. However, no
new tasks
+ /// will be run until the throttle is resumed.
+ virtual void Pause() = 0;
+ /// Resume the throttle
+ ///
+ /// Allows taks to be submitted again. If there is a max_concurrent_cost
limit then
+ /// it will still apply.
+ virtual void Resume() = 0;
};
- /// Create a throttle
+
+ /// Pause the throttle
+ ///
+ /// Any tasks that have been submitted already will continue. However, no
new tasks
+ /// will be run until the throttle is resumed.
+ virtual void Pause() = 0;
+ /// Resume the throttle
///
- /// This throttle is used to limit how many tasks can run at once. The
- /// user should keep the throttle alive for the lifetime of the scheduler.
- /// The same throttle can be used in multiple schedulers.
- static std::unique_ptr<Throttle> MakeThrottle(int max_concurrent_cost);
+ /// Allows taks to be submitted again. If there is a max_concurrent_cost
limit then
+ /// it will still apply.
+ virtual void Resume() = 0;
- /// Add a task to the scheduler
+ /// Create a throttled view of a scheduler
///
- /// If the scheduler is in an aborted state this call will return false and
the task
- /// will never be run. This is harmless and does not need to be guarded
against.
+ /// Tasks added via this view will be subjected to the throttle and, if the
tasks cannot
+ /// run immediately, will be placed into a queue.
///
- /// If the scheduler is in an ended state then this call will cause an
program abort.
- /// This represents a logic error in the program and should be avoidable.
+ /// Using a throttled view after the underlying scheduler has finished is
invalid.
Review Comment:
Sounds good to me, thanks!
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
