org.pageseeder.flint.indexing.IndexBatch Maven / Gradle / Ivy

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

import java.util.Date;

public class IndexBatch {
  private final String index;
  private int          totalCount;
  private boolean      computed     = false;
  private boolean      started      = false;
  private int          currentCount = 0;
  private int          cancelCount  = -1;
  private final long   batchID;
  private final long   createTime;
  private long         startTime;
  private long         computeTime;
  private long         indexTime;
  private long         totalTime;
  private boolean      hasClearJob = false;

  public IndexBatch() {
    this(null);
  }

  public IndexBatch(String idx) {
    this.batchID = System.nanoTime();
    this.createTime = System.currentTimeMillis();
    this.index = idx;
    this.totalCount = 0;
  }

  public IndexBatch(String idx, int total) {
    this.batchID = System.nanoTime();
    this.createTime = System.currentTimeMillis();
    this.index = idx;
    this.totalCount = total;
    this.computeTime = -1;
    this.computed = true;
  }

  public IndexBatch withClearJob() {
    this.hasClearJob = true;
    return this;
  }

  public boolean hasClearJob() {
    return this.hasClearJob;
  }

  protected synchronized void startIndexing() {
    this.started = true;
    this.startTime = System.currentTimeMillis();
  }

  public synchronized void increaseTotal() {
    this.totalCount++;
  }

  public synchronized void increaseTotal(int by) {
    this.totalCount += by;
  }

  /**
   * Increase the current counter.
   * @return true if finished, false otherwise
   */
  protected synchronized boolean increaseCurrent() {
    this.currentCount++;
    boolean finished = isFinished();
    if (finished) {
      long now = System.currentTimeMillis();
      this.indexTime = now - this.startTime;
      this.totalTime = now - this.createTime;
    }
    return finished;
  }

  public synchronized void cancel(int currentCount) {
    this.cancelCount = currentCount;
  }

  public synchronized void remove(int removed) {
    this.totalCount -= removed;
  }

  public synchronized void setComputed() {
    this.computed = true;
    this.computeTime = System.currentTimeMillis() - this.createTime;
  }

  public synchronized boolean isComputed() {
    return this.computed;
  }

  public synchronized boolean isStarted() {
    return this.started;
  }

  public synchronized boolean isFinished() {
    return (this.cancelCount >= 0 && this.currentCount >= this.cancelCount)
        || (this.computed && this.currentCount >= this.totalCount);
  }

  public int getCurrentCount() {
    return this.currentCount;
  }

  public int getTotalDocuments() {
    return isFinished() ? this.currentCount : this.totalCount; // handle cancel
  }

  public String getID() {
    return String.valueOf(this.batchID);
  }

  public Date getCreation() {
    return new Date(this.createTime);
  }

  public Date getStart() {
    return new Date(this.startTime);
  }

  public long getIndexingDuration() {
    return this.indexTime;
  }

  public long getComputingDuration() {
    return this.computeTime;
  }

  public long getTotalDuration() {
    return this.totalTime;
  }

  public String getIndex() {
    return this.index;
  }
}