optimize: once delete GlobalSession locks for db mode when commit success

server/src/main/java/io/seata/server/lock/DefaultLockManager.java

@@ -15,6 +15,7 @@
  */
 package io.seata.server.lock;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import io.seata.common.util.CollectionUtils;
@@ -23,6 +24,7 @@
 import io.seata.core.lock.Locker;
 import io.seata.core.lock.RowLock;
 import io.seata.server.session.BranchSession;
+import io.seata.server.session.GlobalSession;
 
 /**
  * The type Default lock manager.
@@ -53,18 +55,37 @@ public boolean acquireLock(BranchSession branchSession) throws TransactionExcept
     @Override
     public boolean releaseLock(BranchSession branchSession) throws TransactionException {
         List<RowLock> locks = collectRowLocks(branchSession);
-        if (CollectionUtils.isEmpty(locks)) {
-            //no lock
-            return true;
-        }
         try {
-            return getLocker(branchSession).releaseLock(locks);
+            return this.doReleaseLock(locks, branchSession);
         } catch (Exception t) {
             LOGGER.error("unLock error, branchSession:" + branchSession, t);
             return false;
         }
     }
 
+    @Override
+    public boolean releaseDBGlobalSessionLock(GlobalSession globalSession) throws TransactionException {
+        List<RowLock> locks = new ArrayList<>();
+        ArrayList<BranchSession> branchSessions = globalSession.getBranchSessions();
+        for (BranchSession branchSession : branchSessions) {
+            locks.addAll(collectRowLocks(branchSession));
+        }
+        try {
+            return this.doReleaseLock(locks, null);
+        } catch (Exception t) {
+            LOGGER.error("unLock globalSession error, xid:" + globalSession.getXid(), t);
+            return false;
+        }
+    }
+
+    private boolean doReleaseLock(List<RowLock> locks, BranchSession branchSession) {
+        if (CollectionUtils.isEmpty(locks)) {
+            //no lock
+            return true;
+        }
+        return getLocker(branchSession).releaseLock(locks);
+    }
+
     @Override
     public boolean isLockable(String xid, String resourceId, String lockKey) throws TransactionException {
         List<RowLock> locks = collectRowLocks(lockKey, resourceId, xid);

server/src/main/java/io/seata/server/lock/LockManager.java

@@ -17,6 +17,7 @@
 
 import io.seata.core.exception.TransactionException;
 import io.seata.server.session.BranchSession;
+import io.seata.server.session.GlobalSession;
 
 /**
  * The interface Lock manager.
@@ -43,6 +44,16 @@ public interface LockManager {
      */
     boolean releaseLock(BranchSession branchSession) throws TransactionException;
 
+    /**
+     * Un lock boolean.
+     * only used by db mode
+     *
+     * @param globalSession the global session
+     * @return the boolean
+     * @throws TransactionException the transaction exception
+     */
+    boolean releaseDBGlobalSessionLock(GlobalSession globalSession) throws TransactionException;
+
     /**
      * Is lockable boolean.
      *

server/src/main/java/io/seata/server/session/GlobalSession.java

@@ -25,12 +25,17 @@
 import java.util.concurrent.locks.ReentrantLock;
 
 import io.seata.common.XID;
+import io.seata.config.Configuration;
+import io.seata.config.ConfigurationFactory;
+import io.seata.core.constants.ConfigurationKeys;
 import io.seata.core.exception.TransactionException;
 import io.seata.core.exception.TransactionExceptionCode;
 import io.seata.core.model.BranchStatus;
 import io.seata.core.model.BranchType;
 import io.seata.core.model.GlobalStatus;
+import io.seata.core.store.StoreMode;
 import io.seata.server.UUIDGenerator;
+import io.seata.server.lock.LockerFactory;
 import io.seata.server.store.SessionStorable;
 import io.seata.server.store.StoreConfig;
 import org.slf4j.Logger;
@@ -74,6 +79,11 @@ public class GlobalSession implements SessionLifecycle, SessionStorable {
 
     private GlobalSessionLock globalSessionLock = new GlobalSessionLock();
 
+    /**
+     * The constant CONFIG.
+     */
+    protected static final Configuration CONFIG = ConfigurationFactory.getInstance();
+
     /**
      * Add boolean.
      *
@@ -173,8 +183,13 @@ public void end() throws TransactionException {
     }
 
     public void clean() throws TransactionException {
-        for (BranchSession branchSession : branchSessions) {
-            branchSession.unlock();
+        String storeMode = CONFIG.getConfig(ConfigurationKeys.STORE_MODE);
+        if (StoreMode.DB.name().equalsIgnoreCase(storeMode)) {
+            LockerFactory.getLockManager().releaseDBGlobalSessionLock(this);
+        } else {
+            for (BranchSession branchSession : branchSessions) {
+                branchSession.unlock();
+            }
         }
 
     }
@@ -631,4 +646,8 @@ public interface LockCallable<V> {
 
         V call() throws TransactionException;
     }
+
+    public ArrayList<BranchSession> getBranchSessions() {
+        return branchSessions;
+    }
 }