解析Mybatis Porxy動(dòng)態(tài)代理和sql解析替換問(wèn)題

目錄

JDK常用核心原理

概述

在 Mybatis 中，常用的作用就是講數據庫中的表的字段映射為對象的屬性，在進(jìn)入Mybatis之前，原生的 JDBC 有幾個(gè)步驟：導入 JDBC 驅動(dòng)包，通過(guò) DriverManager 注冊驅動(dòng)，創(chuàng )建連接，創(chuàng )建 Statement，增刪改查，操作結果集，關(guān)閉連接

過(guò)程詳解

首先進(jìn)行類(lèi)的加載，通過(guò) DriverManager 注冊驅動(dòng)

Class.forName("com.mysql.jdbc.Driver");
Connection connection = DriverManager.getConnection("");

為什么在這里可以直接注冊進(jìn)去，com.mysql.jdbc.Driver 被加載到 Driver.class ，在 DriverManager 中，首先有一個(gè)靜態(tài)代碼塊來(lái)進(jìn)行初始化加載 Driver

static {
    loadInitialDrivers();
    println("JDBC DriverManager initialized");
}

通過(guò) loadInitialDrivers()，來(lái)加載 Driver，拿出 jdbc.drivers，通過(guò) ServiceLoader 讀取 Driver.class，讀取拿出 driver 和 所有迭代器，一直迭代

private static void loadInitialDrivers() {
    String drivers;
    // 訪(fǎng)問(wèn)修飾符，在這里把 jdbc.drivers 拿出來(lái)
    try {
        drivers = AccessController.doPrivileged(new PrivilegedAction<String>() {
            public String run() {
                return System.getProperty("jdbc.drivers");
            }
        });
    } catch (Exception ex) {
        drivers = null;
    }

    AccessController.doPrivileged(new PrivilegedAction<Void>() {
        public Void run() {
            // 讀取拿出 driver 和 所有迭代器
            ServiceLoader<Driver> loadedDrivers = ServiceLoader.load(Driver.class);
            Iterator<Driver> driversIterator = loadedDrivers.iterator();
            
            // 一直進(jìn)行迭代
            try{
                while(driversIterator.hasNext()) {
                    driversIterator.next();
                }
            } catch(Throwable t) {
            // Do nothing
            }
            return null;
        }
    });

    println("DriverManager.initialize: jdbc.drivers = " + drivers);

    if (drivers == null || drivers.equals("")) {
        return;
    }
    String[] driversList = drivers.split(":");
    println("number of Drivers:" + driversList.length);
    for (String aDriver : driversList) {
        try {
            println("DriverManager.Initialize: loading " + aDriver);
            Class.forName(aDriver, true,
                    ClassLoader.getSystemClassLoader());
        } catch (Exception ex) {
            println("DriverManager.Initialize: load failed: " + ex);
        }
    }
}

從 Driver 加載完后，就可以得到一個(gè)和數據庫的連接 connection ，connection 就可以創(chuàng )建一個(gè) Statement，Statement 就可以進(jìn)行執行 sql 語(yǔ)句，將結果返回一個(gè)結果集，獲取出來(lái)的結果集遍歷放進(jìn)一個(gè) List 集合中

Statement statement = connection.createStatement();
ResultSet resultSet = statement.executeQuery("select * from mybatis.user");
    while (resultSet.next()) {
      int id = resultSet.getInt(1);
      String username = resultSet.getString(2);
      list.add(new User(id,username));
}

在原生的 JDBC 直接操作中，繁雜的步驟在業(yè)務(wù)代碼中不會(huì )使用，而 Mybatis 可以在更好的便利度上使用

JDK動(dòng)態(tài)代理

sql語(yǔ)句解析替換

在 JDK 動(dòng)態(tài)代理中，利用了 Proxy 這個(gè)類(lèi)來(lái)實(shí)現，在 Proxy 中，有著(zhù) newProxyInstance() 方法，創(chuàng )建一個(gè)動(dòng)態(tài)代理實(shí)例

interface UserMapper {
  @Select("select * from mybatis.user where id =#{id}")
  List<User> selectUserList();
}

public static void main(String[] args) {
  UserMapper userMapper = (UserMapper) Proxy.newProxyInstance(
    JDKMybatis.class.getClassLoader(),
    new Class<?>[]{UserMapper.class},
    new InvocationHandler() {
      /**
       * 在 invoke() 方法中就可以進(jìn)行查找 method，args
       * @param proxy  動(dòng)態(tài)代理
       * @param method 方法
       * @param args   參數
       */
      @Override
      public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        // 把注解類(lèi)獲取，可以查出注解的值等多種其他值
        Select annotation = method.getAnnotation(Select.class);
        if (annotation != null) {
          String[] value = annotation.value();
          System.out.println(Arrays.toString(value));
        }
        return null;
      }
    });

  userMapper.selectUserList(1);
}

newProxyInstance() 的創(chuàng )建需要三個(gè)參數，查看源碼，可以知道需要 ClassLoader 類(lèi)加載器，interfaces 接口（Mapper 接口），InvocationHandler 處理器，來(lái)進(jìn)行處理

public static Object newProxyInstance(ClassLoader loader,
                                      Class<?>[] interfaces,
                                      InvocationHandler h)

把 sql 語(yǔ)句中的參數取出來(lái)放進(jìn) args，這時(shí)需要一個(gè) Map 來(lái)進(jìn)行傳值
問(wèn)題
當在通過(guò)反射獲取方法的參數名，method.getParameters() 獲取出來(lái)的參數都是 arg0，arg1...無(wú)意義參數

在Java8之前，代碼編譯為class文件后，方法參數的類(lèi)型是固定的，但參數名稱(chēng)卻丟失了，在編譯的時(shí)候，需要有編譯的選項，javac -parameters 默認是關(guān)閉的，需要在 idea 中設置開(kāi)啟，開(kāi)啟完成后，重新編譯源文件

這種方式只能臨時(shí)解決當前環(huán)境設置，在其他人運行代碼時(shí)還是要重新設置
另一種解決方式，在pom文件中添加編譯參數：

<plugins>
    <plugin>
        <groupId>org.apache.maven.plugins</groupId>
        <artifactId>maven-compiler-plugin</artifactId>
        <configuration>
            <compilerArgument>-parameters</compilerArgument>
            <source>1.8</source>
            <target>1.8</target>
        </configuration>
    </plugin>
</plugins>

編譯完成后，重新執行，再次通過(guò)method.getParameters()獲取參數：

解析原來(lái)的 sql ，就要把 #{} 給替換掉，這時(shí)候可以使用 StringBuffer 類(lèi)來(lái)實(shí)現替換

private static String parseSql(String sql, Map<String, Object> argsNameMap) {
  // 定義為常量數組
  char[] str = {'#', '{'};
  StringBuilder stringBuilder = new StringBuilder();
  for (int i = 0; i < sql.length(); i++) {
    char aloneParseSql = sql.charAt(i);

    if (str[0] == aloneParseSql) {
      int nextIndex = i + 1;
      char nextChar = sql.charAt(nextIndex);

      // # 后應該是 { ，不匹配直接拋出異常
      if (str[1] != nextChar) {
        throw new RuntimeException(String.format(
          "此處應該是：#{\n sql：%s\n index：%d",
          stringBuilder.toString(), nextIndex));
      }
      /*
        1 已經(jīng)解析完的下標
        2 解析完的 #{} 內的參數名
        3 把對應的 argsName 的值 argsValue 取出來(lái)
        4 追加到原來(lái)的 stringBuilder 中的 sql 語(yǔ)句后面
       */
      StringBuilder partStringBuilder = new StringBuilder();
      i = partParseSql(partStringBuilder, sql, nextIndex);
      String argsName = partStringBuilder.toString();
      Object argsValue = argsNameMap.get(argsName);
      stringBuilder.append(argsValue.toString());
    }

    // 如果沒(méi)有條件，直接追加
    stringBuilder.append(aloneParseSql);
  }
  return stringBuilder.toString();
}

在其中需要把需要替換的值，再用 StringBuffer 類(lèi)來(lái)實(shí)現

private static int partParseSql(StringBuilder partStringBuilder, String sql, int nextIndex) {
  // 由于 nextIndex 當前指針指向的是 { 所以要加一位，把后面內容解析
  nextIndex++;
  char[] str = {'}'};
  for (; nextIndex < sql.length(); nextIndex++) {
    char indexSql = sql.charAt(nextIndex);
    if (str[0] != indexSql) {
      partStringBuilder.append(indexSql);
    }
    if (str[0] == indexSql) {
      return nextIndex;
    }
  }
  throw new RuntimeException(String.format(
    "缺少：}\n index：%d",
    nextIndex));
}

再重新在 invoke 方法中進(jìn)行調用，完成 sql 語(yǔ)句的動(dòng)態(tài)拼裝

@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
  // 把注解類(lèi)獲取，可以查出注解的值等多種其他值
  Select annotation = method.getAnnotation(Select.class);
  Map<String, Object> argsNameMap = MapBuildArgsName(method, args);
  if (annotation != null) {
    String[] value = annotation.value();
    String sql = value[0];
    sql = parseSql(sql, argsNameMap);
    System.out.println(sql);
  }
  return null;
}

到此這篇關(guān)于Mybatis（一）Porxy動(dòng)態(tài)代理和sql解析替換的文章就介紹到這了,更多相關(guān)Mybatis（一）Porxy動(dòng)態(tài)代理和sql解析替換內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！