mirror of
https://github.com/tiennm99/zfoo.git
synced 2026-05-26 16:01:55 +00:00
feat[hashmap]: add int-short primitive type hash map
This commit is contained in:
godotg
@@ -41,28 +41,28 @@ public abstract class ArrayUtils {
|
||||
return value != null && value;
|
||||
}
|
||||
|
||||
public static byte byteValue(Byte value) {
|
||||
return value == null ? 0 : value;
|
||||
public static byte byteValue(Number value) {
|
||||
return value == null ? 0 : value.byteValue();
|
||||
}
|
||||
|
||||
public static short shortValue(Short value) {
|
||||
return value == null ? 0 : value;
|
||||
public static short shortValue(Number value) {
|
||||
return value == null ? 0 : value.shortValue();
|
||||
}
|
||||
|
||||
public static int intValue(Integer value) {
|
||||
return value == null ? 0 : value;
|
||||
public static int intValue(Number value) {
|
||||
return value == null ? 0 : value.intValue();
|
||||
}
|
||||
|
||||
public static long longValue(Long value) {
|
||||
return value == null ? 0 : value;
|
||||
public static long longValue(Number value) {
|
||||
return value == null ? 0 : value.longValue();
|
||||
}
|
||||
|
||||
public static float floatValue(Float value) {
|
||||
return value == null ? 0 : value;
|
||||
public static float floatValue(Number value) {
|
||||
return value == null ? 0 : value.floatValue();
|
||||
}
|
||||
|
||||
public static double doubleValue(Double value) {
|
||||
return value == null ? 0 : value;
|
||||
public static double doubleValue(Number value) {
|
||||
return value == null ? 0 : value.doubleValue();
|
||||
}
|
||||
|
||||
public static char charValue(Character value) {
|
||||
|
||||
@@ -289,8 +289,8 @@ public class HashMapIntInt implements Map<Integer, Integer> {
|
||||
|
||||
@Override
|
||||
public Integer setValue(Integer value) {
|
||||
var prevValue = ArrayUtils.intValue(values[entryIndex]);
|
||||
values[entryIndex] = value;
|
||||
var prevValue = values[entryIndex];
|
||||
values[entryIndex] = ArrayUtils.intValue(value);
|
||||
return prevValue;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -86,10 +86,10 @@ public class HashMapIntLong implements Map<Integer, Long> {
|
||||
|
||||
@Override
|
||||
public boolean containsValue(Object value) {
|
||||
return containsValuePrimitive(ArrayUtils.intValue((Integer) value));
|
||||
return containsValuePrimitive(ArrayUtils.longValue((Long) value));
|
||||
}
|
||||
|
||||
public boolean containsValuePrimitive(int value) {
|
||||
public boolean containsValuePrimitive(long value) {
|
||||
for (var i = 0; i < statuses.length; i++) {
|
||||
if (statuses[i] == FILLED && values[i] == value) {
|
||||
return true;
|
||||
@@ -271,8 +271,8 @@ public class HashMapIntLong implements Map<Integer, Long> {
|
||||
|
||||
@Override
|
||||
public Long setValue(Long value) {
|
||||
var prevValue = ArrayUtils.longValue(values[entryIndex]);
|
||||
values[entryIndex] = value;
|
||||
var prevValue = values[entryIndex];
|
||||
values[entryIndex] = ArrayUtils.longValue(value);
|
||||
return prevValue;
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,411 @@
|
||||
/*
|
||||
* Copyright (C) 2020 The zfoo Authors
|
||||
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
|
||||
* in compliance with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software distributed under the License is distributed
|
||||
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and limitations under the License.
|
||||
*/
|
||||
|
||||
package com.zfoo.protocol.collection;
|
||||
|
||||
import com.zfoo.protocol.util.StringUtils;
|
||||
import io.netty.util.collection.IntObjectHashMap;
|
||||
import io.netty.util.internal.MathUtil;
|
||||
|
||||
import java.util.*;
|
||||
|
||||
import static com.zfoo.protocol.collection.HashMapIntInt.*;
|
||||
|
||||
|
||||
/**
|
||||
* @author godotg
|
||||
* @version 3.0
|
||||
*/
|
||||
public class HashMapIntShort implements Map<Integer, Short> {
|
||||
|
||||
private int[] keys;
|
||||
private short[] values;
|
||||
private byte[] statuses;
|
||||
private int size;
|
||||
private int maxSize;
|
||||
private int mask;
|
||||
|
||||
|
||||
public HashMapIntShort() {
|
||||
this(IntObjectHashMap.DEFAULT_CAPACITY);
|
||||
}
|
||||
|
||||
public HashMapIntShort(int initialCapacity) {
|
||||
var capacity = MathUtil.safeFindNextPositivePowerOfTwo(initialCapacity);
|
||||
initCapacity(capacity);
|
||||
}
|
||||
|
||||
private void initCapacity(int capacity) {
|
||||
mask = capacity - 1;
|
||||
|
||||
keys = new int[capacity];
|
||||
values = new short[capacity];
|
||||
statuses = new byte[capacity];
|
||||
|
||||
maxSize = calcMaxSize(capacity);
|
||||
}
|
||||
|
||||
private void ensureCapacity() {
|
||||
if (size > maxSize) {
|
||||
if (keys.length == Integer.MAX_VALUE) {
|
||||
throw new IllegalStateException("Max capacity reached at size=" + size);
|
||||
}
|
||||
// Double the capacity.
|
||||
rehash(keys.length << 1);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public int size() {
|
||||
return size;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isEmpty() {
|
||||
return size == 0;
|
||||
}
|
||||
|
||||
public boolean containsKeyPrimitive(int key) {
|
||||
return indexOf(key) >= 0;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean containsKey(Object key) {
|
||||
return containsKeyPrimitive(ArrayUtils.intValue((Integer) key));
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean containsValue(Object value) {
|
||||
return containsValuePrimitive(ArrayUtils.shortValue((Short) value));
|
||||
}
|
||||
|
||||
public boolean containsValuePrimitive(short value) {
|
||||
for (var i = 0; i < statuses.length; i++) {
|
||||
if (statuses[i] == FILLED && values[i] == value) {
|
||||
return true;
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short get(Object key) {
|
||||
var index = indexOf(ArrayUtils.intValue((Integer) key));
|
||||
return index == -1 ? null : values[index];
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short put(Integer key, Short value) {
|
||||
return putPrimitive(ArrayUtils.intValue(key), ArrayUtils.shortValue(value));
|
||||
}
|
||||
|
||||
public Short putPrimitive(int key, short value) {
|
||||
var startIndex = hashIndex(key);
|
||||
var index = startIndex;
|
||||
|
||||
var firstRemoveIndex = -1;
|
||||
for (; ; ) {
|
||||
var status = statuses[index];
|
||||
if (status == FREE) {
|
||||
index = firstRemoveIndex < 0 ? index : firstRemoveIndex;
|
||||
set(index, key, value, FILLED);
|
||||
size++;
|
||||
ensureCapacity();
|
||||
return null;
|
||||
} else if (status == REMOVED) {
|
||||
firstRemoveIndex = firstRemoveIndex < 0 ? index : firstRemoveIndex;
|
||||
} else if (keys[index] == key) { // status == FILLED
|
||||
// Found existing entry with this key, just replace the value.
|
||||
var previousValue = values[index];
|
||||
values[index] = value;
|
||||
return previousValue;
|
||||
}
|
||||
|
||||
// Conflict, keep probing ...
|
||||
if ((index = probeNext(index, mask)) == startIndex) {
|
||||
if (firstRemoveIndex < 0) {
|
||||
throw new IllegalStateException("Unable to insert, the map was full at MAX_ARRAY_SIZE and couldn't grow");
|
||||
} else {
|
||||
set(firstRemoveIndex, key, value, FILLED);
|
||||
size++;
|
||||
ensureCapacity();
|
||||
return null;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short remove(Object key) {
|
||||
return removePrimitive(ArrayUtils.intValue((Integer) key));
|
||||
}
|
||||
|
||||
public Short removePrimitive(int key) {
|
||||
var index = indexOf(key);
|
||||
if (index == -1) {
|
||||
return null;
|
||||
}
|
||||
var prev = values[index];
|
||||
removeAt(index);
|
||||
return prev;
|
||||
}
|
||||
|
||||
private void removeAt(int index) {
|
||||
set(index, 0, (short) 0, REMOVED);
|
||||
size--;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void putAll(Map<? extends Integer, ? extends Short> m) {
|
||||
for (Entry<? extends Integer, ? extends Short> entry : m.entrySet()) {
|
||||
put(entry.getKey(), entry.getValue());
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void clear() {
|
||||
Arrays.fill(keys, 0);
|
||||
Arrays.fill(values, (short) 0);
|
||||
Arrays.fill(statuses, FREE);
|
||||
size = 0;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Set<Integer> keySet() {
|
||||
return new KeySet();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Collection<Short> values() {
|
||||
return new ValueSet();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Set<Entry<Integer, Short>> entrySet() {
|
||||
return new EntrySet();
|
||||
}
|
||||
|
||||
private int hashIndex(int key) {
|
||||
return key & mask;
|
||||
}
|
||||
|
||||
private void set(int index, int key, short value, byte status) {
|
||||
keys[index] = key;
|
||||
values[index] = value;
|
||||
statuses[index] = status;
|
||||
}
|
||||
|
||||
private void rehash(int newCapacity) {
|
||||
var oldKeys = keys;
|
||||
var oldValues = values;
|
||||
var oldStatuses = statuses;
|
||||
|
||||
initCapacity(newCapacity);
|
||||
|
||||
for (var i = 0; i < oldStatuses.length; ++i) {
|
||||
var oldStatus = oldStatuses[i];
|
||||
if (oldStatus == FILLED) {
|
||||
var oldKey = oldKeys[i];
|
||||
var oldValue = oldValues[i];
|
||||
int index = hashIndex(oldKey);
|
||||
|
||||
for (; ; ) {
|
||||
if (statuses[index] == FREE) {
|
||||
set(index, oldKey, oldValue, FILLED);
|
||||
break;
|
||||
}
|
||||
|
||||
index = probeNext(index, mask);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private int indexOf(int key) {
|
||||
int startIndex = hashIndex(key);
|
||||
int index = startIndex;
|
||||
|
||||
for (; ; ) {
|
||||
var status = statuses[index];
|
||||
if (status == FREE) {
|
||||
// It's available, so no chance that this value exists anywhere in the map.
|
||||
return -1;
|
||||
}
|
||||
if (key == keys[index] && status == FILLED) {
|
||||
return index;
|
||||
}
|
||||
|
||||
// Conflict, keep probing ...
|
||||
if ((index = probeNext(index, mask)) == startIndex) {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private class PrimitiveEntry implements Entry<Integer, Short> {
|
||||
int entryIndex;
|
||||
|
||||
PrimitiveEntry(int entryIndex) {
|
||||
this.entryIndex = entryIndex;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Integer getKey() {
|
||||
return keys[entryIndex];
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short getValue() {
|
||||
return values[entryIndex];
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short setValue(Short value) {
|
||||
var prevValue = values[entryIndex];
|
||||
values[entryIndex] = ArrayUtils.shortValue(value);
|
||||
return prevValue;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private class FastIterator implements Iterator<Entry<Integer, Short>> {
|
||||
int lastCursor = -1;
|
||||
int cursor = -1;
|
||||
|
||||
private void scanNext() {
|
||||
while (++cursor != statuses.length && statuses[cursor] != FILLED) {
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean hasNext() {
|
||||
if (cursor == -1) {
|
||||
scanNext();
|
||||
}
|
||||
return cursor != statuses.length;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Entry<Integer, Short> next() {
|
||||
if (!hasNext()) {
|
||||
throw new NoSuchElementException();
|
||||
}
|
||||
|
||||
lastCursor = cursor;
|
||||
scanNext();
|
||||
|
||||
return new PrimitiveEntry(lastCursor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void remove() {
|
||||
if (lastCursor == -1) {
|
||||
throw new IllegalStateException("next must be called before each remove.");
|
||||
}
|
||||
removeAt(lastCursor);
|
||||
cursor = -1;
|
||||
lastCursor = -1;
|
||||
}
|
||||
}
|
||||
|
||||
private final class KeySet extends AbstractSet<Integer> {
|
||||
FastIterator fastIterator = new FastIterator();
|
||||
|
||||
@Override
|
||||
public Iterator<Integer> iterator() {
|
||||
return new Iterator<Integer>() {
|
||||
@Override
|
||||
public boolean hasNext() {
|
||||
return fastIterator.hasNext();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Integer next() {
|
||||
return fastIterator.next().getKey();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void remove() {
|
||||
fastIterator.remove();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
@Override
|
||||
public int size() {
|
||||
return HashMapIntShort.this.size();
|
||||
}
|
||||
}
|
||||
|
||||
private final class ValueSet extends AbstractSet<Short> {
|
||||
FastIterator fastIterator = new FastIterator();
|
||||
|
||||
@Override
|
||||
public Iterator<Short> iterator() {
|
||||
return new Iterator<Short>() {
|
||||
@Override
|
||||
public boolean hasNext() {
|
||||
return fastIterator.hasNext();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Short next() {
|
||||
return fastIterator.next().getValue();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void remove() {
|
||||
fastIterator.remove();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
@Override
|
||||
public int size() {
|
||||
return HashMapIntShort.this.size();
|
||||
}
|
||||
}
|
||||
|
||||
private final class EntrySet extends AbstractSet<Entry<Integer, Short>> {
|
||||
@Override
|
||||
public Iterator<Entry<Integer, Short>> iterator() {
|
||||
return new FastIterator();
|
||||
}
|
||||
|
||||
@Override
|
||||
public int size() {
|
||||
return HashMapIntShort.this.size();
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
if (isEmpty()) {
|
||||
return StringUtils.EMPTY_JSON;
|
||||
}
|
||||
var builder = new StringBuilder(4 * size);
|
||||
builder.append('{');
|
||||
var first = true;
|
||||
for (int i = 0; i < values.length; ++i) {
|
||||
if (statuses[i] != FILLED) {
|
||||
continue;
|
||||
}
|
||||
if (!first) {
|
||||
builder.append(", ");
|
||||
}
|
||||
builder.append(keys[i]).append('=').append(values[i]);
|
||||
first = false;
|
||||
}
|
||||
return builder.append('}').toString();
|
||||
}
|
||||
}
|
||||
@@ -271,8 +271,8 @@ public class HashMapLongInt implements Map<Long, Integer> {
|
||||
|
||||
@Override
|
||||
public Integer setValue(Integer value) {
|
||||
var prevValue = ArrayUtils.intValue(values[entryIndex]);
|
||||
values[entryIndex] = value;
|
||||
var prevValue = values[entryIndex];
|
||||
values[entryIndex] = ArrayUtils.intValue(value);
|
||||
return prevValue;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -86,10 +86,10 @@ public class HashMapLongLong implements Map<Long, Long> {
|
||||
|
||||
@Override
|
||||
public boolean containsValue(Object value) {
|
||||
return containsValuePrimitive(ArrayUtils.intValue((Integer) value));
|
||||
return containsValuePrimitive(ArrayUtils.longValue((Long) value));
|
||||
}
|
||||
|
||||
public boolean containsValuePrimitive(int value) {
|
||||
public boolean containsValuePrimitive(long value) {
|
||||
for (var i = 0; i < statuses.length; i++) {
|
||||
if (statuses[i] == FILLED && values[i] == value) {
|
||||
return true;
|
||||
@@ -271,8 +271,8 @@ public class HashMapLongLong implements Map<Long, Long> {
|
||||
|
||||
@Override
|
||||
public Long setValue(Long value) {
|
||||
var prevValue = ArrayUtils.longValue(values[entryIndex]);
|
||||
values[entryIndex] = value;
|
||||
var prevValue = values[entryIndex];
|
||||
values[entryIndex] = ArrayUtils.longValue(value);
|
||||
return prevValue;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user