OpenHashMap.mergeX: Avoid double-hashing key

Towards vigna#336
mhansen · Nov 18, 2024 · a61950b · a61950b
1 parent 355d8eb
commit a61950b
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/drv/OpenHashMap.drv b/drv/OpenHashMap.drv
@@ -36,6 +36,7 @@ import VALUE_PACKAGE.VALUE_COLLECTION;
 import VALUE_PACKAGE.VALUE_ABSTRACT_COLLECTION;
 
 #if VALUES_PRIMITIVE
+import VALUE_PACKAGE.VALUE_BINARY_OPERATOR;
 import VALUE_PACKAGE.VALUE_ITERATOR;
 import VALUE_PACKAGE.VALUE_SPLITERATOR;
 import VALUE_PACKAGE.VALUE_SPLITERATORS;
@@ -1253,6 +1254,25 @@ public class OPEN_HASH_MAP KEY_VALUE_GENERIC extends ABSTRACT_MAP KEY_VALUE_GENE
 		return value[pos] = newVal;
 	}
 
+#if VALUES_PRIMITIVE && ! VALUE_CLASS_Boolean
+	/** {@inheritDoc} */
+	@Override
+	public VALUE_GENERIC_TYPE MERGE_VALUE(final KEY_GENERIC_TYPE k, final VALUE_GENERIC_TYPE v, METHOD_ARG_VALUE_BINARY_OPERATOR remappingFunction) {
+		java.util.Objects.requireNonNull(remappingFunction);
+		REQUIRE_VALUE_NON_NULL(v)
+
+		final int pos = find(k);
+		if (pos < 0) {
+			insert(-pos - 1, k, v);
+			return v;
+		}
+
+		final VALUE_GENERIC_TYPE newValue = remappingFunction.VALUE_OPERATOR_APPLY(value[pos], v);
+
+		return value[pos] = newValue;
+	}
+#endif
+
 	/** {@inheritDoc} */
 	@Override
 	public VALUE_GENERIC_TYPE merge(final KEY_GENERIC_TYPE k, final VALUE_GENERIC_TYPE v, final java.util.function.BiFunction<? super VALUE_GENERIC_CLASS, ? super VALUE_GENERIC_CLASS, ? extends VALUE_GENERIC_CLASS> remappingFunction) {