From 879056f353d5aa047c44deb65af3dce528a3b665 Mon Sep 17 00:00:00 2001
From: "MSI\\derek" <abc@gmail.com>
Date: Tue, 4 Jun 2024 17:39:37 +0800
Subject: [PATCH] update

---
 .../com/ffii/tsms/modules/data/web/TeamController.kt     | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/ffii/tsms/modules/data/web/TeamController.kt b/src/main/java/com/ffii/tsms/modules/data/web/TeamController.kt
index 95d9831..66bbcb3 100644
--- a/src/main/java/com/ffii/tsms/modules/data/web/TeamController.kt
+++ b/src/main/java/com/ffii/tsms/modules/data/web/TeamController.kt
@@ -43,14 +43,9 @@ class TeamController(
     @GetMapping("/{id}")
     fun getStaff(@PathVariable id: Long): Map<String, Any> {
         val staffList = staffsService.findAllByTeamId(id).orElseThrow { NotFoundException() }
-        val staffIdList: MutableList<Long> = mutableListOf()
-        for (staff in staffList) {
-            staffIdList.add(staff.id as Long)
-        }
-//        val map: Map<String, Any> =  java.util.Map.of("team" to teamService.find(id).orElseThrow { NotFoundException() })
-//        map["staffIds"] = staffIdList
+        val staffIdList: List<Long?> = staffList.map { it.id }
         return java.util.Map.of(
-            "team", staffsService.find(id).orElseThrow { NotFoundException() },
+            "team", teamService.find(id).orElseThrow { NotFoundException() },
             "staffIds", staffIdList
         )
     }