com.arextest.diff.compare.feature.IndexSelectorFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of compare-sdk Show documentation
There is a newer version: 0.2.15
package com.arextest.diff.compare.feature;

import com.arextest.diff.model.compare.CompareContext;
import com.arextest.diff.model.log.NodeEntity;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.logging.Logger;

public class IndexSelectorFactory {

  private static Logger LOGGER = Logger.getLogger(IndexSelectorFactory.class.getName());

  public static IndexSelector getIndexSelector(List currentNodeLeft,
      List currentNodeRight, CompareContext compareContext) {
    Map indexKeysLeft = compareContext.listIndexKeysLeft.get(currentNodeLeft);
    Map indexKeysRight = compareContext.listIndexKeysRight.get(currentNodeRight);

    if (indexKeysLeft != null || indexKeysRight != null) {
      return new ListKeyIndexSelector(indexKeysLeft, indexKeysRight, compareContext);
    } else if (isPrimitiveArrayAndEqual(compareContext)) {
      LOGGER.info("use PrimitiveArrayIndexSelector");
      return new PrimitiveArrayIndexSelector(compareContext);
    } else {
      return new GeneralIndexSelector();
    }
  }

  private static boolean isPrimitiveArrayAndEqual(CompareContext compareContext) {
    Object currentBaseObj = compareContext.currentBaseObj;
    Object currentTestObj = compareContext.currentTestObj;

    if (!(currentBaseObj instanceof ArrayNode) || !(currentTestObj instanceof ArrayNode)) {
      return false;
    }

    ArrayNode baseArray = (ArrayNode) currentBaseObj;
    ArrayNode testArray = (ArrayNode) currentTestObj;

    if (baseArray.size() != testArray.size()) {
      return false;
    }

    if ((!baseArray.isEmpty() && !(baseArray.get(0).isValueNode())) ||
        (!testArray.isEmpty() && !(testArray.get(0).isValueNode()))) {
      return false;
    }

    List baseList = new ArrayList<>();
    List testList = new ArrayList<>();

    for (JsonNode jsonNode : baseArray) {
      if (!jsonNode.isValueNode()) {
        return false;
      }
      baseList.add(jsonNode.asText());
    }

    for (JsonNode jsonNode : testArray) {
      if (!jsonNode.isValueNode()) {
        return false;
      }
      testList.add(jsonNode.asText());
    }

    // If it can be directly compared before sorting,
    // do not use PrimitiveArrayIndexSelector.
    if (Objects.equals(baseList, testList)) {
      return false;
    }

    Collections.sort(baseList);
    Collections.sort(testList);
    return Objects.equals(baseList, testList);
  }
}