fix: mySession 返回 session 信息

1 files changed, 4 insertions, 3 deletions

diff --git a/src/main/java/net/geedge/asw/module/environment/service/impl/EnvironmentServiceImpl.java b/src/main/java/net/geedge/asw/module/environment/service/impl/EnvironmentServiceImpl.java
index a641193..ed6f45a 100644
--- a/src/main/java/net/geedge/asw/module/environment/service/impl/EnvironmentServiceImpl.java
+++ b/src/main/java/net/geedge/asw/module/environment/service/impl/EnvironmentServiceImpl.java
@@ -131,13 +131,14 @@ public class EnvironmentServiceImpl extends ServiceImpl<EnvironmentDao, Environm
         params.put("currentUserId", currentUserId);
         Page page = new Query(EnvironmentEntity.class).getPage(params);
 
-        List<EnvironmentSessionEntity> sessionEntityList = environmentSessionService.list(new LambdaQueryWrapper<EnvironmentSessionEntity>().eq(EnvironmentSessionEntity::getStatus, 1));
-        List<String> envIdList = sessionEntityList.stream().map(x -> x.getEnvId()).toList();
-
+        List<EnvironmentSessionEntity> sessionEntityList = environmentSessionService.queryListByUsed();
         List<EnvironmentEntity> packageList = this.getBaseMapper().mySession(page, params);
+        List<String> envIdList = sessionEntityList.stream().map(x -> x.getEnvId()).toList();
+        Map<String, EnvironmentSessionEntity> sessionByEnvId = sessionEntityList.stream().collect(Collectors.toMap(EnvironmentSessionEntity::getEnvId, Function.identity()));
         for (EnvironmentEntity entity : packageList) {
             entity.setParam(entity.getParamJSONObject());
             entity.setStatus(envIdList.contains(entity.getId()) ? 2 : entity.getStatus());
+            entity.setSession(sessionByEnvId.get(entity.getId()));
         }
         page.setRecords(packageList);
         return page;