tk.mybatis擴(kuò)展通用接口使用詳解

發(fā)布時(shí)間：2020-10-24 09:40:20 來源：腳本之家閱讀：336 作者：夢飛翔up 欄目：編程語言

一.tk.mybatis已經(jīng)為我們封裝好了許多拆箱即用的通用mapper,但在實(shí)際的項(xiàng)目開發(fā)中想必不少小伙伴在數(shù)據(jù)庫設(shè)計(jì)中都會采用邏輯刪除這種方案，再去使用通用的mapper接口就不行了。

這時(shí)候就需要我們封裝一些擴(kuò)展的通用Mapper接口。

二.項(xiàng)目中提供了大量現(xiàn)成的方法，這些方法可以作為擴(kuò)展時(shí)的參考。

例如 selectAll 方法。

首先定義接口：

@RegisterMapper
public interface SelectAllMapper<T> {
  /**
   * 查詢?nèi)拷Y(jié)果
   *
   * @return
   */
  @SelectProvider(type = MySelectProvider.class, method = "dynamicSQL")
  List<T> selectAll();
}

其中 MySelectProvider 是你要實(shí)現(xiàn)的一個(gè)類，該類需要繼承 MapperTemplate。@RegisterMapper 注解可以避免 mappers 參數(shù)配置，通用 Mapper 檢測到該接口被繼承時(shí)，會自動注冊。

import org.apache.ibatis.mapping.MappedStatement;
import tk.mybatis.mapper.mapperhelper.MapperHelper;
import tk.mybatis.mapper.mapperhelper.MapperTemplate;
import tk.mybatis.mapper.mapperhelper.SqlHelper;
public class MySelectProvider extends MapperTemplate {
  public BaseSelectProvider(Class<?> mapperClass, MapperHelper mapperHelper) {
    super(mapperClass, mapperHelper);
  } 
  /**
   * 查詢?nèi)拷Y(jié)果
   *
   * @param ms
   * @return
   */
  public String selectAll(MappedStatement ms) {
    final Class<?> entityClass = getEntityClass(ms);
    //修改返回值類型為實(shí)體類型
    setResultType(ms, entityClass);
    StringBuilder sql = new StringBuilder();
    sql.append(SqlHelper.selectAllColumns(entityClass));
    sql.append(SqlHelper.fromTable(entityClass, tableName(entityClass)));
    sql.append(SqlHelper.orderByDefault(entityClass));
    return sql.toString();
  }
}

其中 selectAll 方法名要和接口中定義的方法名一致。其次就是該方法的參數(shù)為 MappedStatement類型。

在 selectAll 方法中，首先是獲取了當(dāng)前接口的實(shí)體類型：

final Class<?> entityClass = getEntityClass(ms);

因?yàn)榻涌诜祷刂殿愋蜑?List<T>，MyBatis 會認(rèn)為返回值類型為 List<Object>，這和我們想要的實(shí)體類型不一樣，所以下一行代碼就是設(shè)置返回值類型：

setResultType(ms, entityClass);

注意，只有返回 T 或者 List 時(shí)需要設(shè)置，返回 int 類型時(shí)不需要設(shè)置。

接下來就是純粹的拼接 XML 形式的 SQL 了。

/select col1,col2...
sql.append(SqlHelper.selectAllColumns(entityClass));
//from tablename - 支持動態(tài)表名
sql.append(SqlHelper.fromTable(entityClass, tableName(entityClass)));
//order by xxx
sql.append(SqlHelper.orderByDefault(entityClass));

當(dāng)你想要實(shí)現(xiàn)某種方法時(shí)，可以從已有的例子中找一個(gè)最接近的方法，在此基礎(chǔ)上進(jìn)行修改

三.例：根據(jù)主鍵查詢單個(gè)實(shí)體對象（過濾掉邏輯刪除的實(shí)體，注：我的數(shù)據(jù)表邏輯刪除字段定義為enabled_status）

首先定義mapper

@RegisterMapper
public interface SelectByKeyAndNotDeletedMapper<T> {
  /**
   * 根據(jù)主鍵查詢沒有被邏輯刪除的實(shí)體
   *
   * @return
   */
  @SelectProvider(type = SelectByKeyNotDeletedProvider.class, method = "dynamicSQL")
  T selectByKeyNotDeleted(Object key);
}

其次定義SelectByKeyNotDeletedProvider

public class SelectByKeyNotDeletedProvider extends MapperTemplate {
  public SelectByKeyNotDeletedProvider(Class<?> mapperClass, MapperHelper mapperHelper) {
    super(mapperClass, mapperHelper);
  }

  public String selectByKeyNotDeleted(MappedStatement ms) {
    final Class<?> entityClass = getEntityClass(ms);
    //將返回值修改為實(shí)體類型
    setResultType(ms, entityClass);
    StringBuilder sql = new StringBuilder();
    sql.append(SqlHelper.selectAllColumns(entityClass));
    sql.append(SqlHelper.fromTable(entityClass, tableName(entityClass)));
    sql.append(wherePKColumns(entityClass, false));
    return sql.toString();
  }
  private String wherePKColumns(Class<?> entityClass, boolean useVersion) {
    StringBuilder sql = new StringBuilder();
    sql.append("<where>");
    //獲取全部列
    Set<EntityColumn> columnSet = EntityHelper.getPKColumns(entityClass);
    //當(dāng)某個(gè)列有主鍵策略時(shí)，不需要考慮他的屬性是否為空，因?yàn)槿绻麨榭?，一定會根?jù)主鍵策略給他生成一個(gè)值
    for (EntityColumn column : columnSet) {
      sql.append(" AND " + column.getColumnEqualsHolder());
    }
    if (useVersion) {
      sql.append(whereVersion(entityClass));
    }
    //過濾被邏輯刪除的數(shù)據(jù)
    sql.append(" AND enabled_status = 1 ");
    sql.append("</where>");
    return sql.toString();
  }
}

然后定義BasicMapper,讓其繼承通用Mapper接口以及上面自定義的SelectByKeyAndNotDeletedMapper接口。

@tk.mybatis.mapper.annotation.RegisterMapper public interface BasicMapper<T> extends Mapper<T>, SelectByKeyAndNotDeletedMapper<T> { }

最后在通用service中引入，

@Autowired private BasicMapper<T> mapper;

自定義*.Mapper都去繼承BasicMapper<T>即可

public interface UserPOMapper extends BasicMapper<UserPO> {}

注意：最新版本的tk已經(jīng)支持根據(jù)注解@LogicDelete實(shí)現(xiàn)邏輯刪除

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

tk.mybatis擴(kuò)展通用接口使用詳解

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽