comfortableCapacity size optimize

This commit is contained in:
libin
2023-11-09 15:39:06 +08:00
parent f1b0835fb5
commit f716bfd2f7
2 changed files with 21 additions and 5 deletions
@@ -16,8 +16,8 @@ package com.zfoo.protocol.collection;
import com.zfoo.protocol.model.Pair;
import com.zfoo.protocol.util.AssertionUtils;
import com.zfoo.protocol.util.IOUtils;
import com.zfoo.protocol.util.MathSafeUtil;
import com.zfoo.protocol.util.StringUtils;
import io.netty.util.internal.MathUtil;
import java.util.*;
@@ -116,10 +116,7 @@ public abstract class CollectionUtils {
* CN: 计算HashMap初始化合适的大小,为了安全必须给初始化的集合一个最大上限,防止反序列化一个不合法的包导致内存突然升高
*/
public static int comfortableCapacity(int capacity) {
return MathUtil.safeFindNextPositivePowerOfTwo(capacity);
/* return capacity < 16
? (capacity < 8 ? 16 : 32)
: (capacity < 32 ? 64 : Math.min(capacity << 1, IOUtils.BYTES_PER_MB));*/
return MathSafeUtil.safeFindNextPositivePowerOfTwo(capacity);
}
public static int capacity(int expectedSize) {
@@ -0,0 +1,19 @@
package com.zfoo.protocol.util;
/**
* @author Boone Jack
*/
public class MathSafeUtil {
private MathSafeUtil() {
}
public static int findNextPositivePowerOfTwo(int value) {
assert value > Integer.MIN_VALUE && value < IOUtils.BYTES_PER_MB;
return 1 << 32 - Integer.numberOfLeadingZeros(value - 1);
}
public static int safeFindNextPositivePowerOfTwo(int value) {
return value <= 0 ? 1 : (value >= IOUtils.BYTES_PER_MB ? IOUtils.BYTES_PER_MB : findNextPositivePowerOfTwo(value));
}
}