summary refs log tree commit diff stats
path: root/lib/pure/collections/tables.nim
diff options
context:
space:
mode:
authorCharles Blake <cblake@csail.mit.edu>2015-02-16 06:02:31 -0500
committerCharles Blake <cblake@csail.mit.edu>2015-02-16 06:02:31 -0500
commit614a1f9d5a9d87dce230fffe1f59f81c0fe6025a (patch)
treeefd8e2dbaa8ca149c73defa386d7c6ed08fc7ad3 /lib/pure/collections/tables.nim
parent7c1c9a6a9d264feb47b5346c1953b8a72862292e (diff)
downloadNim-614a1f9d5a9d87dce230fffe1f59f81c0fe6025a.tar.gz
Add TableRef version of mgetOrPut.
Diffstat (limited to 'lib/pure/collections/tables.nim')
-rw-r--r--lib/pure/collections/tables.nim5
1 files changed, 5 insertions, 0 deletions
diff --git a/lib/pure/collections/tables.nim b/lib/pure/collections/tables.nim
index 96189baff..6719c79de 100644
--- a/lib/pure/collections/tables.nim
+++ b/lib/pure/collections/tables.nim
@@ -397,6 +397,11 @@ proc mget*[A, B](t: TableRef[A, B], key: A): var B =
   ## If `key` is not in `t`, the ``EInvalidKey`` exception is raised.
   t[].mget(key)
 
+proc mgetOrPut*[A, B](t: TableRef[A, B], key: A, value: B): var B =
+  ## retrieves value at ``t[key]`` or puts ``value`` if not present, either way
+  ## returning a value which can be modified.
+  t[].mgetOrPut(key, value)
+
 proc hasKey*[A, B](t: TableRef[A, B], key: A): bool =
   ## returns true iff `key` is in the table `t`.
   result = t[].hasKey(key)