工作笔记

package manager.dao.impl; import java.io.Serializable; import java.lang.reflect.Field; import java.util.HashMap; import java.util.Iterator; import java.util.List; import java.util.Map; import javax.persistence.Column; import javax.persistence.EmbeddedId; import javax.persistence.Entity; import javax.persistence.EntityManager; import javax.persistence.NoResultException; import javax.persistence.Query; import manager.dao.BaseDao; import mlink.jpa.JPADAO; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; public class BaseDaoImpl implements BaseDao { protected static Log logger = LogFactory .getLog(BaseDaoImpl.class); @SuppressWarnings("unchecked") public Integer countObject(Object example, String extraQuery, Map extraParas) { EntityManager em = JPADAO.getEntityManager(); Query query = countByExample(em, example, extraQuery, extraParas); Integer count = null; try { List li = (List) query.getResultList(); count = ((Long) li.get(0)).intValue(); } catch (NoResultException e) { logger.error(e.getMessage()); } finally { JPADAO.closeEntityManager(em); } return count; } public Integer countObject(Object example) { return countObject(example, "", new HashMap()); } public List getListBySQL(String sql, Map paras) { List list = null; EntityManager em = JPADAO.getEntityManager(); Query query = em.createQuery(sql); try { for (String key : paras.keySet()) { query.setParameter(key, paras.get(key)); } list = query.getResultList(); } catch (NoResultException e) { logger.error(e.getMessage()); } finally { JPADAO.closeEntityManager(em); } return list; } public Object getObjectBySQL(String sql, Map paras) { List list = getListBySQL(sql, paras); return (list != null && list.size() == 1) ? list.get(0) : null; } public T getSingleObject(T example) { List list = searchObjects(example); return (list != null && list.size() == 1) ? list.get(0) : null; } public void insertObject(Object o) { EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); em.persist(o); JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } } public T queryById(Class refClass, Serializable key) { T result = null; EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); result = em.find(refClass, key); JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } return result; } public void removeObject(Class clazz, Serializable id) { EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); T entity = em.find(clazz, id); em.remove(entity); JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } } public void removeObject(T obj) { EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); // obj = getSingleObject(obj); // if (obj != null) em.remove(obj); JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } } public List searchObjects(T example, String extraQuery, Map extraParas, int firstResult, int maxResults, String orderByName) { return searchObjects(example, extraQuery, extraParas, firstResult, maxResults, orderByName, false); } public List searchObjects(T example, String extraQuery, Map extraParas, int firstResult, int maxResults, String orderByName, boolean descOrder) { EntityManager em = JPADAO.getEntityManager(); Query query = queryByExample(em, example, extraQuery, extraParas, orderByName, descOrder); if (maxResults > 0) { query.setFirstResult(firstResult); query.setMaxResults(maxResults); } List li = null; try { li = (List ) query.getResultList(); } catch (NoResultException e) { logger.error(e.getMessage()); } finally { JPADAO.closeEntityManager(em); } return li; } public List searchObjects(T example, String extraQuery, Map extraParas, int firstResult, int maxResults) { return searchObjects(example, extraQuery, extraParas, firstResult, maxResults, ""); } public List searchObjects(T example, int firstResult, int maxResults, String orderByName, boolean descOrder) { return searchObjects(example, "", new HashMap(), firstResult, maxResults, orderByName, descOrder); } public List searchObjects(T example, int firstResult, int maxResults, String orderByName) { return searchObjects(example, "", new HashMap(), firstResult, maxResults, orderByName); } public List searchObjects(T example, int firstResult, int maxResults) { return searchObjects(example, firstResult, maxResults, ""); } public List searchObjects(T example) { return searchObjects(example, -1, -1); } public T updateObject(T o) { EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); em.merge(o); JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } return o; } protected Query countByExample(EntityManager em, T example, String extraQuery, Map extraParas) { StringBuffer sb = new StringBuffer(); Object[] params = null; try { Class c = example.getClass(); sb.append("select count(*) from " + c.getSimpleName() + " o "); if (c.isAnnotationPresent(Entity.class)) { // param count int nParams = 0; Field[] f = c.getDeclaredFields(); for (int i = 0; i < f.length; i++) { if (f[i].isAnnotationPresent(Column.class)) { nParams++; } else if (f[i] .isAnnotationPresent(EmbeddedId.class)) { Class c2 = f[i].getType(); Field[] f2 = c2.getDeclaredFields(); nParams = nParams + f2.length; } } params = new Object[nParams + 1]; int j = 1; for (int i = 0; i < f.length; i++) { if (f[i].isAnnotationPresent(Column.class)) { f[i].setAccessible(true); if (f[i].get(example) != null) { params[j] = f[i].get(example); if (params[j] instanceof String && ((String) params[j]).length() == 0) continue; if (j == 1) { sb.append(" where o."); } else { sb.append(" and o."); } sb.append(f[i].getName()); sb.append(" = "); sb.append("?" + j); j++; } } else if (f[i] .isAnnotationPresent(EmbeddedId.class)) { Class c2 = f[i].getType(); Field[] f2 = c2.getDeclaredFields(); for (int k = 0; k < f2.length; k++) { if (f2[k] .isAnnotationPresent(Column.class)) { f2[k].setAccessible(true); f[i].setAccessible(true); if (f2[k].get(f[i].get(example)) != null) { params[j] = f2[k].get(f[i] .get(example)); if (params[j] instanceof String && ((String) params[j]) .length() == 0) continue; if (j == 1) { sb.append(" where o."); } else { sb.append(" and o."); } sb.append(f[i].getName()); sb.append("."); sb.append(f2[k].getName()); sb.append(" = "); sb.append("?" + j); j++; } } } } } if (extraQuery != null && extraQuery.length() > 0) { if (j == 1) { sb.append(" where " + extraQuery); } else { sb.append(" and " + extraQuery); } } } } catch (Exception e) { e.printStackTrace(); logger.error(e.getMessage()); logger.error("Query is:" + sb.toString()); } Query query = em.createQuery(sb.toString()); for (int i = 1; i < params.length; i++) { if (params[i] != null) { if (params[i] instanceof String && ((String) params[i]).length() == 0) continue; query.setParameter(i, params[i]); } } for (String key : extraParas.keySet()) { query.setParameter(key, extraParas.get(key)); } return query; } protected Query queryByExample(EntityManager em, Object example, String extraQuery, Map extraParas, String orderByName, boolean descOrder) { StringBuffer sb = new StringBuffer(); Object[] params = null; try { Class c = example.getClass(); sb.append("select o from " + c.getSimpleName() + " o "); if (c.isAnnotationPresent(Entity.class)) { // param count int nParams = 0; Field[] f = c.getDeclaredFields(); for (int i = 0; i < f.length; i++) { if (f[i].isAnnotationPresent(Column.class)) { nParams++; } else if (f[i] .isAnnotationPresent(EmbeddedId.class)) { Class c2 = f[i].getType(); Field[] f2 = c2.getDeclaredFields(); nParams = nParams + f2.length; } } params = new Object[nParams + 1]; int j = 1; for (int i = 0; i < f.length; i++) { if (f[i].isAnnotationPresent(Column.class)) { f[i].setAccessible(true); if (f[i].get(example) != null) { params[j] = f[i].get(example); if (params[j] instanceof String && ((String) params[j]).length() == 0) // continue; if (j == 1) { sb.append(" where o."); } else { sb.append(" and o."); } sb.append(f[i].getName()); sb.append(" = "); sb.append("?" + j); j++; } } else if (f[i] .isAnnotationPresent(EmbeddedId.class)) { Class c2 = f[i].getType(); Field[] f2 = c2.getDeclaredFields(); for (int k = 0; k < f2.length; k++) { if (f2[k] .isAnnotationPresent(Column.class)) { f2[k].setAccessible(true); f[i].setAccessible(true); if (f2[k].get(f[i].get(example)) != null) { params[j] = f2[k].get(f[i] .get(example)); if (params[j] instanceof String && ((String) params[j]) .length() == 0) continue; if (j == 1) { sb.append(" where o."); } else { sb.append(" and o."); } sb.append(f[i].getName()); sb.append("."); sb.append(f2[k].getName()); sb.append(" = "); sb.append("?" + j); j++; } } } } } if (extraQuery != null && extraQuery.length() > 0) { if (j == 1) { sb.append(" where " + extraQuery); } else { sb.append(" and " + extraQuery); } } if (orderByName != null && orderByName.length() > 0) { sb.append(" order by o."); sb.append(orderByName); if (descOrder) { sb.append(" desc"); } } } } catch (Exception e) { logger.error(e.getMessage()); logger.error("Query is:" + sb.toString()); } Query query = em.createQuery(sb.toString()); for (int i = 1; i < params.length; i++) { if (params[i] != null) { if (params[i] instanceof String && ((String) params[i]).length() == 0) continue; query.setParameter(i, params[i]); } } for (String key : extraParas.keySet()) { query.setParameter(key, extraParas.get(key)); } return query; } public void beginTransaction() { JPADAO.beginTransaction(JPADAO.getEntityManager()); } public void commitTransaction() { JPADAO.commitTransaction(JPADAO.getEntityManager()); } public void rollBackTransaction() { JPADAO.rollbackTransaction(JPADAO.getEntityManager()); } public void insertForInput(Listo) { EntityManager em = JPADAO.getEntityManager(); try { JPADAO.beginTransaction(em); for (Iterator iterator = o.iterator(); iterator.hasNext();) { Object object = (Object) iterator.next(); em.persist(object); } JPADAO.commitTransaction(em); } catch (RuntimeException e) { logger.error(e, e); JPADAO.rollbackTransaction(em); throw e; } finally { JPADAO.closeEntityManager(em); } } }