org.zht.framework.zhtdao.base.impl.BaseDaoImpl.java Source code

Introduction

Here is the source code for org.zht.framework.zhtdao.base.impl.BaseDaoImpl.java

Source

/**
 * Copyright (c) 2015 https://github.com/zhaohuatai
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 */
package org.zht.framework.zhtdao.base.impl;

import java.util.Collection;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang3.StringUtils;
import org.hibernate.Query;
import org.hibernate.transform.Transformers;
import org.zht.framework.data.DataSet;
import org.zht.framework.data.ParamObject;
import org.zht.framework.exception.DaoException;
import org.zht.framework.zhtdao.base.IBaseDao;
import org.zht.framework.zhtdao.hibernate.impl.HibernateBaseDaoImpl;

//@Component("baseDaoImpl")
public class BaseDaoImpl extends HibernateBaseDaoImpl implements IBaseDao {

    public DataSet loadDataSet(String queryStr, ParamObject paramObject) throws DaoException {
        if (paramObject == null) {
            throw new DaoException("?");
        }
        if (paramObject.getIsNeedCount() == null) {
            throw new DaoException("[NeedCount]?");
        }
        if (paramObject.getIsSql() == null) {
            throw new DaoException("[isSql]?");
        }
        if (paramObject.getIsOffset() == null) {
            throw new DaoException("[isOffset]?");
        }
        if (queryStr == null || !(queryStr.contains("@from") || queryStr.contains("@FROM"))) {
            throw new DaoException("?,[@from]");
        }
        String queryCountStr = " select count(*) from " + StringUtils.substringAfter(queryStr, "@from");
        //?grouby ?
        String countTemp = queryCountStr.replaceAll("\\((.*)\\)", "").replaceAll("[\\s]+", " ").toLowerCase();
        if (countTemp.contains("group by")) {
            queryCountStr = "SELECT count(*) from ( " + queryCountStr + " )";
        }
        queryStr = queryStr.replace("@from", "from").replace("@FROM", "FROM");

        Query queryData, queryCount;
        if (paramObject.getIsSql()) {
            queryData = this.getCurrentSession().createSQLQuery(queryStr)
                    .setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
            queryCount = this.getCurrentSession().createSQLQuery(queryCountStr);
        } else {
            queryData = this.getCurrentSession().createQuery(queryStr)
                    .setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
            queryCount = this.getCurrentSession().createQuery(queryCountStr);
        }

        String tempSql = new String(queryStr);
        Map<String, Object> parmeterSet = paramObject.getQueryParams();
        if (parmeterSet != null && parmeterSet.size() > 0) {
            while (tempSql.contains(":")) {
                tempSql = StringUtils.substringAfter(tempSql, ":");
                String key = StringUtils.substringBefore(tempSql, " ");
                if (key != null && key.trim().endsWith(")")) {
                    key = key.replace(")", "");
                }
                if (parmeterSet.keySet().contains(key)) {
                    Object paramValue = parmeterSet.get(key);
                    //?null////
                    if (paramValue == null) {
                        queryData.setParameter(key, paramValue);
                        queryCount.setParameter(key, paramValue);
                    } else {
                        if (paramValue instanceof Collection) {
                            queryData.setParameterList(key, (Collection<?>) paramValue);
                            queryCount.setParameterList(key, (Collection<?>) paramValue);
                        } else if (paramValue.getClass().isArray()) {
                            queryData.setParameterList(key, (Object[]) paramValue);
                            queryCount.setParameterList(key, (Object[]) paramValue);
                        } else if ("true".equalsIgnoreCase("" + paramValue)
                                || "false".equalsIgnoreCase("" + paramValue)) {
                            queryData.setParameter(key, Boolean.valueOf("" + paramValue));
                        } else {
                            queryData.setParameter(key, paramValue);
                            queryCount.setParameter(key, paramValue);
                        }
                    }

                }
            }
        }

        if (paramObject.getIsOffset()) {
            queryData.setMaxResults(paramObject.getRows());
            queryData.setFirstResult((paramObject.getPage() - 1) * paramObject.getRows());
        }
        @SuppressWarnings("unchecked")
        List<Map<String, Object>> entityList = queryData.list();

        Long total = 0L;
        if (paramObject.getIsNeedCount()) {
            total = (Long) queryCount.uniqueResult();
        }
        return new DataSet(total, entityList);
    }

    @Override
    public List<?> findJustList(String queryStr, ParamObject paramObject) throws DaoException {
        if (paramObject == null) {
            throw new DaoException("?");
        }
        if (paramObject.getIsNeedCount() == null) {
            throw new DaoException("[NeedCount]?");
        }
        if (paramObject.getIsSql() == null) {
            throw new DaoException("[isSql]?");
        }
        if (paramObject.getIsOffset() == null) {
            throw new DaoException("[isOffset]?");
        }
        //      if(queryStr==null||!(queryStr.contains("@from")||queryStr.contains("@FROM"))){
        //         throw new DaoException("?,[@from]");
        //      }
        queryStr = queryStr.replace("@from", "from").replace("@FROM", "FROM");
        Query queryData;
        if (paramObject.getIsSql()) {
            queryData = this.getCurrentSession().createSQLQuery(queryStr);
        } else {
            queryData = this.getCurrentSession().createQuery(queryStr);
        }
        Map<String, Object> parmeterSet = paramObject.getQueryParams();
        String tempSql = new String(queryStr);
        if (parmeterSet != null && parmeterSet.size() > 0) {
            while (tempSql.contains(":")) {
                tempSql = StringUtils.substringAfter(tempSql, ":");
                String key = StringUtils.substringBefore(tempSql, " ");
                if (key != null && key.trim().endsWith(")")) {
                    key = key.replace(")", "");
                }
                if (parmeterSet.keySet().contains(key)) {
                    Object paramValue = parmeterSet.get(key);
                    //?null////
                    if (paramValue == null) {
                        queryData.setParameter(key, paramValue);
                    } else {
                        if (paramValue instanceof Collection) {
                            queryData.setParameterList(key, (Collection<?>) paramValue);
                        } else if (paramValue.getClass().isArray()) {
                            queryData.setParameterList(key, (Object[]) paramValue);
                        } else if ("true".equalsIgnoreCase("" + paramValue)
                                || "false".equalsIgnoreCase("" + paramValue)) {
                            queryData.setParameter(key, Boolean.valueOf("" + paramValue));
                        } else {
                            queryData.setParameter(key, paramValue);
                        }
                    }

                }
            }
        }
        List<?> entityList = queryData.list();
        if (entityList != null && entityList.size() > 0) {
            return entityList;
        }
        return null;
    }

    //=====================================================================
}