From: Alexis Laferrière <alexis.laf@xymus.net>
Date: Sun, 26 Oct 2014 16:21:22 +0000 (-0400)
Subject: lib/threads: optional redefs of collections to be thread-safe
X-Git-Tag: v0.6.10~8^2~2
X-Git-Url: http://nitlanguage.org

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
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