Updated mcollective.init according to OSCI-658

[packages/precise/mcollective.git] / lib / mcollective / vendor / json / java / src / json / ext / RuntimeInfo.java

diff --git a/lib/mcollective/vendor/json/java/src/json/ext/RuntimeInfo.java b/lib/mcollective/vendor/json/java/src/json/ext/RuntimeInfo.java
new file mode 100644 (file)
index 0000000..5de5740
--- /dev/null
+++ b/lib/mcollective/vendor/json/java/src/json/ext/RuntimeInfo.java
@@ -0,0 +1,121 @@
+/*
+ * This code is copyrighted work by Daniel Luz <dev at mernen dot com>.
+ *
+ * Distributed under the Ruby and GPLv2 licenses; see COPYING and GPL files
+ * for details.
+ */
+package json.ext;
+
+import java.lang.ref.WeakReference;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.WeakHashMap;
+import org.jruby.Ruby;
+import org.jruby.RubyClass;
+import org.jruby.RubyEncoding;
+import org.jruby.RubyModule;
+import org.jruby.runtime.ThreadContext;
+import org.jruby.runtime.builtin.IRubyObject;
+
+
+final class RuntimeInfo {
+    // since the vast majority of cases runs just one runtime,
+    // we optimize for that
+    private static WeakReference<Ruby> runtime1 = new WeakReference<Ruby>(null);
+    private static RuntimeInfo info1;
+    // store remaining runtimes here (does not include runtime1)
+    private static Map<Ruby, RuntimeInfo> runtimes;
+
+    // these fields are filled by the service loaders
+    // Use WeakReferences so that RuntimeInfo doesn't indirectly hold a hard reference to
+    // the Ruby runtime object, which would cause memory leaks in the runtimes map above.
+    /** JSON */
+    WeakReference<RubyModule> jsonModule;
+    /** JSON::Ext::Generator::GeneratorMethods::String::Extend */
+    WeakReference<RubyModule> stringExtendModule;
+    /** JSON::Ext::Generator::State */
+    WeakReference<RubyClass> generatorStateClass;
+    /** JSON::SAFE_STATE_PROTOTYPE */
+    WeakReference<GeneratorState> safeStatePrototype;
+
+    final WeakReference<RubyEncoding> utf8;
+    final WeakReference<RubyEncoding> ascii8bit;
+    // other encodings
+    private final Map<String, WeakReference<RubyEncoding>> encodings;
+
+    private RuntimeInfo(Ruby runtime) {
+        RubyClass encodingClass = runtime.getEncoding();
+        if (encodingClass == null) { // 1.8 mode
+            utf8 = ascii8bit = null;
+            encodings = null;
+        } else {
+            ThreadContext context = runtime.getCurrentContext();
+
+            utf8 = new WeakReference<RubyEncoding>((RubyEncoding)RubyEncoding.find(context,
+                    encodingClass, runtime.newString("utf-8")));
+            ascii8bit = new WeakReference<RubyEncoding>((RubyEncoding)RubyEncoding.find(context,
+                    encodingClass, runtime.newString("ascii-8bit")));
+            encodings = new HashMap<String, WeakReference<RubyEncoding>>();
+        }
+    }
+
+    static RuntimeInfo initRuntime(Ruby runtime) {
+        synchronized (RuntimeInfo.class) {
+            if (runtime1.get() == runtime) {
+                return info1;
+            } else if (runtime1.get() == null) {
+                runtime1 = new WeakReference<Ruby>(runtime);
+                info1 = new RuntimeInfo(runtime);
+                return info1;
+            } else {
+                if (runtimes == null) {
+                    runtimes = new WeakHashMap<Ruby, RuntimeInfo>(1);
+                }
+                RuntimeInfo cache = runtimes.get(runtime);
+                if (cache == null) {
+                    cache = new RuntimeInfo(runtime);
+                    runtimes.put(runtime, cache);
+                }
+                return cache;
+            }
+        }
+    }
+
+    public static RuntimeInfo forRuntime(Ruby runtime) {
+        synchronized (RuntimeInfo.class) {
+            if (runtime1.get() == runtime) return info1;
+            RuntimeInfo cache = null;
+            if (runtimes != null) cache = runtimes.get(runtime);
+            assert cache != null : "Runtime given has not initialized JSON::Ext";
+            return cache;
+        }
+    }
+
+    public boolean encodingsSupported() {
+        return utf8 != null && utf8.get() != null;
+    }
+
+    public RubyEncoding getEncoding(ThreadContext context, String name) {
+        synchronized (encodings) {
+            WeakReference<RubyEncoding> encoding = encodings.get(name);
+            if (encoding == null) {
+                Ruby runtime = context.getRuntime();
+                encoding = new WeakReference<RubyEncoding>((RubyEncoding)RubyEncoding.find(context,
+                        runtime.getEncoding(), runtime.newString(name)));
+                encodings.put(name, encoding);
+            }
+            return encoding.get();
+        }
+    }
+
+    public GeneratorState getSafeStatePrototype(ThreadContext context) {
+        if (safeStatePrototype == null) {
+            IRubyObject value = jsonModule.get().getConstant("SAFE_STATE_PROTOTYPE");
+            if (!(value instanceof GeneratorState)) {
+                throw context.getRuntime().newTypeError(value, generatorStateClass.get());
+            }
+            safeStatePrototype = new WeakReference<GeneratorState>((GeneratorState)value);
+        }
+        return safeStatePrototype.get();
+    }
+}