0005764: Added method to DataService that returns table reload reques…

…ts mapped by load id
JumpMind · Mar 29, 2023 · a837250 · a837250
1 parent 61df8b4
commit a837250
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/symmetric-core/src/main/java/org/jumpmind/symmetric/service/IDataService.java b/symmetric-core/src/main/java/org/jumpmind/symmetric/service/IDataService.java
@@ -65,6 +65,8 @@ public interface IDataService {
     public List<TableReloadRequest> getTableReloadRequestToProcess(final String sourceNodeId);
 
     public List<TableReloadRequest> getTableReloadRequestToProcessByTarget(final String targetNodeId);
+
+    public Map<Long, List<TableReloadRequest>> getTableReloadRequestByLoadIdMap();
 
     public List<TableReloadStatus> getTableReloadStatus();
 

diff --git a/symmetric-core/src/main/java/org/jumpmind/symmetric/service/impl/DataService.java b/symmetric-core/src/main/java/org/jumpmind/symmetric/service/impl/DataService.java
@@ -442,6 +442,20 @@ public List<TableReloadRequest> collapseTableReloadRequestsByLoadId(List<TableRe
         return collapsedRequests;
     }
 
+    @Override
+    public Map<Long, List<TableReloadRequest>> getTableReloadRequestByLoadIdMap() {
+        Map<Long, List<TableReloadRequest>> requestMap = new HashMap<Long, List<TableReloadRequest>>();
+        for (TableReloadRequest request : getTableReloadRequests()) {
+            List<TableReloadRequest> requestList = requestMap.get(request.getLoadId());
+            if (requestList == null) {
+                requestList = new ArrayList<TableReloadRequest>();
+                requestMap.put(request.getLoadId(), requestList);
+            }
+            requestList.add(request);
+        }
+        return requestMap;
+    }
+
     public TableReloadStatus updateTableReloadStatusDataLoaded(ISqlTransaction transaction, long loadId, long batchId, int batchCount, boolean isBulkLoaded) {
         int idType = symmetricDialect.getSqlTypeForIds();
         int count;