Merge pull request #38 from SentryMan/recycle

Use JVM 19 Buffer Recycling

Author: rbygrave

jsonb/src/main/java/io/avaje/jsonb/stream/JsonStream.java

@@ -112,12 +112,8 @@ public JsonReader reader(Reader reader) {
 
   @Override
   public JsonReader reader(InputStream inputStream) {
-    try {
-      JsonParser parser = Recycle.parser(inputStream);
-      return new JsonReadAdapter(parser, failOnUnknown);
-    } catch (IOException e) {
-      throw new UncheckedIOException(e);
-    }
+    JsonParser parser = Recycle.parser(inputStream);
+    return new JsonReadAdapter(parser, failOnUnknown);
   }
 
   @Override

jsonb/src/main/java/io/avaje/jsonb/stream/Recycle.java

@@ -1,38 +1,58 @@
 package io.avaje.jsonb.stream;
 
-import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 
 final class Recycle {
+  private Recycle() {}
+
+  private static boolean jvmRecycle;
+  private static ThreadLocal<JParser> read;
+  private static ThreadLocal<JGenerator> managed;
+
+  static {
+    if (Float.parseFloat(System.getProperty("java.specification.version")) >= 19
+        && !Boolean.getBoolean("jsonb.UseTLBuffers")) {
+      jvmRecycle = true;
+    } else {
+      managed = ThreadLocal.withInitial(Recycle::getGenerator);
+      read = ThreadLocal.withInitial(Recycle::getParser);
+    }
+  }
 
-  static ThreadLocal<JGenerator> managed = ThreadLocal.withInitial(() -> new JGenerator(4096));
-
-  static ThreadLocal<JParser> read = ThreadLocal.withInitial(() -> {
-    char[] ch = new char[4096];
-    byte[] by = new byte[4096];
-    return new JParser(ch, by, 0, JParser.ErrorInfo.MINIMAL, JParser.DoublePrecision.DEFAULT, JParser.UnknownNumberParsing.BIGDECIMAL, 100, 50_000);
-  });
-
-  /**
-   * Return a recycled generator with the given target OutputStream.
-   */
+  /** Return a recycled generator with the given target OutputStream. */
   static JsonGenerator generator(OutputStream target) {
-    return managed.get().prepare(target);
+    return (jvmRecycle ? getGenerator() : managed.get()).prepare(target);
   }
 
-  /**
-   * Return a recycled generator with expected "to String" result.
-   */
+  /** Return a recycled generator with expected "to String" result. */
   static JsonGenerator generator() {
-    return managed.get().prepare(null);
+    return (jvmRecycle ? getGenerator() : managed.get()).prepare(null);
   }
 
   static JsonParser parser(byte[] bytes) {
-    return read.get().process(bytes, bytes.length);
+    return (jvmRecycle ? getParser() : read.get()).process(bytes, bytes.length);
+  }
+
+  static JsonParser parser(InputStream in) {
+    return (jvmRecycle ? getParser() : read.get()).process(in);
+  }
+
+  static JGenerator getGenerator() {
+    return new JGenerator(4096);
   }
 
-  static JsonParser parser(InputStream in) throws IOException {
-    return read.get().process(in);
+  static JParser getParser() {
+    final char[] ch = new char[4096];
+    final byte[] by = new byte[4096];
+    return new JParser(
+        ch,
+        by,
+        0,
+        JParser.ErrorInfo.MINIMAL,
+        JParser.DoublePrecision.DEFAULT,
+        JParser.UnknownNumberParsing.BIGDECIMAL,
+        100,
+        50_000);
   }
 }