From 6def3f791aee37bdbd98d6748b96d7c9babafbdb Mon Sep 17 00:00:00 2001
From: yinghuochongfly <39123775+yinghuochongfly@users.noreply.github.com>
Date: Fri, 15 Dec 2023 10:15:46 +0800
Subject: [PATCH] <perf>(xatransaction): Improve the xaTransaction list
 interface  (#596)

* Improve the xaTransaction list interface

* feat:add annotation

* feat:Formatted code

* feat:Version changed to 1.0

---------

Co-authored-by: ylxiao5 <ylxiao5@iflytek.com>
---
 .../rpc/handler/XATransactionHandler.java     | 10 ++++
 .../routine/xa/XATransactionManager.java      | 53 +++++++++++++++++--
 2 files changed, 60 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/webank/wecross/network/rpc/handler/XATransactionHandler.java b/src/main/java/com/webank/wecross/network/rpc/handler/XATransactionHandler.java
index 4e5ddb560..f5c0f2078 100644
--- a/src/main/java/com/webank/wecross/network/rpc/handler/XATransactionHandler.java
+++ b/src/main/java/com/webank/wecross/network/rpc/handler/XATransactionHandler.java
@@ -58,6 +58,7 @@ public void setPaths(Set<String> paths) {
 
     public static class ListXATransactionsRequest {
         private int size;
+        private String path;
         private Map<String, Long> offsets = Collections.synchronizedMap(new HashMap<>());
 
         public int getSize() {
@@ -75,6 +76,14 @@ public Map<String, Long> getOffsets() {
         public void setOffsets(Map<String, Long> offsets) {
             this.offsets = offsets;
         }
+
+        public String getPath() {
+            return path;
+        }
+
+        public void setPath(String path) {
+            this.path = path;
+        }
     }
 
     @Override
@@ -229,6 +238,7 @@ public void handle(
                                 host.getAccountManager().getAdminUA(),
                                 xaRequest.getData().getOffsets(),
                                 xaRequest.getData().getSize(),
+                                xaRequest.getData().getPath(),
                                 (exception, xaTransactionListResponse) -> {
                                     if (logger.isDebugEnabled()) {
                                         logger.debug(
diff --git a/src/main/java/com/webank/wecross/routine/xa/XATransactionManager.java b/src/main/java/com/webank/wecross/routine/xa/XATransactionManager.java
index 875fffd11..988010774 100644
--- a/src/main/java/com/webank/wecross/routine/xa/XATransactionManager.java
+++ b/src/main/java/com/webank/wecross/routine/xa/XATransactionManager.java
@@ -576,6 +576,44 @@ public void asyncGetXATransaction(
         }
     }
 
+    private ListXAReduceCallback getListXACallback(
+            int count, Map<String, Long> offsets, int size, ListXATransactionsCallback callback) {
+        List<XAResponse.ChainErrorMessage> errors =
+                Collections.synchronizedList(new LinkedList<>());
+        Map<String, Long> nextOffsets = new ConcurrentHashMap<>(offsets);
+
+        return (chain, listXAResponse) -> {
+
+            // first time to list, reset offsets
+            if (nextOffsets.get(chain) == -1) {
+                nextOffsets.put(chain, listXAResponse.getTotal() - 1);
+            }
+
+            XATransactionListResponse response = new XATransactionListResponse();
+            if (Objects.nonNull(listXAResponse.getChainErrorMessage())) {
+                errors.add(listXAResponse.getChainErrorMessage());
+                if (!errors.isEmpty()) {
+                    XAResponse xaResponse = new XAResponse();
+                    xaResponse.setStatus(-1);
+                    xaResponse.setChainErrorMessages(errors);
+                    response.setXaResponse(xaResponse);
+                }
+            } else {
+                response.setXaList(listXAResponse.getXaTransactions());
+                // update offsets
+                Long nextOffset =
+                        nextOffsets.get(chain) - listXAResponse.getXaTransactions().size();
+                nextOffsets.put(chain, nextOffset);
+                response.setNextOffsets(nextOffsets);
+                if (nextOffsets.get(chain) == -1) {
+                    response.setFinished(true);
+                }
+                response.recoverUsername(accountManager);
+            }
+            callback.onResponse(null, response);
+        };
+    }
+
     public interface ListXATransactionsCallback {
         void onResponse(WeCrossException e, XATransactionListResponse xaTransactionListResponse);
     }
@@ -728,6 +766,7 @@ public void asyncListXATransactions(
             UniversalAccount ua,
             Map<String, Long> offsets,
             int size,
+            String chainPath,
             ListXATransactionsCallback callback) {
 
         try {
@@ -742,7 +781,17 @@ public void asyncListXATransactions(
 
             XATransactionListResponse response = new XATransactionListResponse();
 
-            List<Path> chainPaths = setToList(zoneManager.getAllChainsInfo(false).keySet());
+            ListXAReduceCallback reduceCallback = null;
+            List<Path> chainPaths = new ArrayList<>();
+            if (Objects.isNull(chainPath) || chainPath.isEmpty()) {
+                chainPaths = setToList(zoneManager.getAllChainsInfo(false).keySet());
+                // has sort operation callback
+                reduceCallback = getListXAReduceCallback(offsets.size(), offsets, size, callback);
+            } else {
+                chainPaths.add(Path.decode(chainPath));
+                // Remove sort operation callback
+                reduceCallback = getListXACallback(offsets.size(), offsets, size, callback);
+            }
 
             int chainNum = chainPaths.size();
             if (chainNum == 0) {
@@ -761,8 +810,6 @@ public void asyncListXATransactions(
                 }
             }
 
-            ListXAReduceCallback reduceCallback =
-                    getListXAReduceCallback(offsets.size(), offsets, size, callback);
             for (String chain : offsets.keySet()) {
                 if (!requireIgnore || offsets.get(chain) != -1L) {
                     asyncListXATransactions(