org.gra4j.gazelle.structure.JpaStructure Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gazelle Show documentation
The newest version!
package org.gra4j.gazelle.structure;

import org.gra4j.gazelle.JPAQuery.core.Operation;

import javax.persistence.EntityManager;
import javax.persistence.criteria.*;
import java.io.Serializable;

/**
 * gazelle
 * org.gra4j.gazelle.structure
 *
 * @author tom.long
 */
public class JpaStructure implements Serializable {

    private CriteriaBuilder cb;

    private Root root;

    private CriteriaQuery query;

    private CriteriaUpdate update;

    private CriteriaDelete delete;

    private Boolean isAggregated;

    public JpaStructure (EntityManager entityManager, Operation operation, Class clazz) {
        cb = entityManager.getCriteriaBuilder();
        isAggregated = Boolean.FALSE;
        switch (operation) {
            case tupleQuery:
                query = cb.createTupleQuery();
                root = query.from(clazz);
                break;
            case select:
                query = cb.createQuery(clazz);
                root = query.from(clazz);
                break;
            case update:
                update = cb.createCriteriaUpdate(clazz);
                root = update.from(clazz);
                break;
            case create:
                break;
            case delete:
                delete = cb.createCriteriaDelete(clazz);
                root = delete.from(clazz);
                break;
        }
    }

    public void setCb (CriteriaBuilder cb) {
        this.cb = cb;
    }

    public CriteriaBuilder getCb () {
        return this.cb;
    }

    public void setRoot (Root root) {
        this.root = root;
    }

    public Root getRoot () {
        return this.root;
    }

    public void setQuery (CriteriaQuery query) {
        this.query = query;
    }

    public CriteriaQuery getQuery () {
        return this.query;
    }

    public void setUpdate (CriteriaUpdate update) {
        this.update = update;
    }

    public CriteriaUpdate getUpdate () {
        return this.update;
    }

    public void setDelete (CriteriaDelete delete) {
        this.delete = delete;
    }

    public CriteriaDelete getDelete () {
        return this.delete;
    }

    public void setIsAggregated (Boolean aggregated) {
        this.isAggregated = aggregated;
    }

    public Boolean getIsAggregated () {
        return this.isAggregated;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof JpaStructure)) return false;

        JpaStructure that = (JpaStructure) o;

        if (getCb() != null ? !getCb().equals(that.getCb()) : that.getCb() != null) return false;
        if (getRoot() != null ? !getRoot().equals(that.getRoot()) : that.getRoot() != null) return false;
        if (getQuery() != null ? !getQuery().equals(that.getQuery()) : that.getQuery() != null) return false;
        if (getUpdate() != null ? !getUpdate().equals(that.getUpdate()) : that.getUpdate() != null) return false;
        if (getDelete() != null ? !getDelete().equals(that.getDelete()) : that.getDelete() != null) return false;
        return getIsAggregated() != null ? getIsAggregated().equals(that.getIsAggregated()) : that.getIsAggregated() == null;
    }

    @Override
    public int hashCode() {
        int result = getCb() != null ? getCb().hashCode() : 0;
        result = 31 * result + (getRoot() != null ? getRoot().hashCode() : 0);
        result = 31 * result + (getQuery() != null ? getQuery().hashCode() : 0);
        result = 31 * result + (getUpdate() != null ? getUpdate().hashCode() : 0);
        result = 31 * result + (getDelete() != null ? getDelete().hashCode() : 0);
        result = 31 * result + (getIsAggregated() != null ? getIsAggregated().hashCode() : 0);
        return result;
    }

    @Override
    public String toString() {
        return "JpaStructure{" +
                "cb=" + cb +
                ", root=" + root +
                ", query=" + query +
                ", update=" + update +
                ", delete=" + delete +
                ", isAggregated=" + isAggregated +
                '}';
    }
}