fastjsonFeatures behind JsonProvider

Author: wenshao

src/main/java/com/github/fabienrenaud/jjb/data/JsonSource.java

@@ -1,5 +1,10 @@
 package com.github.fabienrenaud.jjb.data;
 
+import com.alibaba.fastjson2.JSONFactory;
+import com.alibaba.fastjson2.JSONReader;
+import com.alibaba.fastjson2.JSONWriter;
+import com.alibaba.fastjson2.reader.ObjectReaderProvider;
+import com.alibaba.fastjson2.writer.ObjectWriterProvider;
 import com.github.fabienrenaud.jjb.RandomUtils;
 import com.github.fabienrenaud.jjb.data.gen.DataGenerator;
 import com.github.fabienrenaud.jjb.provider.JsonProvider;
@@ -29,6 +34,8 @@ public abstract class JsonSource<T> {
     private final StreamSerializer<T> streamSerializer;
     private final StreamDeserializer<T> streamDeserializer;
 
+    private final FastjsonProvider fastjsonFeatures;
+
     JsonSource(int quantity, int individualSize, JsonProvider provider, DataGenerator<T> dataGenerator, StreamSerializer<T> streamSerializer, StreamDeserializer<T> streamDeserializer) {
         this.provider = provider;
 
@@ -49,6 +56,24 @@ public abstract class JsonSource<T> {
             }
             return arr;
         });
+
+        ObjectWriterProvider featuresWriterProvider = new ObjectWriterProvider();
+        featuresWriterProvider.setDisableAutoType(true);
+        featuresWriterProvider.setDisableArrayMapping(true);
+        featuresWriterProvider.setDisableJSONB(true);
+        featuresWriterProvider.setDisableReferenceDetect(true);
+        JSONFactory.createWriteContext(featuresWriterProvider);
+
+        ObjectReaderProvider featuresReaderProvider = new ObjectReaderProvider();
+        featuresReaderProvider.setDisableArrayMapping(true);
+        featuresReaderProvider.setDisableAutoType(true);
+        featuresReaderProvider.setDisableJSONB(true);
+        featuresReaderProvider.setDisableReferenceDetect(true);
+        featuresReaderProvider.setDisableSmartMatch(true);
+
+        fastjsonFeatures = new FastjsonProvider(
+                JSONFactory.createReadContext(featuresReaderProvider),
+                JSONFactory.createWriteContext(featuresWriterProvider));
     }
 
     private void populateFields(int quantity, int individualSize) {
@@ -124,4 +149,10 @@ public StreamDeserializer<T> streamDeserializer() {
     private int index(int bound) {
         return bound == 1 ? 0 : RandomUtils.nextInt(bound);
     }
+
+    public FastjsonProvider fastjsonFeatures() {
+        return fastjsonFeatures;
+    }
+
+    public record FastjsonProvider (JSONReader.Context readerContext, JSONWriter.Context writerContext) {}
 }

src/main/java/com/github/fabienrenaud/jjb/databind/Deserialization.java

@@ -1,8 +1,6 @@
 package com.github.fabienrenaud.jjb.databind;
 
 import com.alibaba.fastjson2.JSON;
-import com.alibaba.fastjson2.JSONReader;
-import com.alibaba.fastjson2.reader.ObjectReaderProvider;
 import com.bluelinelabs.logansquare.LoganSquare;
 import com.github.fabienrenaud.jjb.JsonBench;
 import com.github.fabienrenaud.jjb.data.JsonSource;
@@ -17,17 +15,6 @@
  * @author Fabien Renaud
  */
 public class Deserialization extends JsonBench {
-    static final ObjectReaderProvider featuresProvider = new ObjectReaderProvider();
-    static final JSONReader.Context featuresContext;
-    static {
-        featuresProvider.setDisableArrayMapping(true);
-        featuresProvider.setDisableAutoType(true);
-        featuresProvider.setDisableJSONB(true);
-        featuresProvider.setDisableReferenceDetect(true);
-        featuresProvider.setDisableSmartMatch(true);
-        featuresContext = new JSONReader.Context(featuresProvider);
-    }
-
     public JsonSource JSON_SOURCE() {
         return CLI_JSON_SOURCE;
     }
@@ -77,7 +64,7 @@ public Object fastjson() {
     @Benchmark
     @Override
     public Object fastjson_features() {
-        return JSON.parseObject(JSON_SOURCE().nextByteArray(), JSON_SOURCE().pojoType(), featuresContext);
+        return JSON.parseObject(JSON_SOURCE().nextByteArray(), JSON_SOURCE().pojoType(), JSON_SOURCE().fastjsonFeatures().readerContext());
     }
 
     @Benchmark

src/main/java/com/github/fabienrenaud/jjb/databind/Serialization.java

@@ -1,9 +1,6 @@
 package com.github.fabienrenaud.jjb.databind;
 
 import com.alibaba.fastjson2.JSON;
-import com.alibaba.fastjson2.JSONFactory;
-import com.alibaba.fastjson2.JSONWriter;
-import com.alibaba.fastjson2.writer.ObjectWriterProvider;
 import com.bluelinelabs.logansquare.LoganSquare;
 import com.github.fabienrenaud.jjb.JsonBench;
 import com.github.fabienrenaud.jjb.JsonUtils;
@@ -16,16 +13,6 @@
 import java.io.ByteArrayOutputStream;
 
 public class Serialization extends JsonBench {
-    static final ObjectWriterProvider featuresProvider = new ObjectWriterProvider();
-    static final JSONWriter.Context featuresContext;
-    static {
-        featuresProvider.setDisableAutoType(true);
-        featuresProvider.setDisableArrayMapping(true);
-        featuresProvider.setDisableJSONB(true);
-        featuresProvider.setDisableReferenceDetect(true);
-        featuresContext = JSONFactory.createWriteContext(featuresProvider);
-    }
-
     public JsonSource JSON_SOURCE() {
         return CLI_JSON_SOURCE;
     }
@@ -90,7 +77,7 @@ public Object fastjson() throws Exception {
     @Override
     public Object fastjson_features() throws Exception {
         ByteArrayOutputStream baos = JsonUtils.byteArrayOutputStream();
-        JSON.writeTo(baos, JSON_SOURCE().nextPojo(), featuresContext);
+        JSON.writeTo(baos, JSON_SOURCE().nextPojo(), JSON_SOURCE().fastjsonFeatures().writerContext());
         return baos;
     }