關於mybatis plus 中的查詢優化問題

Posted on by

傳統的mybatis plus 查詢寫法

對於常規的mybatis單表查詢，我們既可以采用LambdaQueryWrapper查詢，也可以使用QueryWrapper查詢。
LambdaQueryWrapper具有防誤寫、規范代碼等好處，但是缺點是無法在復雜的多表查詢中使用。
相比較來說，使用QueryWrapper編寫查詢更加靈活，可以適應更復雜的查詢場景。
我們首先看一個QueryWrapper查詢的例子

public List<UserMo> list (UserForm userForm) {
 QueryWrapper<UserMo> queryWrapper = new QueryWrapper<>();
 queryWrapper.like(StringTool.isNotEmpty(userForm.getUserName(), "name", userForm.getUserName());
 queryWrapper.eq(StringTool.isNotEmpty(userForm.getMobile(), "mobile", userForm.getMobile());
 // 其它的查詢條件... 
 return userMapper.selectList(queryWrapper);
}

對於上面的查詢語句來說，可以很好的對前端傳值進行處理，當userForm中有前端傳值的話，就會往SQL語句中加一條where條件。
但是這樣做的話會有一個相對來說比較復雜的點，那就是當UserForm中的字段過於多的時候，我們也許得寫十幾行的這種重復判斷的語句。  

通過自定義註解來解決通用查詢條件過多問題

通過觀察mybatis plus 對於queryWrapper相關查詢方法的列子，我們可以找出一類通用方法

可以看出來這幾個方法都是傳的同樣的三個參數。
我想對於這些簡單的通用的查詢條件，也許可以有一個通用的方法來填充。
我首先設置瞭一個枚舉類，將這些查詢條件列出來,並在構造方法中，將對應的方法以反射的方式取到。 

public enum QueryConditionEnum {
 
 EQ("eq"),
 NE("ne"),
 GT("gt"),
 GE("ge"),
 LT("lt"),
 LE("le"),
 LIKE("like"),
 NOT_LIKE("notLike"),
 LIKE_LEFT("likeLeft"),
 LIKE_RIGHT("likeRight");
 
 private String name;
 
 private Method method;
 
 
 QueryConditionEnum (String name) {
  this.name = name;
  try {
   Method method = AbstractWrapper.class.getDeclaredMethod(name, boolean.class, Object.class, Object.class);
   this.method = method;
  } catch (NoSuchMethodException e) {
  }
 }
 
}

再者，我想通過註解的方式來規定需要以什麼方法填充，默認為EQ,對此寫瞭一個QueryCondition註解。  

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD})
public @interface QueryCondition {
 
 /**
  * 默認查詢方式
  * 
  * @return
  */
 QueryConditionEnum value() default QueryConditionEnum.EQ;
 
 /**
  * 是否填充默認查詢條件
  * 
  * @return
  */
 boolean isCondition() default true;
 
}

然後就可以這樣構造UserForm  

public class UserForm {
 private String name;
 
 @QueryCondition(QueryConditionEnum.LIKE)
 private String mobile;
}

我們需要一個工具類填充查詢條件,這裡我們新增瞭一個參數 mo對象，這是因為我們的主查詢對象是Mo對象，Mo對象存儲瞭相關表格名稱、表格字段名信息。  

@TableName("user")
public class UserMo {
 @TableField("name")
 private String name;
 
 @TableField("mobile")
 private String mobile;
}
public class QueryTool {
 
 /**
  * 填充默認查詢
  * @param baseClazz mo對象class
  * @param queryWrapper 查詢條件
  * @param form 請求對象
  */
 public static void paddingDefaultConditionQuery(Class baseClazz, QueryWrapper queryWrapper, Object form) {
  try {
   for (Field declaredField : form.getClass().getDeclaredFields()) {
    declaredField.setAccessible(true);
    Object fieldValue = declaredField.get(form);
    QueryCondition queryCondition = declaredField.getAnnotation(QueryCondition.class);
    if (fieldValue == null) {
     continue;
    }
    if (queryCondition == null) {
     queryWrapper.eq(StringTool.isNotEmpty(fieldValue.toString()),
      QueryTool.getTableName(baseClazz) + "." + QueryTool.getTableFieldName(baseClazz, declaredField),
      fieldValue.toString());
     continue;
    }
    if (queryCondition.isCondition() == false) {
     continue;
    }
    Method method = queryCondition.value().getMethod();
    method.invoke(queryWrapper, StringTool.isNotEmpty(fieldValue.toString()),
     QueryTool.getTableName(baseClazz) + "." + QueryTool.getTableFieldName(baseClazz, declaredField),
     fieldValue.toString());
   }
  } catch (Exception e) {
   throw new RuntimeException("填充默認的SQL條件出錯", e);
  }
 }
 
 /**
  * 填充默認排序
  * 
  * @param queryWrapper
  * @param pageForm
  */
 public static void paddingDefaultOrderQuery(QueryWrapper queryWrapper, PageForm pageForm) {
  queryWrapper.orderBy(pageForm != null && StringTool.isNotEmpty(pageForm.getColumnName()),
   pageForm.getIsAsc() == null ? false : pageForm.getIsAsc(), pageForm.getColumnName());
 }
 
 /**
  * 獲取表名稱
  *
  * @return
  */
 public static String getTableName(Class baseClazz) {
  TableName tableName = (TableName) baseClazz.getDeclaredAnnotation(TableName.class);
  if (tableName != null && StringTool.isNotEmpty(tableName.value())) {
   return tableName.value();
  }
  return StringTool.toUnderline(baseClazz.getClass().getName());
 }
 
 /**
  * 獲取字段名
  * 
  * @param field
  * @return
  */
 public static String getTableFieldName(Class baseClazz, Field field) {
  Field baseField = null;
  try {
   baseField = baseClazz.getDeclaredField(field.getName());
  } catch (NoSuchFieldException e) {
   e.printStackTrace();
  }
  if (baseField == null) {
   baseField = field;
  }
  TableId tableId = baseField.getAnnotation(TableId.class);
  if (tableId != null && StringTool.isNotEmpty(tableId.value())) {
   return tableId.value();
  }
  TableField tableField = baseField.getAnnotation(TableField.class);
  if (tableField != null && StringTool.isNotEmpty(tableField.value())) {
   return tableField.value();
  }
  return StringTool.toUnderline(baseField.getName());
 }
 
}

最後我們就可以使用工具類來填充瞭 。

public List<UserMo> list (UserForm userForm) {
 QueryWrapper<UserMo> queryWrapper = new QueryWrapper<>();
 QueryTool.paddingDefaultConditionQuery(UserMo.class, queryWrapper, userForm);
 return userMapper.selectList(queryWrapper);
}

可以看到這樣大大減少瞭需要填充的字段。如果有特殊字段，也能通過註解方式，跳過特殊字段，再自行填充就好。

到此這篇關於關於mybatis plus 中的查詢優化的文章就介紹到這瞭,更多相關mybatis plus 查詢優化內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：