From 7f4e713e1cce924206be6f757c88d66f460548d5 Mon Sep 17 00:00:00 2001
From: Robinson <robinson@dorkbox.com>
Date: Wed, 2 Aug 2023 22:24:47 -0600
Subject: [PATCH] code cleanup

---
 src/dorkbox/collections/ArrayMap.kt | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/src/dorkbox/collections/ArrayMap.kt b/src/dorkbox/collections/ArrayMap.kt
index 74a11a0..bdd356c 100644
--- a/src/dorkbox/collections/ArrayMap.kt
+++ b/src/dorkbox/collections/ArrayMap.kt
@@ -35,7 +35,6 @@ package dorkbox.collections
 import dorkbox.collections.Collections.allocateIterators
 import dorkbox.collections.Collections.random
 import dorkbox.collections.ObjectMap.Companion.dummy
-import dorkbox.collections.ObjectMap.Entry
 import java.lang.IllegalStateException
 import java.util.*
 import kotlin.math.max
@@ -884,6 +883,22 @@ class ArrayMap<K: Any, V> : MutableMap<K, V?>{
         }
     }
 
+    class Entry<K: Any, V>(val map: ArrayMap<K, V?>) : MutableMap.MutableEntry<K, V?> {
+        override lateinit var key: K
+        override var value: V? = null
+
+        override fun setValue(newValue: V?): V? {
+            val oldValue = value
+            map[key] = newValue
+            value = newValue
+            return oldValue
+        }
+
+        override fun toString(): String {
+            return "$key=$value"
+        }
+    }
+
     class Values<V>(map: ArrayMap<Any, V?>) : MutableCollection<V>, Iterable<V>, MutableIterator<V> {
         private val map: ArrayMap<Any, V?>
         var index = 0