• Home
• org.jeecgframework
• minidao-pe
• 1.10.2
• source code
• MiniDaoHandler.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.jeecgframework.minidao.aop.MiniDaoHandler Maven / Gradle / Ivy

package org.jeecgframework.minidao.aop;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.math.BigDecimal;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import ognl.Ognl;
import ognl.OgnlException;

import org.apache.commons.lang3.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jeecgframework.minidao.annotation.Arguments;
import org.jeecgframework.minidao.annotation.IgnoreSaas;
import org.jeecgframework.minidao.annotation.ResultType;
import org.jeecgframework.minidao.annotation.Sql;
import org.jeecgframework.minidao.annotation.id.IdType;
import org.jeecgframework.minidao.annotation.id.TableId;
import org.jeecgframework.minidao.aspect.EmptyInterceptor;
import org.jeecgframework.minidao.def.MiniDaoConstants;
import org.jeecgframework.minidao.pagehelper.dialect.PageAutoDialect;
import org.jeecgframework.minidao.pojo.MiniDaoPage;
import org.jeecgframework.minidao.spring.rowMapper.MiniColumnMapRowMapper;
import org.jeecgframework.minidao.spring.rowMapper.MiniColumnOriginalMapRowMapper;
import org.jeecgframework.minidao.util.*;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.Lazy;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.BeanPropertyRowMapper;
import org.springframework.jdbc.core.ColumnMapRowMapper;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;

import javax.sql.DataSource;

/**
 * 
 * @Title: MiniDao动态代理类
 * @author 张代浩
 * @mail [email protected]
 * @category www.jeecg.com
 * @date 20130817
 * @version V1.0
 */
@SuppressWarnings("rawtypes")
public class MiniDaoHandler implements InvocationHandler {
	private static final Log logger = LogFactory.getLog(MiniDaoHandler.class);
	
	/**自定义的数据源*/
    @Autowired(required = false)
    @Qualifier("minidaoDataSource")
    public DataSource dataSource;
    
	@Autowired
	@Lazy
	private JdbcTemplate jdbcTemplate;
	@Autowired
	@Lazy
	private NamedParameterJdbcTemplate namedParameterJdbcTemplate;
	/** minidao拦截器 */
	@Autowired(required = false)
	@Lazy
	private EmptyInterceptor emptyInterceptor;

	private ApplicationContext applicationContext;
	/**
	 * 多数据源配置情况下，minidao默认的数据源名
	 */
	private String MUTL_DATASOURCES_MINIDAO_DF_DSNAME = "minidaoDataSource";
	private String UPPER_KEY = "upper";

	private String LOWER_KEY = "lower";
	/**
	 * map的关键字类型 三个值
	 */
	private String keyType = "origin";
	private boolean formatSql = false;
	private boolean showSql = false;

	//自定获取方言
	protected PageAutoDialect pageAutoDialect = new PageAutoDialect();
     //序列查询sql
	public static final String SEQ_NEXTVAL_SQL = "SELECT %s.nextval FROM DUAL";

	@Override
	public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
		// 返回结果
		Object returnObj = null;
		// SQL模板
		String templateSql = null;
		// SQL模板参数
		Map sqlParamsMap = new HashMap();
		// 分页参数
		MiniDaoPage pageSetting = new MiniDaoPage();

		// Step.0 判断是否是Hiber实体维护方法，如果是执行Hibernate方式实体维护
//		Map rs = new HashMap();

		// Step.1装载SQL模板，所需参数
		templateSql = installDaoMetaData(pageSetting, method, sqlParamsMap, args);

		// Step.3解析SQL模板，返回可执行SQL
		String executeSql = parseSqlTemplate(method, templateSql, sqlParamsMap);
		//jeecg : Minidao报错“Template java/lang/Object_toString.sql not found”的解决方案
		if(executeSql==null || "".equals(executeSql)){
			return null;
		}

		// Step.4 组装SQL占位符参数
		Map sqlMap = installPlaceholderSqlParam(executeSql, sqlParamsMap);

		//step.5  检查Dialect初始化
		initCheckDialectExists();

		// Step.6 获取SQL执行返回值
		try {
			returnObj = getReturnMinidaoResult(pageSetting, method, executeSql, sqlMap, args);
		} catch (Exception e) {
			returnObj = null;
			if(e instanceof EmptyResultDataAccessException){
				//数据查询为空，不抛出Spring异常
			}else{
				e.printStackTrace();
				throw e;
			}
		}
		if (showSql) {
			logger.info("Print MiniDao-Original-SQL :\n\n" + executeSql);
		}
		return returnObj;
	}

	/**
	 * 判斷是否是執行的方法（非查詢）
	 * 
	 * @param methodName
	 * @return
	 */
	private static boolean checkActiveKey(String methodName) {
		String keys[] = MiniDaoConstants.INF_METHOD_ACTIVE.split(",");
		for (String s : keys) {
			if (methodName.startsWith(s))
				return true;
		}
		return false;
	}
	
	/**
	 * 判斷SQL是否（非查詢）
	 * 
	 * @param methodName
	 * @return
	 */
	private static boolean checkActiveSql(String sql) {
		sql = sql.trim().toLowerCase();
		String keys[] = MiniDaoConstants.INF_METHOD_ACTIVE.split(",");
		for (String s : keys) {
			if (sql.startsWith(s))
				return true;
		}
		return false;
	}

	/**
	 * 判斷是否批處理
	 * 
	 * @param methodName
	 * @return
	 */
	private static boolean checkBatchKey(String methodName) {
		String keys[] = MiniDaoConstants.INF_METHOD_BATCH.split(",");
		for (String s : keys) {
			if (methodName.startsWith(s))
				return true;
		}
		return false;
	}

	/**
	 * 把批量处理的结果拼接起来
	 * 
	 * @Author JueYue
	 * @date 2013-11-17
	 */
	private void addResulArray(int[] result, int index, int[] arr) {
		int length = arr.length;
		for (int i = 0; i < length; i++) {
			result[index - length + i] = arr[i];
		}
	}

	/**
	 * 批处理
	 * 
	 * @Author JueYue
	 * @date 2013-11-17
	 * @return
	 */
	private int[] batchUpdate(String executeSql) {
		String[] sqls = executeSql.split(";");
		if (sqls.length < 100) {
			return jdbcTemplate.batchUpdate(sqls);
		}
		int[] result = new int[sqls.length];
		List sqlList = new ArrayList();
		for (int i = 0; i < sqls.length; i++) {
			sqlList.add(sqls[i]);
			if (i % 100 == 0) {
				addResulArray(result, i + 1, jdbcTemplate.batchUpdate(sqlList.toArray(new String[0])));
				sqlList.clear();
			}
		}
		addResulArray(result, sqls.length, jdbcTemplate.batchUpdate(sqlList.toArray(new String[0])));
		return result;
	}

	/**
	 * 根据参数设置map的key大小写
	 **/
	private RowMapper> getColumnMapRowMapper() {
		if (getKeyType().equalsIgnoreCase(LOWER_KEY)) {
			return new MiniColumnMapRowMapper();
		} else if (getKeyType().equalsIgnoreCase(UPPER_KEY)) {
			return new ColumnMapRowMapper();
		} else {
			return new MiniColumnOriginalMapRowMapper();
		}
	}

//	/**
//	 * 获取总数sql - 如果要支持其他数据库，修改这里就可以
//	 *
//	 * @param sql
//	 * @return
//	 */
//	private String getCountSql(String sql) {
//		//update-begin---author:scott----date:20170803------for:分页count去掉排序，兼容SqlServer，同时提高效率--------
//		sql = removeOrderBy(sql);
//		//update-end---author:scott----date:20170803------for:分页count去掉排序，兼容SqlServer，同时提高效率--------
//		return "select count(0) from (" + sql + ") tmp_count";
//	}
//
//	/**
//	 * 为了兼容SQLServer
//	 * 去除子查询中的order by (也为了提升分页性能)
//	 * @param sql
//	 * @return
//	 */
//	public String removeOrderBy(String sql) {
//		if(sql==null){
//			return null;
//		}
//		//sql = sql.replaceAll("(?i)order by [\\s|\\S]+$", "");
//		try {
//			logger.debug(" --- 去除子查询中的order by (为了兼容SQLServer) --- orig sql="+sql);
//			sql = SqlServerParse.class.newInstance().removeOrderBy(sql);
//			logger.debug(" --- 去除子查询中的order by (为了兼容SQLServer) --- upda sql="+sql);
//		} catch (Exception e) {
//			e.printStackTrace();
//		}
//		return sql;
//	}

	public JdbcTemplate getJdbcTemplate() {
		if (jdbcTemplate == null) {
			try {
				namedParameterJdbcTemplate = applicationContext.getBean(NamedParameterJdbcTemplate.class);
				jdbcTemplate = applicationContext.getBean(JdbcTemplate.class);
			} catch (BeansException e) {
				logger.warn(e.getMessage());
				Map multDataSourceMap = applicationContext.getBeansOfType(DataSource.class);
				if (multDataSourceMap != null) {
					String keyOfTheFirst = multDataSourceMap.entrySet().stream().filter(d -> d.getKey().equals(MUTL_DATASOURCES_MINIDAO_DF_DSNAME)).findFirst().get().getKey();
					namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(multDataSourceMap.get(keyOfTheFirst));
					jdbcTemplate = new JdbcTemplate(multDataSourceMap.get(keyOfTheFirst));
				}
			}
		}else{
		    //update-begin--Author:wangshuai--Date:20211018--for: 用户自定义的数据源是否为空，不为空则走用户自定义的数据源
            if(null != dataSource){
                namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
                jdbcTemplate = new JdbcTemplate(dataSource);  
            }
            //update-end--Author:wangshuai--Date:20211018--for: 用户自定义的数据源是否为空，不为空则走用户自定义的数据源
        }
		
		return jdbcTemplate;
	}

	public String getKeyType() {
		return keyType;
	}

	public NamedParameterJdbcTemplate getNamedParameterJdbcTemplate() {
		return namedParameterJdbcTemplate;
	}

	/**
	 * 获取MiniDao处理结果集
	 * 
	 * @param dbType
	 * @param pageSetting
	 * @param jdbcTemplate
	 * @param method
	 * @param executeSql
	 * @return 结果集
	 */
	@SuppressWarnings("unchecked")
	private Object getReturnMinidaoResult(MiniDaoPage pageSetting, Method method, String executeSql, Map paramMap, Object[] args) {
		//update-begin---author:scott----date:20210608------for:springboot2.5/多数据配置问题, 导致jdbcTemplate、namedParameterJdbcTemplate注入是个null.注入失败.-------
		if (namedParameterJdbcTemplate == null) {
			try {
				namedParameterJdbcTemplate = applicationContext.getBean(NamedParameterJdbcTemplate.class);
				jdbcTemplate = applicationContext.getBean(JdbcTemplate.class);
			} catch (BeansException e) {
				logger.warn(e.getMessage());
				Map multDataSourceMap = applicationContext.getBeansOfType(DataSource.class);
				if (multDataSourceMap != null) {
					String keyOfTheFirst = multDataSourceMap.entrySet().stream().filter(d -> d.getKey().equals(MUTL_DATASOURCES_MINIDAO_DF_DSNAME)).findFirst().get().getKey();
					namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(multDataSourceMap.get(keyOfTheFirst));
					jdbcTemplate = new JdbcTemplate(multDataSourceMap.get(keyOfTheFirst));
				}
			}
		}else{
            //update-begin--Author:wangshuai--Date:20211018--for: 用户自定义的数据源是否为空，不为空则走用户自定义的数据源
            if(null!=dataSource){
                namedParameterJdbcTemplate = new NamedParameterJdbcTemplate(dataSource);
                jdbcTemplate = new JdbcTemplate(dataSource);
            }
            //update-end--Author:wangshuai--Date:20211018--for: 用户自定义的数据源是否为空，不为空则走用户自定义的数据源
        }
		
		if(emptyInterceptor==null){
			try {
				emptyInterceptor = applicationContext.getBean(EmptyInterceptor.class);
			} catch (BeansException e) {
				logger.warn(e.getMessage());
			}
		}
		//update-end---author:scott----date:20210608------for:springboot2.5/多数据配置问题, 导致jdbcTemplate、namedParameterJdbcTemplate注入是个null.注入失败.-------

		// step.4.调用SpringJdbc引擎，执行SQL返回值
		// 5.1获取返回值类型[Map/Object/List