Method Wait

be/src/util/thread-pool.h:273–291

Wait for the operation to complete or time out with the specified limit 'timeout_millis' given that the caller has already waited 'time_used_millis'. If the operation times out, it returns TErrorCode::THREAD_POOL_TASK_TIMED_OUT. Otherwise, it returns the status returned by Execute().

  /// If the operation times out, it returns TErrorCode::THREAD_POOL_TASK_TIMED_OUT.
  /// Otherwise, it returns the status returned by Execute().
  Status Wait(int64_t timeout_millis, int64_t time_used_millis) {
    // If the time used has already exceeded the timeout, go directly to returning
    // THREAD_POOL_TASK_TIMED_OUT.
    bool timed_out = (time_used_millis >= timeout_millis);
    Status status;
    if (!timed_out) {
      int64_t timeout_left_millis = timeout_millis - time_used_millis;
      status = done_promise_.Get(timeout_left_millis, &timed_out);
    }
    if (timed_out) {
      // IMPALA-7946: Always throw an error using the original timeout, not the
      // timeout remaining.
      Status timeout_status = Status(TErrorCode::THREAD_POOL_TASK_TIMED_OUT,
          GetDescription(), timeout_millis / MILLIS_PER_SEC);
      LOG(WARNING) << timeout_status.GetDetail();
      return timeout_status;
    }
    return status;
  }

  // Set to the return status of ExecuteImpl() upon completion
  Promise<Status> done_promise_;