RiskToVersion2.java

/**
 * The contents of this file are subject to the license and copyright
 * detailed in the LICENSE file at the root of the source
 * tree and available online at
 *
 * https://github.com/keeps/roda
 */
package org.roda.core.migration.model;

import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.io.IOUtils;
import org.roda.core.common.iterables.CloseableIterable;
import org.roda.core.data.exceptions.RODAException;
import org.roda.core.data.utils.JsonUtils;
import org.roda.core.data.v2.risks.Risk;
import org.roda.core.migration.MigrationAction;
import org.roda.core.model.utils.ModelUtils;
import org.roda.core.storage.Binary;
import org.roda.core.storage.Resource;
import org.roda.core.storage.StorageService;
import org.roda.core.storage.StringContentPayload;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;

public class RiskToVersion2 implements MigrationAction<Risk> {

  private static final Logger LOGGER = LoggerFactory.getLogger(RiskToVersion2.class);

  private static final Map<String, String> MAPPING = new HashMap<>();

  static {
    MAPPING.put("posMitigationProbability", "postMitigationProbability");
    MAPPING.put("posMitigationImpact", "postMitigationImpact");
    MAPPING.put("posMitigationSeverity", "postMitigationSeverity");
    MAPPING.put("posMitigationSeverityLevel", "postMitigationSeverityLevel");
    MAPPING.put("posMitigationNotes", "postMitigationNotes");
  }

  @Override
  public void migrate(StorageService storage) throws RODAException {
    CloseableIterable<Resource> risks = null;

    try {
      risks = storage.listResourcesUnderDirectory(ModelUtils.getRiskContainerPath(), false);

      for (Resource resource : risks) {
        if (!resource.isDirectory() && resource instanceof Binary) {
          Binary binary = (Binary) resource;
          migrate(storage, binary);
        }
      }
    } finally {
      IOUtils.closeQuietly(risks);
    }
  }

  private void migrate(StorageService storage, Binary binary) {
    InputStream inputStream = null;
    try {
      inputStream = binary.getContent().createInputStream();
      JsonNode json = JsonUtils.parseJson(inputStream);
      if (json instanceof ObjectNode) {
        ObjectNode obj = (ObjectNode) json;
        obj = JsonUtils.refactor(obj, MAPPING);

        StringContentPayload payload = new StringContentPayload(JsonUtils.getJsonFromNode(obj));
        boolean asReference = false;
        boolean createIfNotExists = false;
        storage.updateBinaryContent(binary.getStoragePath(), payload, asReference, createIfNotExists);
      } else {
        LOGGER.error("Could not migrate risk {} because the JSON is not an object node", binary.getStoragePath());
      }

    } catch (IOException | RODAException e) {
      LOGGER.error("Could not migrate risk {}", binary.getStoragePath(), e);
    } finally {
      IOUtils.closeQuietly(inputStream);
    }

  }

  @Override
  public boolean isToVersionValid(int toVersion) {
    return toVersion == 2;
  }

}