Move JSONObjectWithLegibleException to a separate file.

1 files changed, 86 insertions, 0 deletions

diff --git a/container-core/src/main/java/com/yahoo/container/jdisc/state/JSONObjectWithLegibleException.java b/container-core/src/main/java/com/yahoo/container/jdisc/state/JSONObjectWithLegibleException.java
new file mode 100644
index 00000000000..dc1bfb89197
--- /dev/null
+++ b/container-core/src/main/java/com/yahoo/container/jdisc/state/JSONObjectWithLegibleException.java
@@ -0,0 +1,86 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.jdisc.state;
+
+import org.json.JSONException;
+import org.json.JSONObject;
+
+import java.util.Collection;
+import java.util.Map;
+
+/**
+ * A JSONObject that wraps the checked JSONException in a RuntimeException with a legible error message.
+ *
+ * @author gjoranv
+ */
+class JSONObjectWithLegibleException extends JSONObject {
+    @Override
+    public JSONObject put(String s, boolean b) {
+        try {
+            return super.put(s, b);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, b, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, double v) {
+        try {
+            Double guardedVal = (((Double) v).isNaN() || ((Double) v).isInfinite()) ?
+                    0.0 : v;
+            return super.put(s, guardedVal);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, v, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, int i) {
+        try {
+            return super.put(s, i);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, i, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, long l) {
+        try {
+            return super.put(s, l);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, l, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, Collection collection) {
+        try {
+            return super.put(s, collection);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, collection, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, Map map) {
+        try {
+            return super.put(s, map);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, map, e), e);
+        }
+    }
+
+    @Override
+    public JSONObject put(String s, Object o) {
+        try {
+            return super.put(s, o);
+        } catch (JSONException e) {
+            throw new RuntimeException(getErrorMessage(s, o, e), e);
+        }
+    }
+
+    private String getErrorMessage(String key, Object value, JSONException e) {
+        return "Trying to add invalid JSON object with key '" + key +
+                "' and value '" + value +
+                "' - " + e.getMessage();
+    }
+}