|
|
|
@ -244,15 +244,8 @@ void DatasetOp::Print(std::ostream &out, bool show_all) const {
|
|
|
|
|
|
|
|
|
|
// Gets the next buffer from the given child
|
|
|
|
|
Status DatasetOp::GetNextBuffer(std::unique_ptr<DataBuffer> *p_buffer, int32_t worker_id, bool retry_if_eoe) {
|
|
|
|
|
#if defined(_WIN32) || defined(_WIN64)
|
|
|
|
|
RETURN_IF_NOT_OK(out_connector_->PopWithRetry(static_cast<int>(worker_id), p_buffer, retry_if_eoe));
|
|
|
|
|
#else
|
|
|
|
|
std::unique_ptr<DataBuffer> next_buff;
|
|
|
|
|
// pop is a blocked call and will throw an interruption if the whole group shuts down.
|
|
|
|
|
RETURN_IF_NOT_OK(out_connector_->PopWithRetry(static_cast<int>(worker_id), &next_buff, retry_if_eoe));
|
|
|
|
|
|
|
|
|
|
*p_buffer = std::move(next_buff);
|
|
|
|
|
#endif
|
|
|
|
|
RETURN_IF_NOT_OK(out_connector_->PopWithRetry(static_cast<int>(worker_id), p_buffer, retry_if_eoe));
|
|
|
|
|
return Status::OK();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|