org.dromara.trans.extend.JPA2TransableAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of easy-trans-jpa-extend Show documentation

There is a newer version: 2.3.1

Show newest version

package org.dromara.trans.extend;

import org.dromara.core.trans.util.ReflectUtils;
import org.dromara.core.trans.vo.VO;
import org.dromara.trans.service.AutoTransable;

import javax.persistence.EntityManager;
import javax.persistence.TypedQuery;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.List;

public class JPA2TransableAdapter implements AutoTransable {

    private EntityManager em;

    private Class voClass;

    public JPA2TransableAdapter(Class voClass, EntityManager em) {
        this.em = em;
        this.voClass = voClass;
    }


    @Override
    public List findByIds(List ids) {
        if (ids == null || ids.isEmpty()) {
            return new ArrayList();
        }
        TypedQuery query = em.createQuery(getSelectSql()
                + " WHERE tbl." + getPkeyFieldName() + " IN :ids", voClass);
        query.setParameter("ids", ids);
        return query.getResultList();
    }

    private String getSelectSql() {
        return "FROM " + voClass.getSimpleName() + " AS tbl ";
    }

    private String getPkeyFieldName() {
        List fieldList = ReflectUtils.getAnnotationField(voClass, javax.persistence.Id.class);
        if (fieldList == null || fieldList.isEmpty()) {
            throw new RuntimeException(voClass.getName() + "没有@Id标记");
        }
        return fieldList.get(0).getName();
    }

    @Override
    public List select() {
        TypedQuery query = em.createQuery(getSelectSql(), voClass);
        return query.getResultList();
    }

    @Override
    public VO selectById(Object pkey) {
        TypedQuery query = em.createQuery(getSelectSql()
                + " WHERE tbl." + getPkeyFieldName() + " = :id", voClass);
        query.setParameter("id", pkey);
        return (VO) query.getSingleResult();
    }


}