Class StructType

java/tools/src/java/org/apache/orc/tools/json/StructType.java:31–140

Model structs.

 * Model structs.
 */
class StructType extends HiveType {
  private static final String COL_PREFIX = "_col";
  private static final Comparator<String> FIELD_COMPARATOR = (left, right) -> {
    if (left == null) {
      return right == null ? 0 : -1;
    } else if (right == null) {
      return 1;
    } else if (left.startsWith(COL_PREFIX)) {
      if (right.startsWith(COL_PREFIX)) {
        try {
          int leftInt = Integer.parseInt(left.substring(COL_PREFIX.length()));
          int rightInt = Integer.parseInt(right.substring(COL_PREFIX.length()));
          return Integer.compare(leftInt, rightInt);
        } catch (Exception e) {
          // fall back to the normal rules
        }
      } else {
        return 1;
      }
    } else if (right.startsWith(COL_PREFIX)) {
      return 1;
    }
    return left.compareTo(right);
  };

  final Map<String, HiveType> fields = new TreeMap<>(FIELD_COMPARATOR);

  StructType() {
    super(Kind.STRUCT);
  }

  @Override
  public String toString() {
    StringBuilder buf = new StringBuilder("struct<");
    boolean first = true;
    for (Map.Entry<String, HiveType> field : fields.entrySet()) {
      if (!first) {
        buf.append(',');
      } else {
        first = false;
      }
      buf.append(field.getKey());
      buf.append(':');
      buf.append(field.getValue().toString());
    }
    buf.append(">");
    return buf.toString();
  }

  public StructType addField(String name, HiveType fieldType) {
    fields.put(name, fieldType);
    return this;
  }

  @Override
  public boolean equals(Object other) {
    return super.equals(other) && fields.equals(((StructType) other).fields);
  }