Skip to content

Commit bc1cb82

Browse files
committed
avoid overflow
1 parent 708794e commit bc1cb82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/src/main/scala/org/apache/spark/util/SizeEstimator.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,10 +217,10 @@ object SizeEstimator extends Logging {
217217
var arrSize: Long = alignSize(objectSize + INT_SIZE)
218218

219219
if (elementClass.isPrimitive) {
220-
arrSize += alignSize(length * primitiveSize(elementClass))
220+
arrSize += alignSize(length.toLong * primitiveSize(elementClass))
221221
state.size += arrSize
222222
} else {
223-
arrSize += alignSize(length * pointerSize)
223+
arrSize += alignSize(length.toLong * pointerSize)
224224
state.size += arrSize
225225

226226
if (length <= ARRAY_SIZE_FOR_SAMPLING) {

0 commit comments

Comments
 (0)