Changeset 18754 in josm for trunk/src/org/openstreetmap/josm/io/GeoJSONWriter.java

Timestamp:

2023-06-14T00:22:08+02:00 (17 months ago)

Author:

taylor.smock

Message:

Significantly reduce allocations in GeoJSONWriter

When using josm validate -i MesaCountyCO.osm.gz, where Mesa County, CO is the
data inside Mesa County, Colorado, USA, this patch reduces overall runtime by
~75%. This is largely done by reducing the amount of resources it takes to write
errors to file. So GeoJSONWriter#write now takes ~99% fewer CPU cycles and ~97%
fewer memory allocations. Time spent in JVM specific methods (like GC) was
reduced by ~50%.

File:

• trunk/src/org/openstreetmap/josm/io/GeoJSONWriter.java (modified) (20 diffs)

trunk/src/org/openstreetmap/josm/io/GeoJSONWriter.java

@@ -27 +27 @@
 import jakarta.json.JsonValue;
 import jakarta.json.JsonWriter;
+import jakarta.json.spi.JsonProvider;
 import jakarta.json.stream.JsonGenerator;
 import jakarta.json.stream.JsonParser;
@@ -74 +75 @@
     private static final BooleanProperty SKIP_EMPTY_NODES = new BooleanProperty("geojson.export.skip-empty-nodes", true);
     private static final BooleanProperty UNTAGGED_CLOSED_IS_POLYGON = new BooleanProperty("geojson.export.untagged-closed-is-polygon", false);
+
+    /**
+     * Used to avoid many calls to {@link JsonProvider#provider} in {@link #getCoorArray(JsonArrayBuilder, EastNorth)}.
+     * For validating Mesa County, CO, this reduces CPU and memory usage of {@link #write()} by ~80%. By using this for
+     * other {@link Json} calls, {@link #write()} takes ~95% less resources than the original. And the entire process
+     * takes 1/4 of the time (38 minutes -&gt; <10 minutes).
+     * <p>
+     * For more details, see <a href="https://github.com/jakartaee/jsonp-api/issues/346">JSONP #346</a>.
+     */
+    protected static final JsonProvider JSON_PROVIDER = JsonProvider.provider();
     private static final Set<Way> processedMultipolygonWays = new HashSet<>();
-    private EnumSet<Options> options = EnumSet.noneOf(Options.class);
+    private final EnumSet<Options> options = EnumSet.noneOf(Options.class);
 
     /**
@@ -138 +149 @@
     public void write(boolean pretty, Writer writer) {
         Map<String, Object> config = Collections.singletonMap(JsonGenerator.PRETTY_PRINTING, pretty);
-        try (JsonWriter jsonWriter = Json.createWriterFactory(config).createWriter(writer)) {
-            JsonObjectBuilder object = Json.createObjectBuilder()
+        try (JsonWriter jsonWriter = JSON_PROVIDER.createWriterFactory(config).createWriter(writer)) {
+            JsonObjectBuilder object = JSON_PROVIDER.createObjectBuilder()
                     .add("type", "FeatureCollection")
                     .add("generator", "JOSM");
@@ -184 +195 @@
                 if (writeAsPolygon) {
                     geomObj.add("type", "Polygon");
-                    geomObj.add("coordinates", Json.createArrayBuilder().add(array));
+                    geomObj.add("coordinates", JSON_PROVIDER.createArrayBuilder().add(array));
                 } else {
                     geomObj.add("type", "LineString");
@@ -223 +234 @@
          */
         private void visitMultiGeometry(final Relation r) {
-            final JsonArrayBuilder jsonArrayBuilder = Json.createArrayBuilder();
+            final JsonArrayBuilder jsonArrayBuilder = JSON_PROVIDER.createArrayBuilder();
             r.getMemberPrimitives().stream().filter(p -> !(p instanceof Relation))
                     .map(p -> {
-                        final JsonObjectBuilder tempGeomObj = Json.createObjectBuilder();
+                        final JsonObjectBuilder tempGeomObj = JSON_PROVIDER.createObjectBuilder();
                         p.accept(new GeometryPrimitiveVisitor(tempGeomObj));
                         return tempGeomObj.build();
@@ -239 +250 @@
          */
         private void visitMultiPoints(final Relation r) {
-            final JsonArrayBuilder multiPoint = Json.createArrayBuilder();
+            final JsonArrayBuilder multiPoint = JSON_PROVIDER.createArrayBuilder();
             r.getMembers().stream().map(RelationMember::getMember).filter(Node.class::isInstance).map(Node.class::cast)
                     .map(latLon -> getCoorArray(null, latLon))
@@ -252 +263 @@
          */
         private void visitMultiLineString(final Relation r) {
-            final JsonArrayBuilder multiLine = Json.createArrayBuilder();
+            final JsonArrayBuilder multiLine = JSON_PROVIDER.createArrayBuilder();
             r.getMembers().stream().map(RelationMember::getMember).filter(Way.class::isInstance).map(Way.class::cast)
                     .map(Way::getNodes).map(p -> {
@@ -274 +285 @@
                 final Pair<List<MultipolygonBuilder.JoinedPolygon>, List<MultipolygonBuilder.JoinedPolygon>> mp =
                         MultipolygonBuilder.joinWays(r);
-                final JsonArrayBuilder polygon = Json.createArrayBuilder();
+                final JsonArrayBuilder polygon = JSON_PROVIDER.createArrayBuilder();
                 // Peek would theoretically be better for these two streams, but SonarLint doesn't like it.
                 // java:S3864: "Stream.peek" should be used with caution
@@ -303 +314 @@
                         })
                         .forEach(polygon::add);
-                final JsonArrayBuilder multiPolygon = Json.createArrayBuilder().add(polygon);
+                final JsonArrayBuilder multiPolygon = JSON_PROVIDER.createArrayBuilder().add(polygon);
                 geomObj.add("type", "MultiPolygon");
                 geomObj.add("coordinates", multiPolygon);
@@ -310 +321 @@
 
         private JsonArrayBuilder getCoorsArray(Iterable<Node> nodes) {
-            final JsonArrayBuilder builder = Json.createArrayBuilder();
+            final JsonArrayBuilder builder = JSON_PROVIDER.createArrayBuilder();
             for (Node n : nodes) {
                 if (n.isLatLonKnown()) {
@@ -325 +336 @@
 
     private static JsonArrayBuilder getCoorArray(JsonArrayBuilder builder, EastNorth c) {
-        return (builder != null ? builder : Json.createArrayBuilder())
+        return (builder != null ? builder : JSON_PROVIDER.createArrayBuilder())
                 .add(BigDecimal.valueOf(c.getX()).setScale(11, RoundingMode.HALF_UP))
                 .add(BigDecimal.valueOf(c.getY()).setScale(11, RoundingMode.HALF_UP));
@@ -337 +348 @@
 
         // Properties
-        final JsonObjectBuilder propObj = Json.createObjectBuilder();
+        final JsonObjectBuilder propObj = JSON_PROVIDER.createObjectBuilder();
         for (Map.Entry<String, String> t : p.getKeys().entrySet()) {
             // If writing OSM information, follow Overpass syntax (escape `@` with another `@`)
@@ -357 +368 @@
             }
             if (options.contains(Options.WRITE_OSM_INFORMATION) && p.getReferrers(true).stream().anyMatch(Relation.class::isInstance)) {
-                final JsonArrayBuilder jsonArrayBuilder = Json.createArrayBuilder();
+                final JsonArrayBuilder jsonArrayBuilder = JSON_PROVIDER.createArrayBuilder();
                 for (Relation relation : Utils.filteredCollection(p.getReferrers(), Relation.class)) {
-                    final JsonObjectBuilder relationObject = Json.createObjectBuilder();
+                    final JsonObjectBuilder relationObject = JSON_PROVIDER.createObjectBuilder();
                     relationObject.add("rel", relation.getId());
                     Collection<RelationMember> members = relation.getMembersFor(Collections.singleton(p));
@@ -365 +376 @@
                     relationObject.add("role",
                             members.stream().map(RelationMember::getRole).collect(Collectors.joining(";")));
-                    final JsonObjectBuilder relationKeys = Json.createObjectBuilder();
+                    final JsonObjectBuilder relationKeys = JSON_PROVIDER.createObjectBuilder();
                     // Uncertain if the @relation reltags need to be @ escaped. I don't think so, as example output
                     // didn't have any metadata in it.
@@ -379 +390 @@
 
         // Geometry
-        final JsonObjectBuilder geomObj = Json.createObjectBuilder();
+        final JsonObjectBuilder geomObj = JSON_PROVIDER.createObjectBuilder();
         p.accept(new GeometryPrimitiveVisitor(geomObj));
         final JsonObject geom = geomObj.build();
@@ -385 +396 @@
         if (!geom.isEmpty()) {
             // Build primitive JSON object
-            array.add(Json.createObjectBuilder()
+            array.add(JSON_PROVIDER.createObjectBuilder()
                     .add("type", "Feature")
                     .add("properties", prop.isEmpty() ? JsonValue.NULL : prop)
@@ -394 +405 @@
     private static JsonValue convertValueToJson(String value) {
         if (value.startsWith(JSON_VALUE_START_MARKER) && value.endsWith(JSON_VALUE_END_MARKER)) {
-            try (JsonParser parser = Json.createParser(new StringReader(value))) {
+            try (JsonParser parser = JSON_PROVIDER.createParser(new StringReader(value))) {
                 if (parser.hasNext() && parser.next() != null) {
                     return parser.getValue();
@@ -402 +413 @@
             }
         }
-        return Json.createValue(value);
+        return JSON_PROVIDER.createValue(value);
     }
 
@@ -420 +431 @@
     protected void appendBounds(Bounds b, JsonObjectBuilder object) {
         if (b != null) {
-            JsonArrayBuilder builder = Json.createArrayBuilder();
+            JsonArrayBuilder builder = JSON_PROVIDER.createArrayBuilder();
             getCoorArray(builder, b.getMin());
             getCoorArray(builder, b.getMax());
@@ -428 +439 @@
 
     protected void appendLayerFeatures(DataSet ds, JsonObjectBuilder object) {
-        JsonArrayBuilder array = Json.createArrayBuilder();
+        JsonArrayBuilder array = JSON_PROVIDER.createArrayBuilder();
         if (ds != null) {
             processedMultipolygonWays.clear();