Temporary fix for performance regression after powsybl upgrade

Author: antoinebhs

src/main/java/com/powsybl/iidm/serde/NetworkDeserializerContext.java

@@ -0,0 +1,107 @@
+/**
+ * Copyright (c) 2016, RTE (http://www.rte-france.com)
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ * SPDX-License-Identifier: MPL-2.0
+ */
+package com.powsybl.iidm.serde;
+
+import com.powsybl.commons.extensions.ExtensionSerDe;
+import com.powsybl.commons.io.DeserializerContext;
+import com.powsybl.commons.io.TreeDataReader;
+import com.powsybl.commons.report.ReportNode;
+import com.powsybl.iidm.network.Network;
+import com.powsybl.iidm.network.ValidationLevel;
+import com.powsybl.iidm.network.util.Networks;
+import com.powsybl.iidm.serde.anonymizer.Anonymizer;
+
+import java.util.*;
+
+import static com.powsybl.iidm.serde.IidmSerDeConstants.CURRENT_IIDM_VERSION;
+
+/**
+ * @author Geoffroy Jamgotchian {@literal <geoffroy.jamgotchian at rte-france.com>}
+ */
+//FIXME: Fixes performance regression (https://github.com/powsybl/powsybl-core/pull/3469). This file has to be removed on Powsybl upgrade 6.7.2 or 6.8.0
+public class NetworkDeserializerContext extends AbstractNetworkSerDeContext<ImportOptions> implements DeserializerContext {
+
+    private final TreeDataReader reader;
+
+    private final List<DeserializationEndTask> endTasks = new ArrayList<>();
+    private final ImportOptions options;
+
+    private final Map<String, String> extensionVersions;
+
+    private ValidationLevel networkValidationLevel;
+
+    private Set<String> ignoredEquipments = new HashSet<>();
+    private EnumSet<DeserializationEndTask.Step> alreadyProcessedEndTasksSteps = EnumSet.noneOf(DeserializationEndTask.Step.class);
+
+    public NetworkDeserializerContext(Anonymizer anonymizer, TreeDataReader reader) {
+        this(anonymizer, reader, new ImportOptions(), CURRENT_IIDM_VERSION, Collections.emptyMap());
+    }
+
+    public NetworkDeserializerContext(Anonymizer anonymizer, TreeDataReader reader, ImportOptions options, IidmVersion version,
+                                      Map<String, String> extensionVersions) {
+        super(anonymizer, version);
+        this.reader = Objects.requireNonNull(reader);
+        this.options = Objects.requireNonNull(options);
+        this.extensionVersions = extensionVersions;
+    }
+
+    @Override
+    public TreeDataReader getReader() {
+        return reader;
+    }
+
+    public void addEndTask(DeserializationEndTask.Step step, Runnable task) {
+        endTasks.add(new DeserializationEndTask(step, task));
+    }
+
+    public void executeEndTasks(Network network, DeserializationEndTask.Step step, ReportNode reportNode) {
+        if (alreadyProcessedEndTasksSteps.contains(step)) {
+            // Skip if step was already processed
+            return;
+        }
+        alreadyProcessedEndTasksSteps.add(step);
+        Networks.executeWithReportNode(network, reportNode, () -> endTasks.stream()
+                .filter(t -> t.getStep() == step
+                        || step == DeserializationEndTask.Step.AFTER_EXTENSIONS &&
+                        t.getStep() == DeserializationEndTask.Step.BEFORE_EXTENSIONS && !t.isProcessed()) // If no "extensions" tag was found, BEFORE_EXTENSIONS tasks were not processed.
+                .forEach(t -> {
+                    t.setProcessed(true);
+                    t.getTask().run();
+                }));
+    }
+
+    @Override
+    public ImportOptions getOptions() {
+        return options;
+    }
+
+    public boolean containsExtensionVersion(String extensionName, String version) {
+        return version != null && version.equals(extensionVersions.get(extensionName));
+    }
+
+    public Optional<String> getExtensionVersion(ExtensionSerDe<?, ?> extensionSerDe) {
+        return Optional.ofNullable(extensionVersions.get(extensionSerDe.getExtensionName()));
+    }
+
+    public NetworkDeserializerContext setNetworkValidationLevel(ValidationLevel validationLevel) {
+        this.networkValidationLevel = validationLevel;
+        return this;
+    }
+
+    public ValidationLevel getNetworkValidationLevel() {
+        return this.networkValidationLevel;
+    }
+
+    public void addIgnoredEquipment(String equipment) {
+        ignoredEquipments.add(equipment);
+    }
+
+    public boolean isIgnoredEquipment(String equipment) {
+        return ignoredEquipments.contains(equipment);
+    }
+}