增加事务的Mock

kaaass · kaaass · Jan 13, 2021 · Jan 1, 2021 · Jan 1, 2021 · Jan 1, 2021
commit 845c7f9e4e0cbfed8d9fa8c928b2bc3d76c0edff
diff --git a/src/main/java/net/kaaass/rumbase/transaction/LockTable.java b/src/main/java/net/kaaass/rumbase/transaction/LockTable.java
@@ -0,0 +1,37 @@
+package net.kaaass.rumbase.transaction;
+
+/**
+ * 锁表
+ *
+ * <p>
+ * 记录锁的信息
+ * </p>
+ *
+ * @author criki
+ */
+public interface LockTable {
+    /**
+     * 添加共享锁
+     *
+     * @param xid  事务id
+     * @param uuid 记录id
+     */
+    void addSharedLock(int xid, long uuid);
+
+    /**
+     * 添加排他锁
+     *
+     * @param xid  事务id
+     * @param uuid 记录id
+     */
+    void addExclusiveLock(int xid, long uuid);
+
+    /**
+     * 释放事务的锁
+     *
+     * @param xid 事务id
+     */
+    void release(int xid);
+
+
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/LockTableManager.java b/src/main/java/net/kaaass/rumbase/transaction/LockTableManager.java
@@ -0,0 +1,38 @@
+package net.kaaass.rumbase.transaction;
+
+import net.kaaass.rumbase.transaction.mock.MockLockTable;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 锁表管理器
+ * <p>
+ * 管理锁表
+ * </p>
+ *
+ * @author criki
+ */
+public class LockTableManager {
+    /**
+     * 表名与锁表的映射
+     */
+    private final Map<String, LockTable> lockTables;
+
+    /**
+     * 锁表管理器
+     */
+    public LockTableManager() {
+        this.lockTables = new HashMap<>();
+    }
+
+    public LockTable getTable(String tableName) {
+        if (lockTables.containsKey(tableName)) {
+            return lockTables.get(tableName);
+        } else {
+            LockTable table = new MockLockTable();
+            lockTables.put(tableName, table);
+            return table;
+        }
+    }
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/TransactionContext.java b/src/main/java/net/kaaass/rumbase/transaction/TransactionContext.java
@@ -1,20 +1,75 @@
 package net.kaaass.rumbase.transaction;
 
+import net.kaaass.rumbase.transaction.mock.MockTransactionContext;
+
 /**
  * 事务上下文
  *
- * @author
+ * <p>
+ * 存储事务状态，对事务进行操作
+ * </p>
+ *
+ * @author criki
  */
-public class TransactionContext {
+public interface TransactionContext {
 
-    private TransactionContext() {
-    }
 
     /**
      * 返回空事务上下文。空事务或超级事务XID为0，不受事务限制，也不具备ACID性质。
+     *
      * @return 空事务上下文
      */
-    public static TransactionContext empty() {
-        return new TransactionContext();
+    static TransactionContext empty() {
+        return new MockTransactionContext();
     }
+
+    /**
+     * 获取事务隔离度
+     *
+     * @return 事务隔离度
+     */
+    TransactionIsolation getIsolation();
+
+    /**
+     * 获取事务状态
+     *
+     * @return 事务状态
+     */
+    TransactionStatus getStatus();
+
+    /**
+     * 获取事务id
+     *
+     * @return 事务id
+     */
+    int getXid();
+
+    /**
+     * 事务开始
+     */
+    void start();
+
+    /**
+     * 事务提交
+     */
+    void commit();
+
+    /**
+     * 事务撤销
+     */
+    void rollback();
+
+    /**
+     * 对记录加共享锁
+     *
+     * @param uuid 记录id
+     */
+    void sharedLock(long uuid);
+
+    /**
+     * 对记录加排他锁
+     *
+     * @param uuid 记录id
+     */
+    void exclusiveLock(long uuid);
 }
diff --git a/src/main/java/net/kaaass/rumbase/transaction/TransactionIsolation.java b/src/main/java/net/kaaass/rumbase/transaction/TransactionIsolation.java
@@ -0,0 +1,30 @@
+package net.kaaass.rumbase.transaction;
+
+import lombok.Getter;
+
+/**
+ * 事务隔离度
+ *
+ * @author criki
+ */
+public enum TransactionIsolation {
+    READ_UNCOMMITTED(0),   //读未提交
+    READ_COMMITTED(1),     //读已提交
+    REPEATABLE_READ(2),    //重复读
+    SERIALIZABLE(3);        //串行化
+
+    /**
+     * 事务隔离度Id
+     */
+    @Getter
+    private final int isolationId;
+
+    /**
+     * 事务隔离度
+     *
+     * @param isolationId 事务隔离度Id
+     */
+    TransactionIsolation(int isolationId) {
+        this.isolationId = isolationId;
+    }
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/TransactionManager.java b/src/main/java/net/kaaass/rumbase/transaction/TransactionManager.java
@@ -0,0 +1,30 @@
+package net.kaaass.rumbase.transaction;
+
+/**
+ * 事务管理器
+ *
+ * <p>
+ * 管理事务的状态
+ * </p>
+ *
+ * @author criki
+ */
+public interface TransactionManager {
+
+    /**
+     * 创建新事务
+     *
+     * @param isolation 事务隔离度
+     * @return 创建的事物对象
+     */
+    TransactionContext createTransactionContext(TransactionIsolation isolation);
+
+    /**
+     * 改变事务日志中的事务状态
+     *
+     * @param xid    事务id
+     * @param status 新事务状态
+     */
+    void changeTransactionStatus(int xid, TransactionStatus status);
+
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/TransactionStatus.java b/src/main/java/net/kaaass/rumbase/transaction/TransactionStatus.java
@@ -0,0 +1,30 @@
+package net.kaaass.rumbase.transaction;
+
+import lombok.Getter;
+
+/**
+ * 事务状态
+ *
+ * @author criki
+ */
+public enum TransactionStatus {
+    PREPARING(0),   //未开始
+    ACTIVE(1),      //正在进行
+    COMMITTED(2),   //已提交
+    ABORTED(3);     //被撤销
+
+    /**
+     * 事务状态Id
+     */
+    @Getter
+    private final byte statusId;
+
+    /**
+     * 事务状态
+     *
+     * @param statusId 事务状态Id
+     */
+    TransactionStatus(int statusId) {
+        this.statusId = (byte) statusId;
+    }
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/exception/DeadlockException.java b/src/main/java/net/kaaass/rumbase/transaction/exception/DeadlockException.java
@@ -0,0 +1,29 @@
+package net.kaaass.rumbase.transaction.exception;
+
+
+import net.kaaass.rumbase.exception.RumbaseException;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * E6001 死锁异常
+ * <p>
+ * E6001-1 发生死锁
+ * </p>
+ */
+public class DeadlockException extends RumbaseException {
+
+    public static final Map<Integer, String> REASONS = new HashMap<>() {{
+        put(1, "发生死锁");
+    }};
+
+    /**
+     * 死锁异常
+     *
+     * @param subId 子错误号
+     */
+    public DeadlockException(int subId) {
+        super(6001, subId, REASONS.get(subId));
+    }
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/mock/MockLockTable.java b/src/main/java/net/kaaass/rumbase/transaction/mock/MockLockTable.java
@@ -0,0 +1,30 @@
+package net.kaaass.rumbase.transaction.mock;
+
+import net.kaaass.rumbase.transaction.LockTable;
+
+/**
+ * Mock锁表
+ *
+ * @author criki
+ */
+
+@Deprecated
+public class MockLockTable implements LockTable {
+
+
+
+    @Override
+    public void addSharedLock(int xid, long uuid) {
+
+    }
+
+    @Override
+    public void addExclusiveLock(int xid, long uuid) {
+
+    }
+
+    @Override
+    public void release(int xid) {
+
+    }
+}
diff --git a/src/main/java/net/kaaass/rumbase/transaction/mock/MockTransactionContext.java b/src/main/java/net/kaaass/rumbase/transaction/mock/MockTransactionContext.java
@@ -0,0 +1,85 @@
+package net.kaaass.rumbase.transaction.mock;
+
+import lombok.Getter;
+import net.kaaass.rumbase.transaction.*;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Mock事务上下文
+ *
+ * @author criki
+ */
+@Deprecated
+public class MockTransactionContext implements TransactionContext {
+
+    @Getter
+    private static final List<Integer> snapshot;
+
+    static {
+        snapshot = new ArrayList<>();
+    }
+
+    @Getter
+    private final int xid;        //事务Id
+    @Getter
+    private final TransactionIsolation isolation; //事务隔离度
+    private final TransactionManager manager; //存储创建它的管理器
+    @Getter
+    private TransactionStatus status; //事务状态
+
+    public MockTransactionContext() {
+        this.xid = 0;
+        this.status = TransactionStatus.COMMITTED;
+        this.isolation = TransactionIsolation.READ_UNCOMMITTED;
+        this.manager = null;
+    }
+
+    /**
+     * 事务上下文
+     *
+     * @param isolation 事务隔离度
+     * @param manager   创建事务的管理器
+     */
+    public MockTransactionContext(int xid, TransactionIsolation isolation, TransactionManager manager) {
+        this.xid = xid;
+        this.status = TransactionStatus.PREPARING;
+        this.isolation = isolation;
+        this.manager = manager;
+    }
+
+    @Override
+    public void start() {
+        this.status = TransactionStatus.ACTIVE;
+        if (manager != null) {
+            manager.changeTransactionStatus(xid, TransactionStatus.ACTIVE);
+        }
+    }
+
+    @Override
+    public void commit() {
+        this.status = TransactionStatus.COMMITTED;
+        if (manager != null) {
+            manager.changeTransactionStatus(xid, TransactionStatus.COMMITTED);
+        }
+    }
+
+    @Override
+    public void rollback() {
+        this.status = TransactionStatus.ABORTED;
+        if (manager != null) {
+            manager.changeTransactionStatus(xid, TransactionStatus.ABORTED);
+        }
+    }
+
+    @Override
+    public void sharedLock(long uuid) {
+
+    }
+
+    @Override
+    public void exclusiveLock(long uuid) {
+
+    }
+}