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

三、子表删除兼容问题

这个问题，使用SQL配置函数不太好处理，而且就算使用SQL配置函数，也不够直观，有点自动生成SQL的意味，太Hibernate了（不过要是可以兼收Hibernate和Mybatis两家之长，那也是一个不错的主意），下面我们使用自定义命名空间的方式来解决。

1、编写sqlmapper-extend命名空间的XSD文件，引进新的<db>元素

<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<xsd:schema xmlns="http://dysd.org/schema/sqlmapper-extend"
targetNamespace="http://dysd.org/schema/sqlmapper-extend"
xmlns:s="http://dysd.org/schema/sqlmapper"
xmlns:xsd="http://www.w3.org/2001/XMLSchema" elementFormDefault="qualified"
version="1.0">

<!-- 导入由mybatis官方DTD转换而来的XSD命名空间 -->
<xsd:import namespace="http://dysd.org/schema/sqlmapper"
schemaLocation="http://dysd.org/schema/sqlmapper.xsd"/>

<xsd:element name="db">
<xsd:annotation><xsd:documentation><![CDATA[
定义一个新的db元素，有一个枚举列表的属性type，表示数据库类型，以!开头表示否定
可以配置SQL文本、statement级元素、script级元素以及其它扩展空间元素等子元素
]]></xsd:documentation></xsd:annotation>
<xsd:complexType mixed="true">
<xsd:choice maxOccurs="unbounded">
<xsd:group ref="s:statementGroup" minOccurs="0" maxOccurs="1"/>
<xsd:group ref="s:dynaScriptGroup" minOccurs="0" maxOccurs="1" />
</xsd:choice>
<xsd:attribute name="type" use="required">
<xsd:simpleType>
<xsd:list>
<xsd:simpleType>
<xsd:restriction base="xsd:string">
<xsd:enumeration value="!"/>
<xsd:enumeration value="Oracle"/>
<xsd:enumeration value="MySQL"/>
<xsd:enumeration value="DB2"/>
<xsd:enumeration value="H2"/>
<xsd:enumeration value="SybaseASE"/>
<xsd:enumeration value="SybaseIQ"/>
</xsd:restriction>
</xsd:simpleType>
</xsd:list>
</xsd:simpleType>
</xsd:attribute>
</xsd:complexType>
</xsd:element>
</xsd:schema>

View Code
2、将扩展命名空间和XSD文件路径配置到ini文件中

# 用于配置sqlmapper的命名空间

# SqlMapper命名空间
[http://dysd.org/schema/sqlmapper]
schema = namespace/dysd-sqlmapper.xsd
parser = org.dysd.dao.mybatis.schema.SchemaSqlMapperNamespaceParser

# 扩展命名空间，因为不是根元素，所以不需要配置命名空间解析器
[http://dysd.org/schema/sqlmapper-extend]
schema = namespace/dysd-sqlmapper-extend.xsd

3、编写db元素的解析处理器，因为db元素即可作为一级子元素(statement)，也可作为脚本级元素(script)，因此我们实现这两种处理器：

//Statement级处理器实现类
public class DbStatementHandler extends StatementHandlerSupport{

@Override
public void handleStatementNode(Configuration configuration, SchemaSqlMapperParserDelegate delegate, XNode node) {
if(SchemaHandlers.isMatch(configuration.getDatabaseId(), node.getStringAttribute("type"))){
delegate.doParseStatements(node.getNode());
}
}
}

//Script级处理器实现类
public class DbScriptHandler extends ScriptHandlerSupport {

@Override
public void handleScriptNode(Configuration configuration, XNode node, List<SqlNode> targetContents) {
if(!SchemaHandlers.isMatch(configuration.getDatabaseId(), node.getStringAttribute("type"))){
List<SqlNode> contents = parseDynamicTags(configuration, node);
final MixedSqlNode mixedSqlNode = new MixedSqlNode(contents);
targetContents.add(new SqlNode(){
@Override
public boolean apply(DynamicContext context) {
mixedSqlNode.apply(context);
return true;
}
});
}
}
}

//SchemaHandlers中的帮助方法
public static boolean isMatch(String databaseId, String allowDatabaseIds){
if(!Tool.CHECK.isBlank(databaseId) && !Tool.CHECK.isBlank(allowDatabaseIds)){
String[] allows = allowDatabaseIds.split("\\s+");//使用空白字符分隔
boolean mode = !"!".equals(allows[0]);//是否肯定模式
if(mode){//肯定模式
for(int i = 0, l = allows.length; i < l; i++){
if(databaseId.equalsIgnoreCase(allows[i])){
return true;
}
}
}else{// 否定模式，从第1项开始匹配
for(int i = 1, l = allows.length; i < l; i++){
if(databaseId.equalsIgnoreCase(allows[i])){
return false;
}
}
return true;
}
}
return false;
}

4、在SchemaHandlers中注册实现类

// 注册自定义命名空间的处理器
registerExtend("db", new DbStatementHandler(), new DbScriptHandler());

5、修改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.schema.IExampleDao">

<select id="selectString" resultType="string">
select PARAM_NAME
from BF_PARAM_ENUM_DEF
<e:db type="MySQL">
<if test="null != paramName and '' != paramName">
where PARAM_NAME $like{#{paramName, jdbcType=VARCHAR}}
</if>
</e:db>
<e:db type="! MySQL">
<if test="null != paramName and '' != paramName">
where PARAM_CODE = 'DISPLAY_AREA'
</if>
</e:db>
</select>
</mapper>

需要注意，这里引入了两个命名空间。

6、执行测试

（1）H2

20161108 09:48:37,738 [main]-[DEBUG] ==>  Preparing: select PARAM_NAME from BF_PARAM_ENUM_DEF where PARAM_NAME LIKE '%'||?||'%'

（2）MySQL

20161108 09:50:36,453 [main]-[DEBUG] ==>  Preparing: select PARAM_NAME from BF_PARAM_ENUM_DEF where PARAM_CODE = 'DISPLAY_AREA'

从where条件可知已经实现预期的效果。

可能有人会说，其实根本不用麻烦，直接使用mybatis的<if>或<choose>，在其中判断_databaseId参数的值就可以。虽然mybatis可以实现这种情况，但需要开发人员记住_databaseId这个参数名，并且<if>或<choose>只能用作脚本级Script的元素使用，不能作为语句级Statement的元素使用；而且更重要的是，扩展提供了一种机制，上面的例子只是演示怎么引入自定义命名空间的元素，仅仅只是其中一个例子而已，开发人员尽可以发挥自己的聪明才智，自行扩展出更多实用的元素。