fixed the dataset permissions field query by the listDataset is null

WeBankBlockchain · Sep 2, 2024 · dd468ec · dd468ec
1 parent 0e5f6d4
commit dd468ec
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...dataset/src/main/java/com/webank/wedpr/components/dataset/service/DatasetServiceImpl.java b/...dataset/src/main/java/com/webank/wedpr/components/dataset/service/DatasetServiceImpl.java
@@ -532,6 +532,16 @@ public ListDatasetResponse listDataset(
                                 startTime,
                                 endTime);
 
+                if (!datasetList.isEmpty()) {
+                    for (Dataset dataset : datasetList) {
+                        String datasetId = dataset.getDatasetId();
+                        DatasetUserPermissions datasetUserPermissions =
+                                DatasetUserPermissionValidator.confirmUserDatasetPermissions(
+                                        datasetId, userInfo, datasetPermissionMapper, false);
+                        dataset.setPermissions(datasetUserPermissions);
+                    }
+                }
+
                 long totalCount = new PageInfo<>(datasetList).getTotal();
                 long pageEndOffset = (long) pageNum * pageSize;
                 boolean isLast = (pageEndOffset >= totalCount);