lib/threads: optional redefs of collections to be thread-safe

Signed-off-by: Alexis Laferrière <alexis.laf@xymus.net>

diff --git a/lib/pthreads/redef_collections.nit b/lib/pthreads/redef_collections.nit
new file mode 100644 (file)
index 0000000..5e0a18c
--- /dev/null
+++ b/lib/pthreads/redef_collections.nit
@@ -0,0 +1,112 @@
+# This file is part of NIT (http://www.nitlanguage.org).
+#
+# Copyright 2014 Alexis Laferrière <alexis.laf@xymus.net>
+#
+# 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.
+
+# Redef _some_ basic collections to be thread-safe
+#
+# This modules is intended to be used with scripts or quick prototypes.
+# It makes thread safe _all_ instances of _some_ collections which
+# also slightly slow down single threaded use. For more robust software,
+# it is recommended to use `threads::concurrent_collections`.
+#
+# Thread-safe collections:
+#
+# - [x] `Array`
+# - [ ] `List`
+# - [ ] `HashMap`
+# - [ ] `HashSet`
+# - [ ] `Container`
+# - [ ] `Queue`
+module redef_collections
+
+import pthreads
+
+# Thread-safe refinements of most of the known methods (except `enlarge`)
+redef class Array
+       var mutex = new Mutex
+
+       redef fun add(e)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun []=(index, e)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun [](index)
+       do
+               mutex.lock
+               var r = super
+               mutex.unlock
+               return r
+       end
+
+       redef fun remove_at(index)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun shift
+       do
+               mutex.lock
+               var r = super
+               mutex.unlock
+               return r
+       end
+
+       redef fun unshift(e)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun insert_all(from, pos)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun swap_at(a, b)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+
+       redef fun ==(o)
+       do
+               mutex.lock
+               var r = super
+               mutex.unlock
+               return r
+       end
+
+       redef fun enlarge(cap)
+       do
+               mutex.lock
+               super
+               mutex.unlock
+       end
+end