Fix: Using JrSimpleTreeExtension to Provide Feature Context

Author: Shounaks

jr-stree/src/main/java/com/fasterxml/jackson/jr/stree/JacksonJrsTreeCodec.java

@@ -18,7 +18,8 @@ public class JacksonJrsTreeCodec extends TreeCodec
 
     // @since 2.17
     protected boolean _failOnDuplicateKeys;
-    
+    protected boolean _useBigDecimalForDouble;
+
     public JacksonJrsTreeCodec() {
         this(null);
     }
@@ -31,7 +32,11 @@ public JacksonJrsTreeCodec(ObjectCodec codec) {
     public void setFailOnDuplicateKeys(boolean state) {
         _failOnDuplicateKeys = state;
     }
-    
+
+    public void setUseBigDecimalForDouble(boolean state) {
+        _useBigDecimalForDouble = state;
+    }
+
     @SuppressWarnings("unchecked")
     @Override
     public <T extends TreeNode> T readTree(JsonParser p) throws IOException {
@@ -50,6 +55,9 @@ private JrsValue nodeFrom(JsonParser p) throws IOException
             return JrsBoolean.FALSE;
         case JsonTokenId.ID_NUMBER_INT:
         case JsonTokenId.ID_NUMBER_FLOAT:
+            if (_useBigDecimalForDouble) {
+                return new JrsNumber(p.getDecimalValue());
+            }
             return new JrsNumber(p.getNumberValue());
         case JsonTokenId.ID_STRING:
             return new JrsString(p.getText());

jr-stree/src/main/java/com/fasterxml/jackson/jr/stree/JrSimpleTreeExtension.java

@@ -28,6 +28,7 @@ public JrSimpleTreeExtension(JacksonJrsTreeCodec tc) {
     @Override
     protected void register(ExtensionContext ctxt) {
         _codec.setFailOnDuplicateKeys(ctxt.isEnabled(JSON.Feature.FAIL_ON_DUPLICATE_MAP_KEYS));
+        _codec.setUseBigDecimalForDouble(ctxt.isEnabled(JSON.Feature.USE_BIG_DECIMAL_FOR_FLOATS));
         ctxt.setTreeCodec(_codec);
     }
 }

jr-stree/src/test/java/com/fasterxml/jackson/jr/stree/failing/ReadAsBigDecimal90Test.java renamed to jr-stree/src/test/java/com/fasterxml/jackson/jr/stree/ReadAsBigDecimal90Test.java

@@ -1,20 +1,18 @@
-package com.fasterxml.jackson.jr.stree.failing;
-
-import java.math.BigDecimal;
+package com.fasterxml.jackson.jr.stree;
 
 import com.fasterxml.jackson.core.TreeNode;
+import com.fasterxml.jackson.jr.ob.JSON;
 
-import com.fasterxml.jackson.jr.ob.*;
-import com.fasterxml.jackson.jr.stree.*;
+import java.math.BigDecimal;
 
 public class ReadAsBigDecimal90Test extends JacksonJrTreeTestBase
 {
     // [jackson-jr#90]
     public void testReadAsBigDecimal() throws Exception
     {
         JSON json = JSON.builder()
-                .treeCodec(new JacksonJrsTreeCodec())
                 .enable(JSON.Feature.USE_BIG_DECIMAL_FOR_FLOATS)
+                .register(new JrSimpleTreeExtension())
                 .build();
 
         String input = "[1.1]";
@@ -27,4 +25,21 @@ public void testReadAsBigDecimal() throws Exception
         assertEquals(BigDecimal.class,
                 ((JrsNumber) elemNode).getValue().getClass());
     }
+
+    public void testDefaultBehaviourWithBigDecimalFlag() throws Exception
+    {
+        JSON json = JSON.builder()
+                .register(new JrSimpleTreeExtension())
+                .build();
+
+        String input = "[1.1]";
+
+        TreeNode node = json.treeFrom(input);
+        TreeNode elemNode = node.get(0);
+
+        assertTrue(elemNode.isValueNode());
+        assertTrue(elemNode instanceof JrsNumber);
+        assertEquals(Double.class,
+                ((JrsNumber) elemNode).getValue().getClass());
+    }
 }