念奴娇 赤壁怀古
     [北宋]苏轼
大江东去,浪淘尽,千古风流人物。
故垒西边,人道是,三国周郎赤壁。
乱石穿空,惊涛拍岸,卷起千堆雪。
江山如画,一时多少豪杰。

遥想公瑾当年,小乔初嫁了,雄姿英发。
羽扇纶巾,谈笑间,樯橹灰飞烟灭。
故国神游,多情应笑我,早生华发。
人生如梦,一樽还酹江月。

Mybatis中SqlMapper配置的扩展与应用(3)

隔了两周,首先回顾一下,在Mybatis中的SqlMapper配置文件中引入的几个扩展机制:

1.引入SQL配置函数,简化配置、屏蔽DB底层差异性
2.引入自定义命名空间,允许自定义语句级元素、脚本级元素
3.引入表达式配置,扩充SqlMapper配置的表达能力

前面两条已经举过例子,现在来看看怎么使用表达式配置。说到表达式语言,最为富丽堂皇的自然就是OGNL,但这也正是Mybatis内部访问数据的固有方式,所以也轮不到我们在这里来扩充了(事实上Mybatis的参数设置并不能使用完全的OGNL)。那么,除了OGNL,还有哪些表达式语言呢?别忘了,我们的前提是Spring环境,自然,SpEL表达式也就走入我们的视野,因此这篇文章就重点记录在SqlMapper中使用SpEL表达式

四、在Mybatis中的SqlMapper使用SpEL表达式

1.SpEL工具类

SpEL就是Spring提供的EL表达式,虽然到Spring3才开始推出,但已经是Spring的一个基础核心模块了,地位已经差不多等同于IoC和AOP了。SpEL和OGNL类似,也有表达式、上下文环境、root对象等概念,但和OGNL不同的是,SpEL还提供了访问Spring中bean的能力——这是非常强悍的,试问一个Spring应用有多少类不是Spring管理的呢?具体的SpEL语法细节可以参考Spring的官方文档。
SpEL目前主要应用于Spring的配置,使用起来非常方便,但是在Java类中使用则比较繁琐,稍微实用一点的例子都需要创建解析器实例、创建执行环境、解析表达式、对表达式求值等步骤,如果需要访问Spring的Bean,还要设置BeanFactoryResolver等,因此,为了简化SpEL在Java中的应用,我编写了一个SpEL的帮助类:

这个工具类分成四个部分:

  1. 实现ApplicationContextAware接口,注入ApplicationContext(BeanFactory)对象
  2. 表达式求值方法
    • 对表达式简单求值(还可指定返回的目标类型)
    • 指定root对象,对表达式求值(还可指定返回的目标类型)
    • 指定root对象和其它变量,对表达式求值(还可指定返回的目标类型)
  3. 表达式设置方法
    • 设置表达式的值
    • 指定root对象,设置表达式的值
    • 指定root对象和其它变量,设置表达式的值
  4. 变量管理方法
    • 添加变量
    • 移除变量

此外,还内置了一个保护变量Tool。
编写一个测试类验证一下:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={
	"classpath:applicationContext.xml"	
})
@Component  // 该测试类本身作为一个Spring管理的bean,便于后面的测试
public class SpringHelpTest {
	
	public String getBeanValue(String arg){//bean的一个方法
		return "beanValue:"+arg;
	}

	@Test
	public void testSpelHelp(){
		// 准备root对象 {key1 : 'root-value1', key2 : 'root-value2'}
		Root root = new Root("root-value1", "root-value2");
		// 准备一般变量
		Map<String, Object> vars = new HashMap<String, Object>();
		vars.put("var1", "value1");
		vars.put("var2", "value2");
		// 直接计算简单表达式
		Object rs = SpringHelp.evaluate("1+2");
		Assert.assertEquals(3, rs);
		// 按指定类型计算简单表达式
		rs = SpringHelp.evaluate("1+2", String.class);
		Assert.assertEquals("3", rs);
		// 访问root对象的属性
		rs = SpringHelp.evaluate(root, "key1");
		Assert.assertEquals("root-value1", rs);
		// 访问一般变量
		rs = SpringHelp.evaluate(root, "#var2", vars);
		Assert.assertEquals("value2", rs);
		// 访问root对象
		rs = SpringHelp.evaluate(root, "#root", vars);
		Assert.assertTrue(rs == root);
		// 访问Spring管理的bean,同时传入的参数又是root对象的属性
		rs = SpringHelp.evaluate(root, "@springHelpTest.getBeanValue(key2)", vars);
		Assert.assertEquals("beanValue:root-value2", rs);
		// 设置root对象的属性
		SpringHelp.setValue(root, "key1", "new-root-value1");
		rs = SpringHelp.evaluate(root, "key1");
		Assert.assertEquals("new-root-value1", rs);
		//访问工具类,其中Tool.DATE.getDate()的作用是获取当前日期
		rs = SpringHelp.evaluate("#Tool.DATE.getDate()");
		Assert.assertEquals(Tool.DATE.getDate(), rs);
	}
	
	public class Root{
		String key1;
		String key2;
		Root(String key1, String key2){
			this.key1 = key1;
			this.key2 = key2;
		}
		// 省略getter/setter方法
	}
}

有了这个静态帮助类,在Java中使用SpEL就方便很多了。

2.编写表达式处理器

利用SpEL帮助类,编写表达式处理器IExpressionHandler的实现,具体逻辑参看代码中的注释

public class SpelExpressionHandler implements IExpressionHandler {
	
	/**
	 * 直接返回true,也就是说不做进一步判断,支持所有的${(exp)}、#{(exp)}内的表达式
	 * 由于支持所有表达式,实际上起到了一种拦截作用,所以需要注意,注册该实现时必须最低优先级
	 */
	@Override
	public boolean isSupport(String expression) {
		return true;
	}

	/**
	 * 对SqlMapper配置中的表达式求值
	 */
	@Override
	public Object eval(String expression, Object parameter, String databaseId) {
		/**
		 * 如果以spel:为前缀,则将mybatis包装后的参数、数据库id以及表达式自身一起封装一个新的root对象
		 * 因此在exp表达式中可以通过params.paramName、databaseId等形式访问
		 */
		if(expression.toLowerCase().startsWith("spel:")){
			expression = expression.substring(5);
			Root root = new Root(parameter, databaseId, expression);
			return SpringHelp.evaluate(root, expression);
		}
		/**
		 * 否则将databaseId作为一个特殊名称的变量
		 * 因此在exp表达式中可以通过paramName、#databaseId等形式访问
		 */
		else{
			Map<String, Object> vars = new HashMap<String, Object>();
			vars.put("databaseId", databaseId);
			return SpringHelp.evaluate(parameter, expression, vars);
		}
	}
	
	public class Root {

		private final Object params;
		private final String databaseId;
		private final String expression;

		public Root(Object params, String databaseId, String expression) {
			this.params = params;
			this.databaseId = databaseId;
			this.expression = expression;
		}

		// 省略getter/setter方法
	}
}

3.注册表达式处理器

如上面的注释,注册的时候需要注意一点,优先级要最低,以避免所有表达式都被拦截,导致其它的处理器不生效。
保证优先级最低,有一种方法,就是实现Spring中的Order接口,并且将该实现类的order值设置为最大,然后按Order排序;另外一种方法,就是干脆另起炉灶,单独一个属性保存默认处理器,只有其它处理器都不支持的时候才使用默认处理器,请看下面的代码:

/**
 * 表达式处理器
 */
private static final Set<IExpressionHandler> expressions = new LinkedHashSet<IExpressionHandler>();
/**
 * 默认表达式处理器
 */
private static final IExpressionHandler defaultExpressionHandler = new SpelExpressionHandler();
/**
 * 获取表达式处理器
 * @param node
 * @return
 */
public static IExpressionHandler getExpressionHandler(String expression){
	for(IExpressionHandler handler : expressions){
		if(handler.isSupport(expression)){
			return handler;
		}
	}
	return defaultExpressionHandler;
}

4.修改SqlMapper中配置

<?xml version="1.0" encoding="UTF-8" ?>
<mapper xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" 
	xmlns="http://dysd.org/schema/sqlmapper"
	xmlns:e="http://dysd.org/schema/sqlmapper-extend"
	xsi:schemaLocation="http://dysd.org/schema/sqlmapper http://dysd.org/schema/sqlmapper.xsd
		http://dysd.org/schema/sqlmapper-extend http://dysd.org/schema/sqlmapper-extend.xsd"
	namespace="org.dysd.dao.mybatis.mapper.IExampleDao">
	
	<select id="selectString" resultType="string">
		select PARAM_NAME, ${(@spelBean.param(paramName))} AS TEST_SPEL
		  from BF_PARAM_ENUM_DEF
		 where PARAM_NAME $like{#{(spel:@spelBean.root(#root,params.paramName)), jdbcType=VARCHAR}}
		 order by SEQNO
	</select>
</mapper>

5.编写配置中的bean

@Component("spelBean")
public class SpelBean {

	public String param(String paramName){
		// 测试的是${()},所以返回结果中添加单引号
		return "'PARAM-"+paramName+"'";
	}
	
	public String root(SpelExpressionHandler.Root root,String paramName){
		// 测试spel:为前缀的表达式,所以可以直接访问SpelExpressionHandler.Root对象
		return "ROOT-"+root.getDatabaseId()+"-"+paramName;
	}
}

6.编写Dao接口

@Repository
public interface IExampleDao {
	
	public String selectString(@Param("paramName")String paramName);
}

7.编写JUnit测试类

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations={
	"classpath:spring/applicationContext.xml"	
})
@Service
public class ExampleDaoTest{

	@Resource
	private IExampleDao dao;
	
	@Test
	public void testSelectString(){
		try {
			String a = dao.selectString("DISPLAY_AREA");
			Assert.assertEquals("显示区域", a);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}

8.执行测试

20161119 19:00:44,298 [main]-[DEBUG] ==>  Preparing: select PARAM_NAME, 'PARAM-DISPLAY_AREA' AS TEST_SPEL from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE CONCAT('%',?,'%') order by SEQNO 
20161119 19:00:48,001 [main]-[DEBUG] ==> Parameters: ROOT-MySQL-DISPLAY_AREA(String)

可以看到,无论是${(exp)}还是#{(exp)},其中的exp都已经得到正确的解析了。

在SqlMapper中可以调用Spring的Bean,大大丰富了SqlMapper的表达能力,但是对于${(exp)}这种情形,由于是字符串的简单替换,也存在SQL注入的风险,因此一般只使用#{(exp)}。

题外话:
1.SqlMapper的扩展与应用系列算是暂告一段落,有朋友希望我能提供实际的案例,我利用这两周的业余时间整理了一下,在GitHub和OSChina同步上传了这个项目,有兴趣的朋友可以看一下,也希望可以多提一点建议给我。因为是maven项目,希望实际运行的朋友最好搭建一个nexus私服,然后git下载,导入至Eclipse中,修改数据库配置即可。
具体地址:
GitHub:https://github.com/linjisong/dysd
OSChina:https://git.oschina.net/linjisong/dysd
2.在博客园中首次使用Markdown,好多地方还不熟悉,比如代码折叠,但也算是一种新的尝试。

posted @ 2016-11-19 19:28  linjisong  阅读(1273)  评论(7编辑  收藏  举报