-
Notifications
You must be signed in to change notification settings - Fork 75
Conversation
998ab09
to
8e06431
Compare
Signed-off-by: Yuan Zhou <yuan.zhou@intel.com>
val delta = iter.next() | ||
rowCount += delta.numRows | ||
if (rowCount > limit) { | ||
val newSize = rowCount - limit |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The newSize calculation logic can be changed like below codes:
if (rowCount < limit) {
val delta = iter.next()
val preRowCount = rowCount
rowCount += delta.numRows
if (rowCount > limit) {
delta.setNumRows(limit - preRowCount)
}
delta
} else {
throw new NoSuchElementException("End of ColumnarBatch iterator")
}
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
cool, thanks for the check
if (rowCount > limit) { | ||
val newSize = limit - preRowCount | ||
delta.setNumRows(newSize) | ||
} |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
hi, delta should be returned
dabe74e
to
8d9d1cb
Compare
hi yuan, here should also be changed into ColumnarLocalLimitExec according to my test. |
right, I just added this change |
Signed-off-by: Yuan Zhou <yuan.zhou@intel.com>
What changes were proposed in this pull request?
This patch implements columnar Limit
Signed-off-by: Yuan Zhou yuan.zhou@intel.com
How was this patch tested?
local unit tests