info.unterrainer.commons.httpserver.daos.BasicJpqlDao Maven / Gradle / Ivy
package info.unterrainer.commons.httpserver.daos;
import java.util.List;
import java.util.Set;
import java.util.function.Function;
import jakarta.persistence.EntityManager;
import jakarta.persistence.EntityManagerFactory;
import jakarta.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();
}
}