org.pageseeder.flint.catalog.Catalog Maven / Gradle / Ivy

Go to download
package org.pageseeder.flint.catalog;

import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.*;

import org.pageseeder.flint.indexing.FlintField;
import org.pageseeder.flint.indexing.FlintField.DocValuesType;
import org.pageseeder.flint.indexing.FlintField.IndexOptions;
import org.pageseeder.flint.indexing.FlintField.NumericType;
import org.pageseeder.flint.indexing.FlintField.Resolution;
import org.pageseeder.xmlwriter.XMLWritable;
import org.pageseeder.xmlwriter.XMLWriter;

public class Catalog implements XMLWritable {

  private final HashMap _fields = new HashMap<>();

  private final String _name;

  public Catalog(String name) {
    this._name = name;
  }

  public String name() {
    return this._name;
  }

  public void addFieldType(FlintField builder) {
    // ignore non-indexed fields
    if (builder.index() == IndexOptions.NONE) return;
    synchronized (this._fields) {
      CatalogEntry existing = this._fields.get(builder.name());
      if (existing == null || !existing.equals(new CatalogEntry(builder, false)))
        this._fields.put(builder.name(), new CatalogEntry(builder, existing != null));
    }
  }

  public void addFieldType(boolean stored, String name, boolean tokenized, DocValuesType dt, NumericType num,
      SimpleDateFormat df, Resolution r) {
    synchronized (this._fields) {
      CatalogEntry newone = new CatalogEntry(stored, dt, tokenized, num, df, r, false);
      CatalogEntry existing = this._fields.get(name);
      if (existing == null || !existing.equals(newone))
        this._fields.put(name, new CatalogEntry(stored, dt, tokenized, num,  df, r, existing != null));
    }
  }

  CatalogEntry get(String fieldname) {
    return this._fields.get(fieldname);
  }

  public NumericType getNumericType(String fieldname) {
    CatalogEntry entry = this._fields.get(fieldname);
    return entry == null ? null : entry.num;
  }

  public boolean isTokenized(String fieldname) {
    CatalogEntry entry = this._fields.get(fieldname);
    return entry != null && entry.tokenized;
  }

  public DocValuesType getDocValuesType(String fieldname) {
    CatalogEntry entry = this._fields.get(fieldname);
    return entry == null ? null : entry.docValues;
  }

  public Collection getFieldsByPrefix(String prefix) {
    List matching = new ArrayList<>();
    synchronized (this._fields) {
      for (String field : this._fields.keySet()) {
        if (field.startsWith(prefix)) {
          matching.add(field);
        }
      }
    }
    return matching;
  }

  public Resolution getResolution(String field) {
    CatalogEntry entry = this._fields.get(field);
    return entry != null ? entry.resolution : null;
  }

  public SimpleDateFormat getDateFormat(String field) {
    CatalogEntry entry = this._fields.get(field);
    return entry != null ? entry.dateFormat : null;
  }

  public void clear() {
    synchronized (this._fields) {
      this._fields.clear();
    }
  }
  @Override
  public void toXML(XMLWriter xml) throws IOException {
    xml.openElement("catalog");
    xml.attribute("name", this._name);
    for (Map.Entry field : this._fields.entrySet()) {
      xml.openElement("field");
      xml.attribute("name", field.getKey());
      CatalogEntry entry = field.getValue();
      xml.attribute("stored",     String.valueOf(entry.stored));
      xml.attribute("tokenized",  String.valueOf(entry.tokenized));
      if (entry.num != null)
        xml.attribute("numeric-type", entry.num.name().toLowerCase());
      if (entry.docValues != null)
        xml.attribute("doc-values", entry.docValues == DocValuesType.SORTED_SET ? "sorted-set" :
                                    entry.docValues == DocValuesType.SORTED ||
                                    entry.docValues == DocValuesType.SORTED_NUMERIC ? "sorted" : "none");
      if (entry.error) xml.attribute("error", "true");
      if (entry.dateFormat != null)
        xml.attribute("date-format", entry.dateFormat.toPattern());
      if (entry.resolution != null)
        xml.attribute("date-resolution", entry.resolution.toString().toLowerCase());
      xml.closeElement();
    }
    xml.closeElement();
  }

  public static class CatalogEntry {
    private final boolean tokenized;
    private boolean error;
    private final boolean stored;
    private final DocValuesType docValues;
    private final NumericType num;
    private final SimpleDateFormat dateFormat;
    private final Resolution resolution;
    public CatalogEntry(boolean s, DocValuesType dv, boolean t, NumericType n, SimpleDateFormat df, Resolution r, boolean e) {
      this.stored = s;
      this.docValues = dv;
      this.tokenized = t;
      this.num = n;
      this.resolution = r;
      this.dateFormat = df;
      this.error = e;
    }
    public CatalogEntry(FlintField builder, boolean err) {
      this.stored = builder.store();
      this.tokenized = builder.tokenize();
      this.num = builder.numericType();
      this.dateFormat = builder.dateformat();
      this.resolution = builder.resolution();
      this.error = err;
      this.docValues = builder.docValues();
    }
    public void update(FlintField field) {
      field.store(this.stored);
      field.dateFormat(this.dateFormat);
      field.tokenize(this.tokenized);
      field.numeric(this.num);
      field.resolution(this.resolution);
      field.docValues(this.docValues);
      this.error = true;
    }
    @Override
    public boolean equals(Object obj) {
      if (obj instanceof CatalogEntry) {
        CatalogEntry entry = (CatalogEntry) obj;
        return this.tokenized  == entry.tokenized &&
               this.stored     == entry.stored &&
               this.num        == entry.num &&
               this.docValues  == entry.docValues &&
               ((this.dateFormat == null && entry.dateFormat == null) ||
                (this.dateFormat != null && this.dateFormat.equals(entry.dateFormat)));
      }
      return false;
    }
    public boolean equalsButDocValues(Object obj) {
      if (obj instanceof CatalogEntry) {
        CatalogEntry entry = (CatalogEntry) obj;
        return this.tokenized  == entry.tokenized &&
            this.stored     == entry.stored &&
            this.num        == entry.num &&
            (this.docValues  == entry.docValues ||
                (this.docValues != null && entry.docValues == DocValuesType.FORCED_NONE) ||
                (entry.docValues != null && this.docValues == DocValuesType.FORCED_NONE)) &&
            ((this.dateFormat == null && entry.dateFormat == null) ||
                (this.dateFormat != null && this.dateFormat.equals(entry.dateFormat)));
      }
      return false;
    }
    @Override
    public int hashCode() {
      return (this.num == null ? 13 : this.num.hashCode() * 17) +
             (this.stored    ? 19 : 2) +
             (this.docValues == null ? 23 : this.docValues.hashCode() * 7) +
             (this.tokenized ? 5 : 11) +
             (this.dateFormat == null ? 31 : this.dateFormat.hashCode() * 31);
    }
  }
}