concurrent-ruby 1.0.5 → 1.1.10

Sign up to get free protection for your applications and to get access to all the features.
Files changed (164) hide show
  1. checksums.yaml +5 -5
  2. data/CHANGELOG.md +155 -0
  3. data/Gemfile +37 -0
  4. data/LICENSE.txt +18 -18
  5. data/README.md +260 -103
  6. data/Rakefile +329 -0
  7. data/ext/concurrent-ruby/ConcurrentRubyService.java +17 -0
  8. data/ext/concurrent-ruby/com/concurrent_ruby/ext/AtomicReferenceLibrary.java +175 -0
  9. data/ext/concurrent-ruby/com/concurrent_ruby/ext/JRubyMapBackendLibrary.java +248 -0
  10. data/ext/concurrent-ruby/com/concurrent_ruby/ext/JavaAtomicBooleanLibrary.java +93 -0
  11. data/ext/concurrent-ruby/com/concurrent_ruby/ext/JavaAtomicFixnumLibrary.java +113 -0
  12. data/ext/concurrent-ruby/com/concurrent_ruby/ext/JavaSemaphoreLibrary.java +189 -0
  13. data/ext/concurrent-ruby/com/concurrent_ruby/ext/SynchronizationLibrary.java +307 -0
  14. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/ConcurrentHashMap.java +31 -0
  15. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/ConcurrentHashMapV8.java +3863 -0
  16. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/LongAdder.java +203 -0
  17. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/Striped64.java +342 -0
  18. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/nounsafe/ConcurrentHashMapV8.java +3800 -0
  19. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/nounsafe/LongAdder.java +204 -0
  20. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166e/nounsafe/Striped64.java +291 -0
  21. data/ext/concurrent-ruby/com/concurrent_ruby/ext/jsr166y/ThreadLocalRandom.java +199 -0
  22. data/lib/{concurrent → concurrent-ruby/concurrent}/agent.rb +7 -7
  23. data/lib/concurrent-ruby/concurrent/array.rb +66 -0
  24. data/lib/{concurrent → concurrent-ruby/concurrent}/async.rb +28 -24
  25. data/lib/{concurrent → concurrent-ruby/concurrent}/atom.rb +10 -10
  26. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/atomic_boolean.rb +26 -22
  27. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/atomic_fixnum.rb +27 -23
  28. data/lib/concurrent-ruby/concurrent/atomic/atomic_markable_reference.rb +164 -0
  29. data/lib/concurrent-ruby/concurrent/atomic/atomic_reference.rb +205 -0
  30. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/count_down_latch.rb +7 -7
  31. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/cyclic_barrier.rb +1 -1
  32. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/event.rb +3 -3
  33. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/java_count_down_latch.rb +9 -6
  34. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/mutex_atomic_boolean.rb +2 -0
  35. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/mutex_count_down_latch.rb +1 -0
  36. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/mutex_semaphore.rb +18 -2
  37. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/read_write_lock.rb +2 -1
  38. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/reentrant_read_write_lock.rb +7 -7
  39. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/ruby_thread_local_var.rb +60 -40
  40. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/semaphore.rb +34 -13
  41. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/thread_local_var.rb +8 -8
  42. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic_reference/mutex_atomic.rb +3 -8
  43. data/lib/{concurrent → concurrent-ruby/concurrent}/atomic_reference/numeric_cas_wrapper.rb +1 -1
  44. data/lib/concurrent-ruby/concurrent/atomics.rb +10 -0
  45. data/lib/concurrent-ruby/concurrent/collection/lock_free_stack.rb +158 -0
  46. data/lib/{concurrent → concurrent-ruby/concurrent}/collection/map/atomic_reference_map_backend.rb +3 -3
  47. data/lib/{concurrent → concurrent-ruby/concurrent}/collection/map/mri_map_backend.rb +1 -1
  48. data/lib/{concurrent → concurrent-ruby/concurrent}/collection/map/non_concurrent_map_backend.rb +1 -2
  49. data/lib/concurrent-ruby/concurrent/collection/map/truffleruby_map_backend.rb +14 -0
  50. data/lib/{concurrent → concurrent-ruby/concurrent}/collection/non_concurrent_priority_queue.rb +30 -30
  51. data/lib/{concurrent → concurrent-ruby/concurrent}/collection/ruby_non_concurrent_priority_queue.rb +11 -1
  52. data/lib/{concurrent → concurrent-ruby/concurrent}/concern/dereferenceable.rb +3 -3
  53. data/lib/{concurrent → concurrent-ruby/concurrent}/concern/logging.rb +6 -1
  54. data/lib/{concurrent → concurrent-ruby/concurrent}/concern/observable.rb +7 -7
  55. data/lib/concurrent-ruby/concurrent/concurrent_ruby.jar +0 -0
  56. data/lib/{concurrent → concurrent-ruby/concurrent}/configuration.rb +15 -15
  57. data/lib/{concurrent → concurrent-ruby/concurrent}/constants.rb +1 -1
  58. data/lib/{concurrent → concurrent-ruby/concurrent}/dataflow.rb +2 -1
  59. data/lib/{concurrent → concurrent-ruby/concurrent}/delay.rb +9 -7
  60. data/lib/{concurrent → concurrent-ruby/concurrent}/exchanger.rb +21 -25
  61. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/abstract_executor_service.rb +35 -38
  62. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/cached_thread_pool.rb +5 -5
  63. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/executor_service.rb +17 -17
  64. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/fixed_thread_pool.rb +47 -33
  65. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/java_executor_service.rb +20 -17
  66. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/java_single_thread_executor.rb +4 -3
  67. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/java_thread_pool_executor.rb +29 -9
  68. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/ruby_executor_service.rb +10 -6
  69. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/ruby_single_thread_executor.rb +0 -1
  70. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/ruby_thread_pool_executor.rb +46 -42
  71. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/safe_task_executor.rb +5 -5
  72. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/simple_executor_service.rb +1 -1
  73. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/single_thread_executor.rb +3 -2
  74. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/thread_pool_executor.rb +7 -6
  75. data/lib/{concurrent → concurrent-ruby/concurrent}/executor/timer_set.rb +14 -17
  76. data/lib/{concurrent → concurrent-ruby/concurrent}/future.rb +4 -1
  77. data/lib/concurrent-ruby/concurrent/hash.rb +59 -0
  78. data/lib/{concurrent → concurrent-ruby/concurrent}/immutable_struct.rb +9 -1
  79. data/lib/{concurrent → concurrent-ruby/concurrent}/ivar.rb +5 -6
  80. data/lib/concurrent-ruby/concurrent/map.rb +346 -0
  81. data/lib/{concurrent → concurrent-ruby/concurrent}/maybe.rb +1 -1
  82. data/lib/{concurrent → concurrent-ruby/concurrent}/mutable_struct.rb +27 -16
  83. data/lib/{concurrent → concurrent-ruby/concurrent}/mvar.rb +2 -2
  84. data/lib/{concurrent → concurrent-ruby/concurrent}/promise.rb +54 -21
  85. data/lib/concurrent-ruby/concurrent/promises.rb +2167 -0
  86. data/lib/concurrent-ruby/concurrent/re_include.rb +58 -0
  87. data/lib/{concurrent → concurrent-ruby/concurrent}/scheduled_task.rb +29 -16
  88. data/lib/concurrent-ruby/concurrent/set.rb +74 -0
  89. data/lib/{concurrent → concurrent-ruby/concurrent}/settable_struct.rb +12 -1
  90. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/abstract_lockable_object.rb +5 -5
  91. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/abstract_struct.rb +18 -4
  92. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/condition.rb +2 -0
  93. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/jruby_object.rb +1 -0
  94. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/lock.rb +2 -0
  95. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/lockable_object.rb +8 -10
  96. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/mri_object.rb +1 -0
  97. data/lib/concurrent-ruby/concurrent/synchronization/mutex_lockable_object.rb +88 -0
  98. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/object.rb +53 -23
  99. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/rbx_lockable_object.rb +6 -0
  100. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/rbx_object.rb +1 -0
  101. data/lib/concurrent-ruby/concurrent/synchronization/truffleruby_object.rb +47 -0
  102. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/volatile.rb +11 -9
  103. data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization.rb +4 -5
  104. data/lib/concurrent-ruby/concurrent/thread_safe/util/data_structures.rb +88 -0
  105. data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/striped64.rb +9 -4
  106. data/lib/{concurrent → concurrent-ruby/concurrent}/timer_task.rb +15 -35
  107. data/lib/{concurrent → concurrent-ruby/concurrent}/tuple.rb +1 -1
  108. data/lib/{concurrent → concurrent-ruby/concurrent}/tvar.rb +21 -58
  109. data/lib/{concurrent → concurrent-ruby/concurrent}/utility/engine.rb +4 -4
  110. data/lib/concurrent-ruby/concurrent/utility/monotonic_time.rb +90 -0
  111. data/lib/concurrent-ruby/concurrent/utility/native_extension_loader.rb +79 -0
  112. data/lib/{concurrent → concurrent-ruby/concurrent}/utility/processor_counter.rb +5 -35
  113. data/lib/concurrent-ruby/concurrent/version.rb +3 -0
  114. data/lib/concurrent-ruby/concurrent-ruby.rb +5 -0
  115. data/lib/{concurrent.rb → concurrent-ruby/concurrent.rb} +24 -20
  116. metadata +149 -134
  117. data/lib/concurrent/array.rb +0 -39
  118. data/lib/concurrent/atomic/atomic_reference.rb +0 -51
  119. data/lib/concurrent/atomic_reference/concurrent_update_error.rb +0 -8
  120. data/lib/concurrent/atomic_reference/direct_update.rb +0 -81
  121. data/lib/concurrent/atomic_reference/jruby+truffle.rb +0 -2
  122. data/lib/concurrent/atomic_reference/jruby.rb +0 -16
  123. data/lib/concurrent/atomic_reference/rbx.rb +0 -22
  124. data/lib/concurrent/atomic_reference/ruby.rb +0 -32
  125. data/lib/concurrent/atomics.rb +0 -53
  126. data/lib/concurrent/edge.rb +0 -26
  127. data/lib/concurrent/hash.rb +0 -36
  128. data/lib/concurrent/lazy_register.rb +0 -81
  129. data/lib/concurrent/map.rb +0 -240
  130. data/lib/concurrent/synchronization/mri_lockable_object.rb +0 -71
  131. data/lib/concurrent/synchronization/truffle_lockable_object.rb +0 -9
  132. data/lib/concurrent/synchronization/truffle_object.rb +0 -31
  133. data/lib/concurrent/thread_safe/util/array_hash_rbx.rb +0 -30
  134. data/lib/concurrent/utility/at_exit.rb +0 -97
  135. data/lib/concurrent/utility/monotonic_time.rb +0 -58
  136. data/lib/concurrent/utility/native_extension_loader.rb +0 -73
  137. data/lib/concurrent/version.rb +0 -4
  138. /data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/abstract_thread_local_var.rb +0 -0
  139. /data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/java_thread_local_var.rb +0 -0
  140. /data/lib/{concurrent → concurrent-ruby/concurrent}/atomic/mutex_atomic_fixnum.rb +0 -0
  141. /data/lib/{concurrent → concurrent-ruby/concurrent}/collection/copy_on_notify_observer_set.rb +0 -0
  142. /data/lib/{concurrent → concurrent-ruby/concurrent}/collection/copy_on_write_observer_set.rb +0 -0
  143. /data/lib/{concurrent → concurrent-ruby/concurrent}/collection/java_non_concurrent_priority_queue.rb +0 -0
  144. /data/lib/{concurrent → concurrent-ruby/concurrent}/collection/map/synchronized_map_backend.rb +0 -0
  145. /data/lib/{concurrent → concurrent-ruby/concurrent}/concern/deprecation.rb +0 -0
  146. /data/lib/{concurrent → concurrent-ruby/concurrent}/concern/obligation.rb +0 -0
  147. /data/lib/{concurrent → concurrent-ruby/concurrent}/errors.rb +0 -0
  148. /data/lib/{concurrent → concurrent-ruby/concurrent}/executor/immediate_executor.rb +0 -0
  149. /data/lib/{concurrent → concurrent-ruby/concurrent}/executor/indirect_immediate_executor.rb +0 -0
  150. /data/lib/{concurrent → concurrent-ruby/concurrent}/executor/serial_executor_service.rb +0 -0
  151. /data/lib/{concurrent → concurrent-ruby/concurrent}/executor/serialized_execution.rb +0 -0
  152. /data/lib/{concurrent → concurrent-ruby/concurrent}/executor/serialized_execution_delegator.rb +0 -0
  153. /data/lib/{concurrent → concurrent-ruby/concurrent}/executors.rb +0 -0
  154. /data/lib/{concurrent → concurrent-ruby/concurrent}/options.rb +0 -0
  155. /data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/abstract_object.rb +0 -0
  156. /data/lib/{concurrent → concurrent-ruby/concurrent}/synchronization/jruby_lockable_object.rb +0 -0
  157. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/synchronized_delegator.rb +0 -0
  158. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/adder.rb +0 -0
  159. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/cheap_lockable.rb +0 -0
  160. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/power_of_two_tuple.rb +0 -0
  161. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/volatile.rb +0 -0
  162. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util/xor_shift_random.rb +0 -0
  163. /data/lib/{concurrent → concurrent-ruby/concurrent}/thread_safe/util.rb +0 -0
  164. /data/lib/{concurrent → concurrent-ruby/concurrent}/utility/native_integer.rb +0 -0
@@ -0,0 +1,248 @@
1
+ package com.concurrent_ruby.ext;
2
+
3
+ import org.jruby.*;
4
+ import org.jruby.anno.JRubyClass;
5
+ import org.jruby.anno.JRubyMethod;
6
+ import com.concurrent_ruby.ext.jsr166e.ConcurrentHashMap;
7
+ import com.concurrent_ruby.ext.jsr166e.ConcurrentHashMapV8;
8
+ import com.concurrent_ruby.ext.jsr166e.nounsafe.*;
9
+ import org.jruby.runtime.Block;
10
+ import org.jruby.runtime.ObjectAllocator;
11
+ import org.jruby.runtime.ThreadContext;
12
+ import org.jruby.runtime.builtin.IRubyObject;
13
+ import org.jruby.runtime.load.Library;
14
+
15
+ import java.io.IOException;
16
+ import java.util.Map;
17
+
18
+ import static org.jruby.runtime.Visibility.PRIVATE;
19
+
20
+ /**
21
+ * Native Java implementation to avoid the JI overhead.
22
+ *
23
+ * @author thedarkone
24
+ */
25
+ public class JRubyMapBackendLibrary implements Library {
26
+ public void load(Ruby runtime, boolean wrap) throws IOException {
27
+
28
+ RubyModule concurrentMod = runtime.defineModule("Concurrent");
29
+ RubyModule thread_safeMod = concurrentMod.defineModuleUnder("Collection");
30
+ RubyClass jrubyRefClass = thread_safeMod.defineClassUnder("JRubyMapBackend", runtime.getObject(), BACKEND_ALLOCATOR);
31
+ jrubyRefClass.setAllocator(BACKEND_ALLOCATOR);
32
+ jrubyRefClass.defineAnnotatedMethods(JRubyMapBackend.class);
33
+ }
34
+
35
+ private static final ObjectAllocator BACKEND_ALLOCATOR = new ObjectAllocator() {
36
+ public IRubyObject allocate(Ruby runtime, RubyClass klazz) {
37
+ return new JRubyMapBackend(runtime, klazz);
38
+ }
39
+ };
40
+
41
+ @JRubyClass(name="JRubyMapBackend", parent="Object")
42
+ public static class JRubyMapBackend extends RubyObject {
43
+ // Defaults used by the CHM
44
+ static final int DEFAULT_INITIAL_CAPACITY = 16;
45
+ static final float DEFAULT_LOAD_FACTOR = 0.75f;
46
+
47
+ public static final boolean CAN_USE_UNSAFE_CHM = canUseUnsafeCHM();
48
+
49
+ private ConcurrentHashMap<IRubyObject, IRubyObject> map;
50
+
51
+ private static ConcurrentHashMap<IRubyObject, IRubyObject> newCHM(int initialCapacity, float loadFactor) {
52
+ if (CAN_USE_UNSAFE_CHM) {
53
+ return new ConcurrentHashMapV8<IRubyObject, IRubyObject>(initialCapacity, loadFactor);
54
+ } else {
55
+ return new com.concurrent_ruby.ext.jsr166e.nounsafe.ConcurrentHashMapV8<IRubyObject, IRubyObject>(initialCapacity, loadFactor);
56
+ }
57
+ }
58
+
59
+ private static ConcurrentHashMap<IRubyObject, IRubyObject> newCHM() {
60
+ return newCHM(DEFAULT_INITIAL_CAPACITY, DEFAULT_LOAD_FACTOR);
61
+ }
62
+
63
+ private static boolean canUseUnsafeCHM() {
64
+ try {
65
+ new com.concurrent_ruby.ext.jsr166e.ConcurrentHashMapV8(); // force class load and initialization
66
+ return true;
67
+ } catch (Throwable t) { // ensuring we really do catch everything
68
+ // Doug's Unsafe setup errors always have this "Could not ini.." message
69
+ if (isCausedBySecurityException(t)) {
70
+ return false;
71
+ }
72
+ throw (t instanceof RuntimeException ? (RuntimeException) t : new RuntimeException(t));
73
+ }
74
+ }
75
+
76
+ private static boolean isCausedBySecurityException(Throwable t) {
77
+ while (t != null) {
78
+ if ((t.getMessage() != null && t.getMessage().contains("Could not initialize intrinsics")) || t instanceof SecurityException) {
79
+ return true;
80
+ }
81
+ t = t.getCause();
82
+ }
83
+ return false;
84
+ }
85
+
86
+ public JRubyMapBackend(Ruby runtime, RubyClass klass) {
87
+ super(runtime, klass);
88
+ }
89
+
90
+ @JRubyMethod
91
+ public IRubyObject initialize(ThreadContext context) {
92
+ map = newCHM();
93
+ return context.getRuntime().getNil();
94
+ }
95
+
96
+ @JRubyMethod
97
+ public IRubyObject initialize(ThreadContext context, IRubyObject options) {
98
+ map = toCHM(context, options);
99
+ return context.getRuntime().getNil();
100
+ }
101
+
102
+ private ConcurrentHashMap<IRubyObject, IRubyObject> toCHM(ThreadContext context, IRubyObject options) {
103
+ Ruby runtime = context.getRuntime();
104
+ if (!options.isNil() && options.respondsTo("[]")) {
105
+ IRubyObject rInitialCapacity = options.callMethod(context, "[]", runtime.newSymbol("initial_capacity"));
106
+ IRubyObject rLoadFactor = options.callMethod(context, "[]", runtime.newSymbol("load_factor"));
107
+ int initialCapacity = !rInitialCapacity.isNil() ? RubyNumeric.num2int(rInitialCapacity.convertToInteger()) : DEFAULT_INITIAL_CAPACITY;
108
+ float loadFactor = !rLoadFactor.isNil() ? (float)RubyNumeric.num2dbl(rLoadFactor.convertToFloat()) : DEFAULT_LOAD_FACTOR;
109
+ return newCHM(initialCapacity, loadFactor);
110
+ } else {
111
+ return newCHM();
112
+ }
113
+ }
114
+
115
+ @JRubyMethod(name = "[]", required = 1)
116
+ public IRubyObject op_aref(ThreadContext context, IRubyObject key) {
117
+ IRubyObject value;
118
+ return ((value = map.get(key)) == null) ? context.getRuntime().getNil() : value;
119
+ }
120
+
121
+ @JRubyMethod(name = {"[]="}, required = 2)
122
+ public IRubyObject op_aset(IRubyObject key, IRubyObject value) {
123
+ map.put(key, value);
124
+ return value;
125
+ }
126
+
127
+ @JRubyMethod
128
+ public IRubyObject put_if_absent(IRubyObject key, IRubyObject value) {
129
+ IRubyObject result = map.putIfAbsent(key, value);
130
+ return result == null ? getRuntime().getNil() : result;
131
+ }
132
+
133
+ @JRubyMethod
134
+ public IRubyObject compute_if_absent(final ThreadContext context, final IRubyObject key, final Block block) {
135
+ return map.computeIfAbsent(key, new ConcurrentHashMap.Fun<IRubyObject, IRubyObject>() {
136
+ @Override
137
+ public IRubyObject apply(IRubyObject key) {
138
+ return block.yieldSpecific(context);
139
+ }
140
+ });
141
+ }
142
+
143
+ @JRubyMethod
144
+ public IRubyObject compute_if_present(final ThreadContext context, final IRubyObject key, final Block block) {
145
+ IRubyObject result = map.computeIfPresent(key, new ConcurrentHashMap.BiFun<IRubyObject, IRubyObject, IRubyObject>() {
146
+ @Override
147
+ public IRubyObject apply(IRubyObject key, IRubyObject oldValue) {
148
+ IRubyObject result = block.yieldSpecific(context, oldValue == null ? context.getRuntime().getNil() : oldValue);
149
+ return result.isNil() ? null : result;
150
+ }
151
+ });
152
+ return result == null ? context.getRuntime().getNil() : result;
153
+ }
154
+
155
+ @JRubyMethod
156
+ public IRubyObject compute(final ThreadContext context, final IRubyObject key, final Block block) {
157
+ IRubyObject result = map.compute(key, new ConcurrentHashMap.BiFun<IRubyObject, IRubyObject, IRubyObject>() {
158
+ @Override
159
+ public IRubyObject apply(IRubyObject key, IRubyObject oldValue) {
160
+ IRubyObject result = block.yieldSpecific(context, oldValue == null ? context.getRuntime().getNil() : oldValue);
161
+ return result.isNil() ? null : result;
162
+ }
163
+ });
164
+ return result == null ? context.getRuntime().getNil() : result;
165
+ }
166
+
167
+ @JRubyMethod
168
+ public IRubyObject merge_pair(final ThreadContext context, final IRubyObject key, final IRubyObject value, final Block block) {
169
+ IRubyObject result = map.merge(key, value, new ConcurrentHashMap.BiFun<IRubyObject, IRubyObject, IRubyObject>() {
170
+ @Override
171
+ public IRubyObject apply(IRubyObject oldValue, IRubyObject newValue) {
172
+ IRubyObject result = block.yieldSpecific(context, oldValue == null ? context.getRuntime().getNil() : oldValue);
173
+ return result.isNil() ? null : result;
174
+ }
175
+ });
176
+ return result == null ? context.getRuntime().getNil() : result;
177
+ }
178
+
179
+ @JRubyMethod
180
+ public RubyBoolean replace_pair(IRubyObject key, IRubyObject oldValue, IRubyObject newValue) {
181
+ return getRuntime().newBoolean(map.replace(key, oldValue, newValue));
182
+ }
183
+
184
+ @JRubyMethod(name = "key?", required = 1)
185
+ public RubyBoolean has_key_p(IRubyObject key) {
186
+ return map.containsKey(key) ? getRuntime().getTrue() : getRuntime().getFalse();
187
+ }
188
+
189
+ @JRubyMethod
190
+ public IRubyObject key(IRubyObject value) {
191
+ final IRubyObject key = map.findKey(value);
192
+ return key == null ? getRuntime().getNil() : key;
193
+ }
194
+
195
+ @JRubyMethod
196
+ public IRubyObject replace_if_exists(IRubyObject key, IRubyObject value) {
197
+ IRubyObject result = map.replace(key, value);
198
+ return result == null ? getRuntime().getNil() : result;
199
+ }
200
+
201
+ @JRubyMethod
202
+ public IRubyObject get_and_set(IRubyObject key, IRubyObject value) {
203
+ IRubyObject result = map.put(key, value);
204
+ return result == null ? getRuntime().getNil() : result;
205
+ }
206
+
207
+ @JRubyMethod
208
+ public IRubyObject delete(IRubyObject key) {
209
+ IRubyObject result = map.remove(key);
210
+ return result == null ? getRuntime().getNil() : result;
211
+ }
212
+
213
+ @JRubyMethod
214
+ public RubyBoolean delete_pair(IRubyObject key, IRubyObject value) {
215
+ return getRuntime().newBoolean(map.remove(key, value));
216
+ }
217
+
218
+ @JRubyMethod
219
+ public IRubyObject clear() {
220
+ map.clear();
221
+ return this;
222
+ }
223
+
224
+ @JRubyMethod
225
+ public IRubyObject each_pair(ThreadContext context, Block block) {
226
+ for (Map.Entry<IRubyObject,IRubyObject> entry : map.entrySet()) {
227
+ block.yieldSpecific(context, entry.getKey(), entry.getValue());
228
+ }
229
+ return this;
230
+ }
231
+
232
+ @JRubyMethod
233
+ public RubyFixnum size(ThreadContext context) {
234
+ return context.getRuntime().newFixnum(map.size());
235
+ }
236
+
237
+ @JRubyMethod
238
+ public IRubyObject get_or_default(IRubyObject key, IRubyObject defaultValue) {
239
+ return map.getValueOrDefault(key, defaultValue);
240
+ }
241
+
242
+ @JRubyMethod(visibility = PRIVATE)
243
+ public JRubyMapBackend initialize_copy(ThreadContext context, IRubyObject other) {
244
+ map = newCHM();
245
+ return this;
246
+ }
247
+ }
248
+ }
@@ -0,0 +1,93 @@
1
+ package com.concurrent_ruby.ext;
2
+
3
+ import org.jruby.Ruby;
4
+ import org.jruby.RubyBoolean;
5
+ import org.jruby.RubyClass;
6
+ import org.jruby.RubyModule;
7
+ import org.jruby.RubyNil;
8
+ import org.jruby.RubyObject;
9
+ import org.jruby.anno.JRubyClass;
10
+ import org.jruby.anno.JRubyMethod;
11
+ import org.jruby.runtime.ObjectAllocator;
12
+ import org.jruby.runtime.ThreadContext;
13
+ import org.jruby.runtime.builtin.IRubyObject;
14
+ import org.jruby.runtime.load.Library;
15
+
16
+ import java.io.IOException;
17
+ import java.util.concurrent.atomic.AtomicBoolean;
18
+
19
+ public class JavaAtomicBooleanLibrary implements Library {
20
+
21
+ public void load(Ruby runtime, boolean wrap) throws IOException {
22
+ RubyModule concurrentMod = runtime.defineModule("Concurrent");
23
+ RubyClass atomicCls = concurrentMod.defineClassUnder("JavaAtomicBoolean", runtime.getObject(), JRUBYREFERENCE_ALLOCATOR);
24
+ atomicCls.defineAnnotatedMethods(JavaAtomicBoolean.class);
25
+ }
26
+
27
+ private static final ObjectAllocator JRUBYREFERENCE_ALLOCATOR = new ObjectAllocator() {
28
+ public IRubyObject allocate(Ruby runtime, RubyClass klazz) {
29
+ return new JavaAtomicBoolean(runtime, klazz);
30
+ }
31
+ };
32
+
33
+ @JRubyClass(name = "JavaAtomicBoolean", parent = "Object")
34
+ public static class JavaAtomicBoolean extends RubyObject {
35
+
36
+ private AtomicBoolean atomicBoolean;
37
+
38
+ public JavaAtomicBoolean(Ruby runtime, RubyClass metaClass) {
39
+ super(runtime, metaClass);
40
+ }
41
+
42
+ @JRubyMethod
43
+ public IRubyObject initialize(ThreadContext context, IRubyObject value) {
44
+ atomicBoolean = new AtomicBoolean(convertRubyBooleanToJavaBoolean(value));
45
+ return context.nil;
46
+ }
47
+
48
+ @JRubyMethod
49
+ public IRubyObject initialize(ThreadContext context) {
50
+ atomicBoolean = new AtomicBoolean();
51
+ return context.nil;
52
+ }
53
+
54
+ @JRubyMethod(name = "value")
55
+ public IRubyObject value() {
56
+ return getRuntime().newBoolean(atomicBoolean.get());
57
+ }
58
+
59
+ @JRubyMethod(name = "true?")
60
+ public IRubyObject isAtomicTrue() {
61
+ return getRuntime().newBoolean(atomicBoolean.get());
62
+ }
63
+
64
+ @JRubyMethod(name = "false?")
65
+ public IRubyObject isAtomicFalse() {
66
+ return getRuntime().newBoolean((atomicBoolean.get() == false));
67
+ }
68
+
69
+ @JRubyMethod(name = "value=")
70
+ public IRubyObject setAtomic(ThreadContext context, IRubyObject newValue) {
71
+ atomicBoolean.set(convertRubyBooleanToJavaBoolean(newValue));
72
+ return context.nil;
73
+ }
74
+
75
+ @JRubyMethod(name = "make_true")
76
+ public IRubyObject makeTrue() {
77
+ return getRuntime().newBoolean(atomicBoolean.compareAndSet(false, true));
78
+ }
79
+
80
+ @JRubyMethod(name = "make_false")
81
+ public IRubyObject makeFalse() {
82
+ return getRuntime().newBoolean(atomicBoolean.compareAndSet(true, false));
83
+ }
84
+
85
+ private boolean convertRubyBooleanToJavaBoolean(IRubyObject newValue) {
86
+ if (newValue instanceof RubyBoolean.False || newValue instanceof RubyNil) {
87
+ return false;
88
+ } else {
89
+ return true;
90
+ }
91
+ }
92
+ }
93
+ }
@@ -0,0 +1,113 @@
1
+ package com.concurrent_ruby.ext;
2
+
3
+ import java.io.IOException;
4
+ import java.util.concurrent.atomic.AtomicLong;
5
+ import org.jruby.Ruby;
6
+ import org.jruby.RubyClass;
7
+ import org.jruby.RubyFixnum;
8
+ import org.jruby.RubyModule;
9
+ import org.jruby.RubyObject;
10
+ import org.jruby.anno.JRubyClass;
11
+ import org.jruby.anno.JRubyMethod;
12
+ import org.jruby.runtime.ObjectAllocator;
13
+ import org.jruby.runtime.ThreadContext;
14
+ import org.jruby.runtime.builtin.IRubyObject;
15
+ import org.jruby.runtime.load.Library;
16
+ import org.jruby.runtime.Block;
17
+
18
+ public class JavaAtomicFixnumLibrary implements Library {
19
+
20
+ public void load(Ruby runtime, boolean wrap) throws IOException {
21
+ RubyModule concurrentMod = runtime.defineModule("Concurrent");
22
+ RubyClass atomicCls = concurrentMod.defineClassUnder("JavaAtomicFixnum", runtime.getObject(), JRUBYREFERENCE_ALLOCATOR);
23
+
24
+ atomicCls.defineAnnotatedMethods(JavaAtomicFixnum.class);
25
+ }
26
+
27
+ private static final ObjectAllocator JRUBYREFERENCE_ALLOCATOR = new ObjectAllocator() {
28
+ public IRubyObject allocate(Ruby runtime, RubyClass klazz) {
29
+ return new JavaAtomicFixnum(runtime, klazz);
30
+ }
31
+ };
32
+
33
+ @JRubyClass(name = "JavaAtomicFixnum", parent = "Object")
34
+ public static class JavaAtomicFixnum extends RubyObject {
35
+
36
+ private AtomicLong atomicLong;
37
+
38
+ public JavaAtomicFixnum(Ruby runtime, RubyClass metaClass) {
39
+ super(runtime, metaClass);
40
+ }
41
+
42
+ @JRubyMethod
43
+ public IRubyObject initialize(ThreadContext context) {
44
+ this.atomicLong = new AtomicLong(0);
45
+ return context.nil;
46
+ }
47
+
48
+ @JRubyMethod
49
+ public IRubyObject initialize(ThreadContext context, IRubyObject value) {
50
+ this.atomicLong = new AtomicLong(rubyFixnumToLong(value));
51
+ return context.nil;
52
+ }
53
+
54
+ @JRubyMethod(name = "value")
55
+ public IRubyObject getValue() {
56
+ return getRuntime().newFixnum(atomicLong.get());
57
+ }
58
+
59
+ @JRubyMethod(name = "value=")
60
+ public IRubyObject setValue(ThreadContext context, IRubyObject newValue) {
61
+ atomicLong.set(rubyFixnumToLong(newValue));
62
+ return context.nil;
63
+ }
64
+
65
+ @JRubyMethod(name = {"increment", "up"})
66
+ public IRubyObject increment() {
67
+ return getRuntime().newFixnum(atomicLong.incrementAndGet());
68
+ }
69
+
70
+ @JRubyMethod(name = {"increment", "up"})
71
+ public IRubyObject increment(IRubyObject value) {
72
+ long delta = rubyFixnumToLong(value);
73
+ return getRuntime().newFixnum(atomicLong.addAndGet(delta));
74
+ }
75
+
76
+ @JRubyMethod(name = {"decrement", "down"})
77
+ public IRubyObject decrement() {
78
+ return getRuntime().newFixnum(atomicLong.decrementAndGet());
79
+ }
80
+
81
+ @JRubyMethod(name = {"decrement", "down"})
82
+ public IRubyObject decrement(IRubyObject value) {
83
+ long delta = rubyFixnumToLong(value);
84
+ return getRuntime().newFixnum(atomicLong.addAndGet(-delta));
85
+ }
86
+
87
+ @JRubyMethod(name = "compare_and_set")
88
+ public IRubyObject compareAndSet(ThreadContext context, IRubyObject expect, IRubyObject update) {
89
+ return getRuntime().newBoolean(atomicLong.compareAndSet(rubyFixnumToLong(expect), rubyFixnumToLong(update)));
90
+ }
91
+
92
+ @JRubyMethod
93
+ public IRubyObject update(ThreadContext context, Block block) {
94
+ for (;;) {
95
+ long _oldValue = atomicLong.get();
96
+ IRubyObject oldValue = getRuntime().newFixnum(_oldValue);
97
+ IRubyObject newValue = block.yield(context, oldValue);
98
+ if (atomicLong.compareAndSet(_oldValue, rubyFixnumToLong(newValue))) {
99
+ return newValue;
100
+ }
101
+ }
102
+ }
103
+
104
+ private long rubyFixnumToLong(IRubyObject value) {
105
+ if (value instanceof RubyFixnum) {
106
+ RubyFixnum fixNum = (RubyFixnum) value;
107
+ return fixNum.getLongValue();
108
+ } else {
109
+ throw getRuntime().newArgumentError("value must be a Fixnum");
110
+ }
111
+ }
112
+ }
113
+ }
@@ -0,0 +1,189 @@
1
+ package com.concurrent_ruby.ext;
2
+
3
+ import java.io.IOException;
4
+ import java.util.concurrent.Semaphore;
5
+ import org.jruby.Ruby;
6
+ import org.jruby.RubyClass;
7
+ import org.jruby.RubyFixnum;
8
+ import org.jruby.RubyModule;
9
+ import org.jruby.RubyNumeric;
10
+ import org.jruby.RubyObject;
11
+ import org.jruby.anno.JRubyClass;
12
+ import org.jruby.anno.JRubyMethod;
13
+ import org.jruby.runtime.Block;
14
+ import org.jruby.runtime.ObjectAllocator;
15
+ import org.jruby.runtime.ThreadContext;
16
+ import org.jruby.runtime.builtin.IRubyObject;
17
+
18
+ public class JavaSemaphoreLibrary {
19
+
20
+ public void load(Ruby runtime, boolean wrap) throws IOException {
21
+ RubyModule concurrentMod = runtime.defineModule("Concurrent");
22
+ RubyClass atomicCls = concurrentMod.defineClassUnder("JavaSemaphore", runtime.getObject(), JRUBYREFERENCE_ALLOCATOR);
23
+
24
+ atomicCls.defineAnnotatedMethods(JavaSemaphore.class);
25
+ }
26
+
27
+ private static final ObjectAllocator JRUBYREFERENCE_ALLOCATOR = new ObjectAllocator() {
28
+ public IRubyObject allocate(Ruby runtime, RubyClass klazz) {
29
+ return new JavaSemaphore(runtime, klazz);
30
+ }
31
+ };
32
+
33
+ @JRubyClass(name = "JavaSemaphore", parent = "Object")
34
+ public static class JavaSemaphore extends RubyObject {
35
+
36
+ private JRubySemaphore semaphore;
37
+
38
+ public JavaSemaphore(Ruby runtime, RubyClass metaClass) {
39
+ super(runtime, metaClass);
40
+ }
41
+
42
+ @JRubyMethod
43
+ public IRubyObject initialize(ThreadContext context, IRubyObject value) {
44
+ this.semaphore = new JRubySemaphore(rubyFixnumInt(value, "count"));
45
+ return context.nil;
46
+ }
47
+
48
+ @JRubyMethod
49
+ public IRubyObject acquire(ThreadContext context, final Block block) throws InterruptedException {
50
+ return this.acquire(context, 1, block);
51
+ }
52
+
53
+ @JRubyMethod
54
+ public IRubyObject acquire(ThreadContext context, IRubyObject permits, final Block block) throws InterruptedException {
55
+ return this.acquire(context, rubyFixnumToPositiveInt(permits, "permits"), block);
56
+ }
57
+
58
+ @JRubyMethod(name = "available_permits")
59
+ public IRubyObject availablePermits(ThreadContext context) {
60
+ return getRuntime().newFixnum(this.semaphore.availablePermits());
61
+ }
62
+
63
+ @JRubyMethod(name = "drain_permits")
64
+ public IRubyObject drainPermits(ThreadContext context) {
65
+ return getRuntime().newFixnum(this.semaphore.drainPermits());
66
+ }
67
+
68
+ @JRubyMethod(name = "try_acquire")
69
+ public IRubyObject tryAcquire(ThreadContext context, final Block block) throws InterruptedException {
70
+ int permitsInt = 1;
71
+ boolean acquired = semaphore.tryAcquire(permitsInt);
72
+
73
+ return triedAcquire(context, permitsInt, acquired, block);
74
+ }
75
+
76
+ @JRubyMethod(name = "try_acquire")
77
+ public IRubyObject tryAcquire(ThreadContext context, IRubyObject permits, final Block block) throws InterruptedException {
78
+ int permitsInt = rubyFixnumToPositiveInt(permits, "permits");
79
+ boolean acquired = semaphore.tryAcquire(permitsInt);
80
+
81
+ return triedAcquire(context, permitsInt, acquired, block);
82
+ }
83
+
84
+ @JRubyMethod(name = "try_acquire")
85
+ public IRubyObject tryAcquire(ThreadContext context, IRubyObject permits, IRubyObject timeout, final Block block) throws InterruptedException {
86
+ int permitsInt = rubyFixnumToPositiveInt(permits, "permits");
87
+ boolean acquired = semaphore.tryAcquire(
88
+ permitsInt,
89
+ rubyNumericToLong(timeout, "timeout"),
90
+ java.util.concurrent.TimeUnit.SECONDS
91
+ );
92
+
93
+ return triedAcquire(context, permitsInt, acquired, block);
94
+ }
95
+
96
+ @JRubyMethod
97
+ public IRubyObject release(ThreadContext context) {
98
+ this.semaphore.release(1);
99
+ return getRuntime().newBoolean(true);
100
+ }
101
+
102
+ @JRubyMethod
103
+ public IRubyObject release(ThreadContext context, IRubyObject permits) {
104
+ this.semaphore.release(rubyFixnumToPositiveInt(permits, "permits"));
105
+ return getRuntime().newBoolean(true);
106
+ }
107
+
108
+ @JRubyMethod(name = "reduce_permits")
109
+ public IRubyObject reducePermits(ThreadContext context, IRubyObject reduction) throws InterruptedException {
110
+ this.semaphore.publicReducePermits(rubyFixnumToNonNegativeInt(reduction, "reduction"));
111
+ return context.nil;
112
+ }
113
+
114
+ private IRubyObject acquire(ThreadContext context, int permits, final Block block) throws InterruptedException {
115
+ this.semaphore.acquire(permits);
116
+
117
+ if (!block.isGiven()) return context.nil;
118
+
119
+ try {
120
+ return block.yieldSpecific(context);
121
+ } finally {
122
+ this.semaphore.release(permits);
123
+ }
124
+ }
125
+
126
+ private IRubyObject triedAcquire(ThreadContext context, int permits, boolean acquired, final Block block) {
127
+ if (!block.isGiven()) return getRuntime().newBoolean(acquired);
128
+ if (!acquired) return context.nil;
129
+
130
+ try {
131
+ return block.yieldSpecific(context);
132
+ } finally {
133
+ this.semaphore.release(permits);
134
+ }
135
+ }
136
+
137
+ private int rubyFixnumInt(IRubyObject value, String paramName) {
138
+ if (value instanceof RubyFixnum) {
139
+ RubyFixnum fixNum = (RubyFixnum) value;
140
+ return (int) fixNum.getLongValue();
141
+ } else {
142
+ throw getRuntime().newArgumentError(paramName + " must be integer");
143
+ }
144
+ }
145
+
146
+ private int rubyFixnumToNonNegativeInt(IRubyObject value, String paramName) {
147
+ if (value instanceof RubyFixnum && ((RubyFixnum) value).getLongValue() >= 0) {
148
+ RubyFixnum fixNum = (RubyFixnum) value;
149
+ return (int) fixNum.getLongValue();
150
+ } else {
151
+ throw getRuntime().newArgumentError(paramName + " must be a non-negative integer");
152
+ }
153
+ }
154
+
155
+ private int rubyFixnumToPositiveInt(IRubyObject value, String paramName) {
156
+ if (value instanceof RubyFixnum && ((RubyFixnum) value).getLongValue() > 0) {
157
+ RubyFixnum fixNum = (RubyFixnum) value;
158
+ return (int) fixNum.getLongValue();
159
+ } else {
160
+ throw getRuntime().newArgumentError(paramName + " must be an integer greater than zero");
161
+ }
162
+ }
163
+
164
+ private long rubyNumericToLong(IRubyObject value, String paramName) {
165
+ if (value instanceof RubyNumeric && ((RubyNumeric) value).getDoubleValue() > 0) {
166
+ RubyNumeric fixNum = (RubyNumeric) value;
167
+ return fixNum.getLongValue();
168
+ } else {
169
+ throw getRuntime().newArgumentError(paramName + " must be a float greater than zero");
170
+ }
171
+ }
172
+
173
+ class JRubySemaphore extends Semaphore {
174
+
175
+ public JRubySemaphore(int permits) {
176
+ super(permits);
177
+ }
178
+
179
+ public JRubySemaphore(int permits, boolean value) {
180
+ super(permits, value);
181
+ }
182
+
183
+ public void publicReducePermits(int i) {
184
+ reducePermits(i);
185
+ }
186
+
187
+ }
188
+ }
189
+ }