info.unterrainer.commons.httpserver.daos.BasicJpqlDao Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http-server Show documentation

There is a newer version: 0.3.14

package info.unterrainer.commons.httpserver.daos;

import java.util.List;
import java.util.Set;
import java.util.function.Function;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.TypedQuery;

import info.unterrainer.commons.rdbutils.entities.BasicJpa;
import io.javalin.http.Context;
import lombok.Getter;

public class BasicJpqlDao implements CoreDaoProvider {

	protected final EntityManagerFactory emf;
	protected final Class
 type;
	@Getter
	protected JpqlCoreDao
 coreDao;

	public BasicJpqlDao(final EntityManagerFactory emf, final Class
 type) {
		super();
		this.emf = emf;
		this.type = type;
		coreDao = new JpqlCoreDao<>(emf, type);
	}

	public BasicJpqlDao(final Function entityManagerFactorySupplier,
			final Class
 type) {
		super();
		this.emf = null;
		this.type = type;
		coreDao = new JpqlCoreDao<>(entityManagerFactorySupplier, type);
	}

	 List getList(final EntityManager em, final TypedQuery query, final long offset, final long size) {
		int s = Integer.MAX_VALUE;
		if (size < s)
			s = (int) size;
		int o = Integer.MAX_VALUE;
		if (offset < o)
			o = (int) offset;
		query.setFirstResult(o);
		query.setMaxResults(s);
		return query.getResultList();
	}

	private  T getFirst(final EntityManager em, final TypedQuery query) {
		List r = getList(em, query, 0, 1);
		if (r.size() == 1) {
			T jpa = r.get(0);
			return jpa;
		}
		return null;
	}

	UpsertResult
 upsert(final EntityManager em, final TypedQuery
 query, final P entity,
			final Set readTenantIds, final Set writeTenantIds) {
		boolean wasInserted = false;
		boolean wasUpdated = false;
		P e = getFirst(em, query);
		if (e == null) {
			e = coreDao.create(em, entity, writeTenantIds);
			wasInserted = true;
		} else {
			entity.setId(e.getId());
			entity.setCreatedOn(e.getCreatedOn());
			e = coreDao.update(em, entity, readTenantIds);
			wasUpdated = true;
		}
		return UpsertResult.builder().wasInserted(wasInserted).wasUpdated(wasUpdated).jpa(e).build();
	}
}