added val to add(), throws exception if key present

1 files changed, 19 insertions, 2 deletions

diff --git a/src/cli/runtime/PersistentHashtableMap.cs b/src/cli/runtime/PersistentHashtableMap.cs
index d0fd4cc0..51759d25 100644
--- a/src/cli/runtime/PersistentHashtableMap.cs
+++ b/src/cli/runtime/PersistentHashtableMap.cs
@@ -82,8 +82,13 @@ public IMapEntry find(Object key) {
     return null;

 }

 

-public IPersistentMap add(Object key) {

-    return put(key, null);

+public IPersistentMap add(Object key, Object val) {

+    if(_count > growAtCount)

+        return grow().add(key, val);

+    int i = bucketFor(key,array);

+    int incr = 1;

+    PersistentArray newArray = doAdd(i, key, val, array);

+    return create(_count + incr, newArray, growAtCount);

 }

 

 public IPersistentMap put(Object key, Object val) {

@@ -115,6 +120,18 @@ PersistentArray doPut(int i,Object key,Object val,PersistentArray array){
 	return (PersistentArray)array.set(i, newEntries);

 }

 

+PersistentArray doAdd(int i,Object key,Object val,PersistentArray array) {

+    IPersistentMap entries = (IPersistentMap) array.get(i);

+    IPersistentMap newEntries;

+    if (entries != null)

+        {

+        newEntries = entries.add(key, val);

+        }

+    else

+	    newEntries = createListMap(key, val);

+

+	return (PersistentArray)array.set(i, newEntries);

+}

 public IPersistentMap remove(Object key) {

     int i = bucketFor(key,array);

     IPersistentMap entries = (IPersistentMap) array.get(i);