io.ebeaninternal.server.querydefn.AbstractQuery Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ebean-core Show documentation

ebean core module

There is a newer version: 15.8.0

package io.ebeaninternal.server.querydefn;

import java.util.concurrent.locks.ReentrantLock;

import jakarta.persistence.PersistenceException;

import io.ebean.CancelableQuery;
import io.ebeaninternal.api.SpiCancelableQuery;

/**
 * Common code for Dto/Orm/RelationalQuery
 *
 * @author Roland Praml, FOCONIS AG
 *
 */
public class AbstractQuery implements SpiCancelableQuery {

  private final ReentrantLock lock = new ReentrantLock();
  private boolean cancelled;
  private CancelableQuery cancelableQuery;
  protected boolean useMaster;

  @Override
  public final void cancel() {
    lock.lock();
    try {
      if (!cancelled) {
        cancelled = true;
        if (cancelableQuery != null) {
          cancelableQuery.cancel();
        }
      }
    } finally {
      lock.unlock();
    }
  }

  @Override
  public final void checkCancelled() {
    if (cancelled) {
      throw new PersistenceException("Query was cancelled");
    }
  }

  @Override
  public final void setCancelableQuery(CancelableQuery cancelableQuery) {
    lock.lock();
    try {
      checkCancelled();
      this.cancelableQuery = cancelableQuery;
    } finally {
      lock.unlock();
    }
  }
}