Mybatis 實(shí)現(xiàn)動(dòng)態(tài)組裝查詢條件,仿SQL模式
以前比較習(xí)慣使用Hibernate，后來(lái)覺(jué)得mybatis不能按我想要的自動(dòng)組裝為SQL查詢條件，所以提供該工具類；
效果圖：
如圖所示，根據(jù)條件自動(dòng)組裝查詢條件，下面來(lái)說(shuō)一下實(shí)現(xiàn)方法：
1. ServiceImpl書(shū)寫注意項(xiàng)Page<SysLogin> resultPage = null;try { PageHelper.startPage(pager.getCurrentPage(), pager.getPageSize()); // 判斷是否有分頁(yè) if (ObjectHelper.isNotEmpty(pager.getDirection()) && ObjectHelper.isNotEmpty(pager.getProperties())) {specification.addOrderBy(pager.getProperties(),pager.getDirection()); } // 判斷是否存在邏輯刪除篩選 String sqlStr = specification.sql(); if (sqlStr.indexOf('deleted') == -1) {specification.eq('deleted', '0'); } resultPage = this.sysLoginMapper.page(specification.sql());} catch (Exception e) { result = Result.newFailure('數(shù)據(jù)錯(cuò)誤', '在獲取分頁(yè)列表時(shí)發(fā)生異常。'); log.error(SimpleLogFormater.formatException(result.getMessage(), e)); return result;}2. Mapper.java 書(shū)寫
查詢條件非Map對(duì)象，直接就是SQL語(yǔ)句了；
/** * 分頁(yè)查詢數(shù)據(jù) * * @return */ Page<T> page(String sqlStr);3. 關(guān)于XML的配置，會(huì)拼裝SQL語(yǔ)句
附：SQL拼裝工具類
/** * @Description: (用一句話描述該文件做什么) * @author heliang * @date 2018-7-6 下午6:43:42 * @version V2.1 */package com.onem2.base.common;import com.onem2.base.helper.ObjectHelper; /** * @ClassName: Specification * @Description: (這里用一句話描述這個(gè)類的作用) * @author heliang * @date 2018-7-6 下午6:43:42 * @version V2.1 * Update Logs: * Name: * Date: * Description: 初始化 */ public class Specification { private StringBuilder where = new StringBuilder(); private String groupBy; private String having; private String orderBy; public StringBuilder getWhere() {return where; } public void setWhere(StringBuilder where) {this.where = where; } public String getGroupBy() {return groupBy; } public void setGroupBy(String groupBy) {this.groupBy = groupBy; } public String getHaving() {return having; } public void setHaving(String having) {this.having = having; } public String getOrderBy() {return orderBy; } public void setOrderBy(String orderBy) {this.orderBy = orderBy; } public Specification addOrderBy(String sort, String order) {if (!isEmpty(sort) && !isEmpty(order)) { this.orderBy = ObjectHelper.underscoreName(sort) + ' ' + order;}return this; } public Specification orLike(String value, String columns) {if (!isEmpty(value)) { StringBuffer strBuf = new StringBuffer(''); for (String column : columns.split(',')) {strBuf.append(ObjectHelper.underscoreName(column) + ' like ’%'+ value + '%’ or '); } String orLikeStr = strBuf.substring(0, strBuf.lastIndexOf('or')); where.append(' and (' + orLikeStr + ')');}return this; } public Specification eq(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' = ’' + sqlParam(value) + '’');}return this; } public Specification ne(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' != ’' + sqlParam(value) + '’');}return this; } public Specification like(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' like ’%' + sqlParam(value) + '%’');}return this; } public Specification notLike(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' not like ’%' + sqlParam(value) + '%’');}return this; } public Specification in(String column, String... values) {if (!isEmpty(values)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' in (' + inValuesString(values) + ')');}return this; } public Specification notIn(String column, String... values) {if (!isEmpty(values)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' not in (' + inValuesString(values) + ')');}return this; } public Specification gt(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' > ’' + sqlParam(value) + '’');}return this; } public Specification gte(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' >= ’' + sqlParam(value) + '’');}return this; } public Specification lt(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' < ’' + sqlParam(value) + '’');}return this; } public Specification lte(String column, String value) {if (!isEmpty(value)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' <= ’' + sqlParam(value) + '’');}return this; } public Specification between(String column, String from, String to) {if (isEmpty(from) && isEmpty(to)) { return this;}if (isEmpty(to)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' >= ’' + sqlParam(from) + '’');} else if (isEmpty(from)) { where.append(' and ' + ObjectHelper.underscoreName(column) + ' <= ’' + sqlParam(to) + '’');} else { where.append(' and ' + ObjectHelper.underscoreName(column) + ' between ’' + sqlParam(from) + '’ and ’' + sqlParam(to) + '’');}return this; } public String sql() {StringBuilder sql = new StringBuilder('');final int a = 4;final int b = 5;if (where.length() > a) { sql.append(' ' + where.substring(b));}if (!isEmpty(groupBy)) { sql.append(' group by ' + groupBy);}if (!isEmpty(having)) { sql.append(' having ' + having);}if (!isEmpty(orderBy)) { sql.append(' order by ' + orderBy);}return sql.toString(); } public String toString() {return sql(); } private static boolean isEmpty(String value) {return value == null || ''.equals(value) || value.trim().length() == 0; } private static boolean isEmpty(String[] values) {if (values == null || values.length == 0) { return true;}for (String value : values) { if (!isEmpty(value)) {return false; }}return true; } private static String inValuesString(String[] values) {StringBuilder string = new StringBuilder();for (String value : values) { if (isEmpty(value)) {continue; } string.append(’’’); string.append(value); string.append(’’’); string.append(’,’);}if (string.length() > 0) { string.deleteCharAt(string.length() - 1);}return string.toString(); } private static String sqlParam(String sqlParam) {return sqlParam.replaceAll('([’;]+|(--)+)', ''); }}
附：ObjectHelper 工具源碼:
package com.onem2.base.helper;import java.util.ArrayList;import java.util.Collection;import java.util.List;/** * Object幫助類 功能：此類提供處理 <Object>對(duì)象一系列方法 * * @author 賀亮 * */public class ObjectHelper { /** * 將id數(shù)組轉(zhuǎn)換為id集合 * * @param ids * @return */ public static List<Long> initIds(String[] ids) {List<Long> list = new ArrayList<Long>();list.add(-1L);for (int i = 0; i < ids.length; i++) { list.add(Long.valueOf(ids[i]));}return list; } /** * 組裝條件 * * @param str * @return */ public static List<String> strToList(String str) {if (isEmpty(str)) { return null;}String[] strs = str.split(',');List<String> list = new ArrayList<String>();for (int i = 0; i < strs.length; i++) { list.add(strs[i]);}return list; } /** * 判斷這個(gè)Object是否為Null或長(zhǎng)度為0 * * @param obj * @return */ public static boolean isEmpty(Object obj) {if (obj == null) { return true;}if (obj instanceof Collection) { return ((Collection<?>) obj).isEmpty();} if (obj instanceof String) { return ((String) obj).equalsIgnoreCase('null') | ((String) obj).trim().toString().equals('');} if (obj instanceof StringBuffer) { return ((StringBuffer) obj).length() == 0;} if (obj.getClass().isArray()) { try {Object[] a = (Object[]) obj; boolean b = true;for (Object o : a) { b = b & isEmpty(o); if (!b) {break; }} return b; } catch (ClassCastException e) { }}return false; } /** * 判斷這個(gè)Object是否不為Null或長(zhǎng)度不為0 * * @param obj * @return */ public static boolean isNotEmpty(Object obj) {return !isEmpty(obj); } /** * 返回首字母大寫單詞 * * @param str * @return */ public static String lcyFirstLetterToUpper(String str) {return str.replaceFirst(str.substring(0, 1), str.substring(0, 1).toUpperCase()); } /** * 轉(zhuǎn)換為下劃線 * * @param camelCaseName * @return */ public static String underscoreName(String camelCaseName) {StringBuilder result = new StringBuilder();if (camelCaseName != null && camelCaseName.length() > 0) { result.append(camelCaseName.substring(0, 1).toLowerCase()); for (int i = 1; i < camelCaseName.length(); i++) {char ch = camelCaseName.charAt(i);if (Character.isUpperCase(ch)) { result.append('_'); result.append(Character.toLowerCase(ch));} else { result.append(ch);} }}return result.toString(); } /** * 轉(zhuǎn)換為駝峰 * * @param underscoreName * @return */ public static String camelCaseName(String underscoreName) {StringBuilder result = new StringBuilder();if (underscoreName != null && underscoreName.length() > 0) { boolean flag = false; for (int i = 0; i < underscoreName.length(); i++) {char ch = underscoreName.charAt(i);if ('_'.charAt(0) == ch) { flag = true;} else { if (flag) {result.append(Character.toUpperCase(ch));flag = false; } else {result.append(ch); }} }}return result.toString(); } public static void main(String[] args) {System.out.println(underscoreName('loginName')); }}
這樣就可以做到動(dòng)態(tài)生成查詢條件，復(fù)雜的查詢條件也不會(huì)去改動(dòng)XML配置了。
mybatis原理:參數(shù)解析與SQL動(dòng)態(tài)組裝過(guò)程mybatis執(zhí)行sql之前， 需要經(jīng)過(guò)參數(shù)解析、sql動(dòng)態(tài)組裝等過(guò)程，本文主要聊聊mybatis的：
（1）參數(shù)解析原理及其過(guò)程
（2）sql動(dòng)態(tài)組裝原理及其過(guò)程
一、數(shù)據(jù)準(zhǔn)備1.實(shí)體類，省略了set、get方法
public class User { private String id; private String username; private String password; private Integer isValid;}
2.mapper接口UserMapper，可以看作是一個(gè)根據(jù)用戶名和密碼的登錄接口
User getUserByUsernameAndPassword(@Param('name') String username, @Param('pwd') String password);
3.mapper映射
<select resultType='com.qxf.pojo.User'>select id,username,password,is_valid as isValid from t_user<where> <if test='name != null and name != ’’'>username = #{name} </if> <if test='pwd != null and pwd != ’’'>and password = #{pwd} </if></where> </select>
4.測(cè)試，mybatis-config.xml配置文件按一般配置即可，這里就不貼代碼了
//讀取配置信息InputStream inputStream = Resources.getResourceAsStream('mybatis-config.xml');//根據(jù)配置信息，創(chuàng)建SqlSession工廠SqlSessionFactory factory = new SqlSessionFactoryBuilder().build(inputStream);//SqlSession工廠創(chuàng)建SqlSessionSqlSession sqlSession = factory.openSession();//獲取接口的代理對(duì)象UserMapper mapper = sqlSession.getMapper(UserMapper.class);//執(zhí)行相應(yīng)的接口方法User user = mapper.getUserByUsernameAndPassword('張三2', null);System.out.println(user);
下面將以這句代碼為入口：
（注意，這里只是為了測(cè)試，給密碼參數(shù)傳遞了null，正常情況不會(huì)這樣傳遞參數(shù)的，不然結(jié)果返回一個(gè)List集合就會(huì)報(bào)錯(cuò)的）
//執(zhí)行相應(yīng)的接口方法User user = mapper.getUserByUsernameAndPassword('張三2', null);二、參數(shù)解析原理及其過(guò)程
首先要明白一點(diǎn)，返回的是mapper接口的代理對(duì)象，所以會(huì)來(lái)到MapperProxy的invoke方法
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {try { // Object對(duì)象的方法，則直接執(zhí)行 if (Object.class.equals(method.getDeclaringClass())) {return method.invoke(this, args); } if (method.isDefault()) {return this.invokeDefaultMethod(proxy, method, args); }} catch (Throwable var5) { throw ExceptionUtil.unwrapThrowable(var5);}// 獲取mapperMethod，這里面就會(huì)進(jìn)行參數(shù)解析MapperMethod mapperMethod = this.cachedMapperMethod(method);// 執(zhí)行方法return mapperMethod.execute(this.sqlSession, args); }
重點(diǎn)關(guān)注這句：
// 獲取mapperMethod，這里面就會(huì)進(jìn)行參數(shù)解析MapperMethod mapperMethod = this.cachedMapperMethod(method);
參數(shù)的解析可以分成兩部：
（1）形參的解析
（2）實(shí)參的封裝
（1）形成的解析
一路跟進(jìn)去，最終會(huì)來(lái)到 ParamNameResolver，暫且叫做參數(shù)名稱解析器吧，首先會(huì)在構(gòu)造器組裝參數(shù)的位置和名稱的對(duì)應(yīng)關(guān)系，如果我們使用了@Param注解，則會(huì)使用我們定義的名稱，否則會(huì)使用arg0、arg1....依次替代，詳細(xì)代碼如下：
public ParamNameResolver(Configuration config, Method method) {// 獲取參數(shù)列表中，每一個(gè)參數(shù)的類型Class<?>[] paramTypes = method.getParameterTypes();// 獲取參數(shù)注解，因?yàn)槊總€(gè)參數(shù)可能有多個(gè)注解，所以是二維數(shù)組Annotation[][] paramAnnotations = method.getParameterAnnotations();// 存放結(jié)果的mapSortedMap<Integer, String> map = new TreeMap();// 參數(shù)個(gè)數(shù)int paramCount = paramAnnotations.length; for(int paramIndex = 0; paramIndex < paramCount; ++paramIndex) { if (!isSpecialParameter(paramTypes[paramIndex])) {// 參數(shù)名稱String name = null;// 參數(shù)的注解數(shù)組Annotation[] var9 = paramAnnotations[paramIndex];// 參數(shù)注解的個(gè)數(shù)int var10 = var9.length;// 遍歷每個(gè)注解，找到Param注解，拿到value作為參數(shù)名稱for(int var11 = 0; var11 < var10; ++var11) { Annotation annotation = var9[var11]; if (annotation instanceof Param) {this.hasParamAnnotation = true;name = ((Param)annotation).value();break; }} if (name == null) { if (config.isUseActualParamName()) {name = this.getActualParamName(method, paramIndex); } if (name == null) {name = String.valueOf(map.size()); }}// 參數(shù)序號(hào)作為key，從0開(kāi)始，參數(shù)名稱作為值map.put(paramIndex, name); }}// 沒(méi)有做什么，再一次封裝而已this.names = Collections.unmodifiableSortedMap(map); }
結(jié)果是這樣的：符合我們的預(yù)期的
（2）實(shí)參的封裝
然后會(huì)來(lái)到getNamedParams方法對(duì)參數(shù)進(jìn)一步的封裝：
public Object getNamedParams(Object[] args) {// 參數(shù)個(gè)數(shù)，這個(gè)names就是上面解析后的map，key是從0開(kāi)始的參數(shù)序號(hào)，value是參數(shù)名稱int paramCount = this.names.size();// 這里的args便是實(shí)參列表// 實(shí)參不為空，形參個(gè)數(shù)不為0if (args != null && paramCount != 0) { if (!this.hasParamAnnotation && paramCount == 1) {// 沒(méi)有使用@Param注解，并且只有一個(gè)參數(shù)return args[(Integer)this.names.firstKey()]; } else {// 將參數(shù)封裝成一個(gè)mapMap<String, Object> param = new ParamMap();int i = 0;// 對(duì)形參循環(huán)迭代for(Iterator var5 = this.names.entrySet().iterator(); var5.hasNext(); ++i) { Entry<Integer, String> entry = (Entry)var5.next(); // names中的參數(shù)名稱為key，值為實(shí)參值 param.put((String)entry.getValue(), args[(Integer)entry.getKey()]); // 并添加key為param1、param2之類的通用參數(shù) String genericParamName = 'param' + String.valueOf(i + 1); if (!this.names.containsValue(genericParamName)) {param.put(genericParamName, args[(Integer)entry.getKey()]); }} return param; }} else { return null;} }
通過(guò)源碼可以發(fā)現(xiàn)，
（1）如果只有一個(gè)參數(shù)，并且沒(méi)有使用@Param注解，就直接返回第一個(gè)參數(shù)
（2）有多個(gè)參數(shù)，則封裝成一個(gè)map，key為參數(shù)參數(shù)名稱，使用了@Param注解，名稱就是注解中的值，否則key為arg0、arg1這種類型，同時(shí)，一定含有key為param1、param2的參數(shù)，值就是傳入的值
封裝后的結(jié)果如下：
這樣就完成了參數(shù)的解析過(guò)程，總結(jié)一下：
（1）解析形參，判斷是否使用了@Param注解
（2）封裝實(shí)參，如果只有一個(gè)，并且沒(méi)有使用@Param注解，就直接返回第一個(gè)參數(shù)值，否則封裝成map
三、動(dòng)態(tài)組裝sql原理及其過(guò)程來(lái)到CachingExecutor的如下方法，作為入口：
public <E> List<E> query(MappedStatement ms, Object parameterObject, RowBounds rowBounds, ResultHandler resultHandler) throws SQLException {// 獲取組裝完成的sqlBoundSql boundSql = ms.getBoundSql(parameterObject);// 創(chuàng)建緩存keyCacheKey key = this.createCacheKey(ms, parameterObject, rowBounds, boundSql);// 執(zhí)行查詢r(jià)eturn this.query(ms, parameterObject, rowBounds, resultHandler, key, boundSql); }
重點(diǎn)看這句：
// 獲取組裝完成的sqlBoundSql boundSql = ms.getBoundSql(parameterObject);
一路跟進(jìn)去，來(lái)到DynamicSqlSource的getBoundSql方法：
public BoundSql getBoundSql(Object parameterObject) {// 將參數(shù)封裝成動(dòng)態(tài)上下文，DynamicContext中sqlBuilder就是最后組裝的sqlDynamicContext context = new DynamicContext(this.configuration, parameterObject);// 根據(jù)條件，動(dòng)態(tài)組裝sqlthis.rootSqlNode.apply(context);SqlSourceBuilder sqlSourceParser = new SqlSourceBuilder(this.configuration);Class<?> parameterType = parameterObject == null ? Object.class : parameterObject.getClass();// 將#{參數(shù)}替換為?SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType, context.getBindings());BoundSql boundSql = sqlSource.getBoundSql(parameterObject);Map var10000 = context.getBindings();Objects.requireNonNull(boundSql);var10000.forEach(boundSql::setAdditionalParameter);return boundSql; }
我們先看下這句：
// 根據(jù)條件，動(dòng)態(tài)組裝sqlthis.rootSqlNode.apply(context);
對(duì)于我們的sql：
<select resultType='com.qxf.pojo.User'>select id,username,password,is_valid as isValid from t_user<where> <if test='name != null and name != ’’'>username = #{name} </if> <if test='pwd != null and pwd != ’’'>and password = #{pwd} </if></where> </select>
每個(gè)標(biāo)簽都有對(duì)應(yīng)的SqlNode來(lái)處理，比如if標(biāo)簽，就由IfSqlNode來(lái)處理，where標(biāo)簽，則會(huì)通過(guò)TrimSqlNode來(lái)處理，SqlNode的具體實(shí)現(xiàn)類如下：
這里以IfSqlNode處理if標(biāo)簽為例：
這是就是兩步：
（1）判斷表達(dá)式的值是否為真，這里最終使用的是Ognl來(lái)判斷
（2）如果表達(dá)式的為真，就將標(biāo)簽內(nèi)容追加到sql中去
處理結(jié)果如下：
因?yàn)槊艽a的參數(shù)傳入為null，所以不會(huì)拼接密碼查詢條件，只拼接了用戶名查詢條件
然后是將#{參數(shù)}替換為?進(jìn)行占位：
// 將#{參數(shù)}替換為?SqlSource sqlSource = sqlSourceParser.parse(context.getSql(), parameterType, context.getBindings());
這個(gè)就比較簡(jiǎn)單了，可以自行看源碼，最終是這樣的：
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持好吧啦網(wǎng)。
網(wǎng)公網(wǎng)安備