手寫mybatis框架-增加快取&事務功能
阿新 • • 發佈：2020-09-10
前言
在學習mybatis原始碼之餘，自己完成了一個簡單的ORM框架。已完成基本SQL的執行和物件關係對映。本週在此基礎上，又加入了快取和事務功能。所有程式碼都沒有copy，如果也對此感興趣，請賞個Star。
專案地址：simple-ibatis
初版博文地址：部落格園博文
增加程式碼詳解
快取 com.simple.ibatis.cache
快取介面-Cache
public interface Cache { /**放入快取*/ void putCache(String key,Object val); /**獲取快取*/ Object getCache(String key);/**清空快取*/ void cleanCache(); /**獲取快取健數量*/ int getSize(); /**移除key的快取*/ void removeCache(String key); }
自定義框架快取介面，提供基本的增刪改查功能。
快取基本實現類-SimpleCache
public class SimpleCache implements Cache{ // 內部使用HashMap作為快取實現 private static Map<String,Object> map = new HashMap<>();// 呼叫map.put()方法實現存快取功能 @Override public void putCache(String key, Object val) { map.put(key,val); } // 呼叫map.get()方法實現取快取功能 @Override public Object getCache(String key) { return map.get(key); } // 呼叫map.clear()方法實現清空快取功能 @Override public void cleanCache() { map.clear(); }// 呼叫map.size()方法獲取快取數量 @Override public int getSize() { return map.size(); } // 呼叫map.remove()方法移除快取 @Override public void removeCache(String key) { map.remove(key); } }
simple-ibatis完成對HasaMap的封裝，實現了基本的快取獲取，刪除，清除等功能。
具備LRU淘汰策略-LruCache
/** * @author xiabing * @description: 快取包裝類，具備Lru淘汰策略功能 */ public class LruCache implements Cache{ // 預設快取數 private static Integer cacheSize = 100; // 負載因子 private static Float loadFactory = 0.75F; // 真實快取 private Cache trueCache; // 重寫LinkedHashMap方法實現Lru功能 private Map<String,Object> linkedCache; // 待移除的快取元素 private static Map.Entry removeEntry; public LruCache(Cache trueCache){ this(cacheSize,loadFactory,trueCache); } public LruCache(Integer cacheSize, Float loadFactory, Cache trueCache) { this.cacheSize = cacheSize; this.loadFactory = loadFactory; this.trueCache = trueCache; this.linkedCache = new LinkedHashMap<String, Object>(cacheSize,loadFactory,true){ @Override // 當快取數大於設定的預設快取數時，linkedHashMap會淘汰最近最少使用的元素，獲取此元素，在真實快取中淘汰即可 protected boolean removeEldestEntry(Map.Entry eldest) { if(getSize() > cacheSize){ removeEntry = eldest; return true; } return false; } }; } @Override public void putCache(String key, Object val) { this.trueCache.putCache(key,val); this.linkedCache.put(key,val); if(removeEntry != null){ // 若找到了最近最少元素，則進行移除 removeCache((String)removeEntry.getKey()); removeEntry = null; } } @Override public Object getCache(String key) { // linkedCache獲取的意義是觸發linkedHashMap元素排序 linkedCache.get(key); return trueCache.getCache(key); } @Override public void cleanCache() { trueCache.cleanCache(); linkedCache.clear(); } @Override public int getSize() { return trueCache.getSize(); } @Override public void removeCache(String key) { trueCache.removeCache(key); } }
LruCache是根據LinkedHashMap的特性來實現，若對此有疑問，可參考mybatis關於LruCache功能的實現 -mybatis快取介紹
專案程式碼測試
@Test // 測試快取獲取 public void shouldGetCache() throws SQLException { // 初始化連線池 PoolDataSource poolDataSource = new PoolDataSource("com.mysql.jdbc.Driver","jdbc:mysql://101.132.150.75:3306/our-auth","root","root"); Config config = new Config("com/simple/ibatis/mapper",poolDataSource); // 設定全域性配置，開啟快取 config.setOpenCache(true); // 獲取執行器 Executor simpleExecutor = config.getExecutor(); UserMapper userMapper = simpleExecutor.getMapper(UserMapper.class); User user = new User(); user.setId(1); user.setName("root"); // 第一次呼叫 List<User> userList = userMapper.getUsers(user); // 第二次呼叫，我在原始碼中有列印輸出，若使用了快取，則列印語句 List<User> userList1 = userMapper.getUsers(user); simpleExecutor.close(); }
結果列印如下 this is cache .感興趣的可以自己試下
cache我設定了全域性可配置，預設生成的是LruCache。並在更新，修改，刪除的SQL操作前強制重新整理快取。詳細程式碼邏輯見專案中SimpleExecutor類。
事務功能com.simple.ibatis.transaction
事務介面-Transaction
/** * @Author xiabing * @Desc 增加事務功能 **/ public interface Transaction { /**獲取連結*/ Connection getConnection() throws SQLException; /**提交*/ void commit() throws SQLException; /**回滾*/ void rollback() throws SQLException; /**關閉*/ void close() throws SQLException; }
JDBC事務-SimpleTransaction
package com.simple.ibatis.transaction; import com.simple.ibatis.datasource.PoolDataSource; import java.sql.Connection; import java.sql.SQLException; /** * @Author xiabing * @Desc 事務的簡單實現 **/ public class SimpleTransaction implements Transaction{ private Connection connection; // 資料庫連線 private PoolDataSource dataSource; // 資料來源 private Integer level = Connection.TRANSACTION_REPEATABLE_READ;; // 事務隔離級別 private Boolean autoCommmit = true; // 是否自動提交 public SimpleTransaction(PoolDataSource dataSource){ this(dataSource,null,null); } public SimpleTransaction(PoolDataSource dataSource, Integer level, Boolean autoCommmit) { this.dataSource = dataSource; if(level != null){ this.level = level; } if(autoCommmit != null){ this.autoCommmit = autoCommmit; } } @Override public Connection getConnection() throws SQLException{ this.connection = dataSource.getConnection(); this.connection.setAutoCommit(autoCommmit); this.connection.setTransactionIsolation(level); return this.connection; } @Override public void commit() throws SQLException{ if(this.connection != null){ this.connection.commit(); } } @Override public void rollback() throws SQLException{ if(this.connection != null){ this.connection.rollback(); } } /**關閉連結前，若設定了自動提交為false，則必須進行回滾操作*/ @Override public void close() throws SQLException{ if(!autoCommmit && connection != null){ connection.rollback(); } /**放回連線池*/ if(connection != null){ dataSource.removeConnection(connection); } /**連結設為null*/ this.connection = null; } }
simpleTransaction主要將事務管理功能交給了資料庫本身（即connection），事務隔離級別默然是mysql的事務隔離級別。通過對Connection的管理，進而實現對connection一系列操作的事務控制。
Test public void shouldOpenTransaction() { /**基本配置*/ PoolDataSource poolDataSource = new PoolDataSource("com.mysql.jdbc.Driver","jdbc:mysql://101.132.150.75:3306/our-auth","root","root"); Config config = new Config("com/simple/ibatis/mapper",poolDataSource); /**設定為啟用事務，關閉自動提交*/ config.setOpenTransaction(true); /**獲取執行器*/ Executor simpleExecutor = config.getExecutor(); UserMapper userMapper = simpleExecutor.getMapper(UserMapper.class); User user = new User(); user.setId(1); user.setName("xiabing"); /**更新名字為xiabing，但未提交*/ userMapper.update(user); User user1 = userMapper.getUserById(1); /**獲取ID為1的名字，為root，說明上文的語句還沒有提交*/ System.out.println(user1.getName()); /**事務提交語句*/ //simpleExecutor.commit(); }
若不提交事物，即執行simpleExecutor.commit()語句，更新語句將不會自動提交到資料庫。上述程式碼在github專案中Test類中shouldOpenTransaction()方法上，可自行debug測試。
總結：
該專案屬於我抱著學習的心態去做的專案，將Mybatis原始碼一步步拆解，在實踐中去領悟其強大的地方。此次在已有的基礎上增加了快取和事務的功能。又是一次學習之旅。因為程式碼全手寫，沒有COPY任何一句程式碼，不是很完善，請見諒。如果覺的感興趣，請給我個star支援下。因為自己想一直去維護這個專案，如果你也感興趣，可以私聊我和我一起做下去，一起寫好這個開源專案。最後，真心求Star了 --------
專案地址：simple-ibatis