Mybatis JPA-整合方案+原始碼

阿新 • • 發佈：2019-02-18

原始碼地址(git)：https://github.com/LittleNewbie/mybatis-jpa

一、Mybatis簡介

mybatis中文官方文件：http://www.mybatis.org/mybatis-3/zh/index.html

簡介是為後面用到的內容做鋪墊，熟悉mybatis的朋友可以直接跳過，到第二章節。

關於mybatis-jpa的使用方式，請參見博文：http://www.cnblogs.com/svili/p/6828077.html

1.1 SqlSession

Mybatis中3個重要的概念：Configuration(容器)，SqlSessionFactory(工廠)，SqlSession；

相對於Spring中的applicationContext，BeanFactory，Bean。

不同之處在於SqlSession包含了所有的SQL方法，即這個SqlSession有且只有一個。SqlSession可以執行mybatis中註冊的所有方法。官方示例說明

<!-- SqlSession 完全包含了面向資料庫執行 SQL 命令所需的所有方法。
你可以通過 SqlSession 例項來直接執行已對映的 SQL 語句。例如：-->
SqlSession session = sqlSessionFactory.openSession();
try {
  Blog blog = (Blog) session.selectOne("org.mybatis.example.BlogMapper.selectBlog", 101);
} finally {
  session.close();
}

 
<!-- 對映器例項（Mapper Instances）-->
SqlSession session = sqlSessionFactory.openSession();
try {
  BlogMapper mapper = session.getMapper(BlogMapper.class);
  // do work
} finally {
  session.close();
}

1.2 Namespace

<!DOCTYPE mapper
    PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
    "http://mybatis.org/dtd/mybatis-3-mapper.dtd" 
>
<!--  xml中名稱空間與java中Mapper介面一致  -->
<mapper namespace="org.apache.ibatis.submitted.rounding.Mapper">

</mapper>

1.3 ResultType 和 ResultMap

ResultMap是myabtis最重要最強大的元素，是SQL列名columnName與POJO屬性名filedName的高階結果對映。對ResultMap不熟悉的朋友可以閱讀官方文件瞭解。

ResultType可以理解為mybatis自動對映生成的簡單的ResultMap，當POJO中的filedName與資料庫ColumnName不一致時，無法完成對映。

1.4 javaType和typeHandler

ResultMap中使用

<resultMap id="userResultMap" type="User">
  <!--  主鍵   -->
  <id property="id" column="user_id" />
  <!--  當column型別與field型別不一致時,需指定javaType或typeHandler,二者選其一即可。   -->
  <!--  簡單的型別轉換隻需指定javaType即可,需要邏輯處理的使用typeHandler  -->
  <result property="remark" column="remark" javaType="string"/>
  <result property="effective" column="is_effective" typeHandler="BooleanTypeHandler"/>
</resultMap>

1.5 jdbcType

官方說明：

jdbcType JDBC 型別是僅需要對插入,更新和刪除操作可能為空的列進行處理。這是 JDBC jdbcType 的需要,而不是 MyBatis 的。

NOTE：如果 null 被當作值來傳遞，對於所有可能為空的列，JDBC Type 是需要的。你可以自己通過閱讀預處理語句的 setNull() 方法的 JavaDocs 文件來研究這種情況。java.sql.PreparedStatement.setNull()

示例：

<insert id="insertAuthor">
  insert into Author
    (id, username)
  values
    <!--  請注意,當sql中引數可能為null時,需要指定jdbcType,不然會出錯  -->
    ('1', #{username,jdbcType=VARCHAR})
</insert>

1.6 MappedStatement

Mapper中的方法(方法簽名和可執行的sql語句)會被封裝為MappedStatement註冊到Configuration中。

詳見mybatis原始碼MapperBuilderAssistant.addMappedStatement(args);

二、Mybatis JPA

2.1 需求

1)首先，我們希望能夠與spring整合，使用spring的依賴注入。

2)其次，我們希望能夠相容spring-mybatis整合的程式碼，拒絕汙染。

3)解析註冊ResultMap和MappedStatement。

2.2 主線

1)參考spring data jpa，使用@RepositoryDefinition註解，標記需要自動生成sql的dao。

我們使用@MapperDefinition和@StatementDefinition註解，標記需要自動生成sql的dao和method。

這個是關鍵，既保證了不汙染原有程式碼，又可以使用spring-mybatis已經實現的依賴注入。

我們只需要在此基礎上，對特定註解標註的mapper類和方法做處理即可。

2)參考spring-mybatis

MapperScannerConfigurer，掃描mapper並註冊到mybatis Configuration中，繼而生成代理類。

MapperAnnotationBuilder實現java註解生成ResultMap和MappedStatement。

2.3 入口-MapperEnhancerScaner

在spring容器初始化後，對@MapperDefinition標註的mapper類進行掃描。

2.4 解析POJO 生成ResultMap

重點：columnName與fieldName對映，特殊欄位的jdbcType和typeHandler。

1)columnName與fieldName對映，使用JPA註解 @Cloumn即可，但是，我們希望能夠自動轉換駝峰與下劃線風格，即對於符合規範命名的，不需要註解，直接對映。參見：PersistentUtil,ColumnNameUtil。

 /**
     * 將駝峰標識轉換為下劃線
     * 
     * @param text
     * @return camel
     */
    public static String camelToUnderline(String text) {
        if (text == null || "".equals(text.trim())) {
            return "";
        }
        StringBuilder result = new StringBuilder(text.length() + 1);
        result.append(text.substring(0, 1));
        for (int i = 1; i < text.length(); i++) {
            if (!Character.isLowerCase(text.charAt(i))) {
                result.append('_');
            }
            result.append(text.substring(i, i + 1));
        }
        return result.toString().toLowerCase();
    }

    /**
     * 將下劃線標識轉換為駝峰
     * 
     * @param text
     * @return underline
     */
    public static String underlineToCamel(String text) {
        if (text == null || "".equals(text.trim())) {
            return "";
        }
        int length = text.length();
        StringBuilder result = new StringBuilder();
        for (int i = 0; i < length; i++) {
            char c = text.charAt(i);
            if (c == '_') {
                if (++i < length) {
                    result.append(Character.toUpperCase(text.charAt(i)));
                }
            } else {
                result.append(c);
            }
        }
        return result.toString();
    }

ColunNameUtil

2)jdbcType和typeHandler

處理了以下3種類型：POJO中的Enum,Boolean,以及資料庫中的CLOB，程式碼見MybatisColumnMeta。

需要強調說明的是，這裡為所有的field都聲明瞭jdbcType，是為了規避sql中引數為null時，產生異常。

/** meta resolver */
    private static class ColumnMetaResolver {

        public static String resolveJdbcAlias(Field field) {

            Class<?> fieldType = field.getType();
            if (field.getType().isEnum()) {
                if (field.isAnnotationPresent(Enumerated.class)) {
                    // 獲取註解物件
                    Enumerated enumerated = field.getAnnotation(Enumerated.class);
                    // 設定了value屬性
                    if (enumerated.value() == EnumType.ORDINAL) {
                        return "INTEGER";
                    }
                }
                return "VARCHAR";
            }
            if (field.isAnnotationPresent(Lob.class)) {
                if (String.class.equals(fieldType)) {
                    return "CLOB";
                }
            }
            if (Integer.class.equals(fieldType)) {
                return "INTEGER";
            }
            if (Double.class.equals(fieldType)) {
                return "DOUBLE";
            }
            if (Float.class.equals(fieldType)) {
                return "FLOAT";
            }
            if (String.class.equals(fieldType)) {
                return "VARCHAR";
            }
            // date型別需宣告
            if (java.util.Date.class.isAssignableFrom(fieldType)) {
                return "TIMESTAMP";
            }
            return null;
        }

        public static JdbcType resolveJdbcType(String alias) {
            if (alias == null) {
                return null;
            }
            try {
                return JdbcType.valueOf(alias);
            } catch (IllegalArgumentException e) {
                throw new BuilderException("Error resolving JdbcType. Cause: " + e, e);
            }
        }

        @SuppressWarnings("unchecked")
        public static Class<? extends TypeHandler<?>> resolveTypeHandler(Field field) {
            Class<? extends TypeHandler<?>> typeHandlerClass = null;
            if (field.getType().isEnum()) {
                typeHandlerClass = (Class<? extends TypeHandler<?>>) EnumTypeHandler.class;
                if (field.isAnnotationPresent(Enumerated.class)) {
                    // 獲取註解物件
                    Enumerated enumerated = field.getAnnotation(Enumerated.class);
                    // 設定了value屬性
                    if (enumerated.value() == EnumType.ORDINAL) {
                        typeHandlerClass = (Class<? extends TypeHandler<?>>) EnumOrdinalTypeHandler.class;
                    }
                }
            }

            if (field.getType().equals(Boolean.class)) {
                typeHandlerClass = (Class<? extends TypeHandler<?>>) BooleanTypeHandler.class;
            }
            return typeHandlerClass;
        }
    }

ColumnMetaResolver

3)ResultMap註冊

見ResultMapAdapter.parseResultMap(args);

2.5 MappedStatement註冊

分類處理，select需要用到ResultMap，預設為Pojo.getSimpleName() + "ResultMap";

insert和insertSelective的區別：在於null值的處理,假設column_1在資料庫設定了預設值,而引數中的field_1為null值,則insert 在資料庫寫入null,而insertSelective寫入資料庫預設值.

需要特別說明的是，動態SQL需要使用"<script></script>"標籤包圍。

對於各種sql方法的語句生成方法，詳見com.mybatis.jpa.statement.builder包下的類。

這裡以InsertSelective和select為例

public class InsertSelectiveBuilder implements StatementBuildable {

    @Override
    public String buildSQL(PersistentMeta persistentMeta, Method method) {
        // columns
        StringBuilder columns = new StringBuilder();
        columns.append("<trim prefix='(' suffix=')' suffixOverrides=',' > ");
        // values
        StringBuilder values = new StringBuilder();
        values.append("<trim prefix='(' suffix=')' suffixOverrides=',' > ");
        for (MybatisColumnMeta columnMeta : persistentMeta.getColumnMetaMap().values()) {
            // columns
            columns.append("<if test='" + columnMeta.getProperty() + "!= null'> ");
            columns.append(columnMeta.getColumnName() + ", ");
            columns.append("</if> ");
            // values
            values.append("<if test='" + columnMeta.getProperty() + "!= null'> ");
            values.append(SqlAssistant.resolveSqlParameter(columnMeta) + ", ");
            values.append("</if> ");
        }

        columns.append("</trim> ");
        values.append("</trim> ");

        return "<script>" + "INSERT INTO " + persistentMeta.getTableName() + columns.toString() + " VALUES "
                + values.toString() + "</script>";
    }

    @Override
    public void parseStatement(MybatisStatementAdapter adapter, PersistentMeta persistentMeta, Method method) {
        // 方法名
        adapter.setMethodName(method.getName());
        // 引數型別
        adapter.setParameterTypeClass(persistentMeta.getType());
        // sqlScript
        adapter.setSqlScript(buildSQL(persistentMeta, method));
        // 返回值型別
        adapter.setResultType(int.class);
        adapter.setResultMapId(null);

        adapter.setSqlCommandType(SqlCommandType.INSERT);

        // 主鍵策略
        adapter.setKeyGenerator(new NoKeyGenerator());
        adapter.setKeyProperty(null);
        adapter.setKeyColumn(null);

        adapter.parseStatement();

    }

}

InsertSelectiveBuilder

public class SelectBuilder implements StatementBuildable {

    @Override
    public String buildSQL(PersistentMeta persistentMeta, Method method) {
        return "SELECT " + persistentMeta.getColumnNames() + " FROM " + persistentMeta.getTableName()
                + SqlAssistant.buildSingleCondition(method, persistentMeta);
    }

    @Override
    public void parseStatement(MybatisStatementAdapter adapter, PersistentMeta persistentMeta, Method method) {
        // 方法名
        adapter.setMethodName(method.getName());
        // 引數型別
        if (method.getParameterTypes().length > 0) {
            // Mybatis mapper 方法最多支援一個引數,先設定成Object.class,mybatis會在sql中解析
            adapter.setParameterTypeClass(Object.class);
        } else {
            adapter.setParameterTypeClass(void.class);
        }

        String orderBy = " ";

        if (method.isAnnotationPresent(OrderBy.class)) {
            orderBy = " order by " + method.getAnnotation(OrderBy.class).value();
        }

        // sqlScript
        adapter.setSqlScript(buildSQL(persistentMeta, method) + orderBy);
        // 返回值型別
        adapter.setResultType(persistentMeta.getType());
        adapter.setResultMapId(persistentMeta.getType().getSimpleName() + "ResultMap");

        adapter.setSqlCommandType(SqlCommandType.SELECT);

        // 主鍵策略
        adapter.setKeyGenerator(new NoKeyGenerator());
        adapter.setKeyProperty(null);
        adapter.setKeyColumn(null);

        adapter.parseStatement();

    }

SelectBuilder

ok，以上就是mybatis-jpa的主要設計思路了，具體的細節，我已經儘可能的在程式碼中增加註釋。

關於mybatis-jpa的程式碼構建使用方式，請參見博文：http://www.cnblogs.com/svili/p/6828077.html

由於個人能力有限，程式碼可能有些簡陋，如有不妥之處，歡迎指正交流。

Mybatis JPA-整合方案+原始碼

一、Mybatis簡介

1.1 SqlSession

1.2 Namespace

1.3 ResultType 和 ResultMap

1.4 javaType和typeHandler

1.5 jdbcType

1.6 MappedStatement

二、Mybatis JPA

2.1 需求

2.2 主線

2.3 入口-MapperEnhancerScaner

2.4 解析POJO 生成ResultMap

2.5 MappedStatement註冊

Mybatis JPA-整合方案+原始碼

Spring整合Mybatis錯誤解決方案

SSM框架整合方案（Spring+SpringMVC+Mybatis）

MyBatis Spring 整合,mapper介面@repository有時候卻不用寫的原因(MyBatis Spring 整合原始碼解析)

mybatis+spring mvc 完美整合方案查詢，保存，更新，刪除自動生成

Mybatis JPA

多工程：基於Maven的SSM（Spring,SpringMvc,Mybatis）整合的web工程(中)

SSM框架Spring+SpringMVC+MyBatis——詳細整合教程

MyBatis框架整合

Spring+Mybatis+Druid 整合Demo

spring-mybatis-springMVC 整合

springboot+mybatis+springmvc整合實例

Springboot與MyBatis簡單整合

mybatis 不整合spring 入門小例子

SpringCloud+MyBatis+Redis整合—— 超詳細實例（二）

SpringBoot Kafka 整合例項原始碼

終極整合SpringMvc+Spring+MyBatis+Maven整合視訊教程

Spring+Mybatis+SpringMVC整合

mybatis-jpa外掛使用教程

SSM(SpringMVC Spring Mybatis)框架整合搭建

Mybatis JPA-整合方案+原始碼

一、Mybatis簡介

1.1 SqlSession

1.2 Namespace

1.3 ResultType 和 ResultMap

1.4 javaType和typeHandler

1.5 jdbcType

1.6 MappedStatement

二、Mybatis JPA

2.1 需求

2.2 主線

2.3 入口-MapperEnhancerScaner

2.4 解析POJO 生成ResultMap

2.5 MappedStatement註冊

相關推薦