diff --git a/src/main/java/icu/samnyan/aqua/sega/ongeki/handler/impl/GetUserStoryHandler.java b/src/main/java/icu/samnyan/aqua/sega/ongeki/handler/impl/GetUserStoryHandler.java index 8c2a89b7..c2db9994 100644 --- a/src/main/java/icu/samnyan/aqua/sega/ongeki/handler/impl/GetUserStoryHandler.java +++ b/src/main/java/icu/samnyan/aqua/sega/ongeki/handler/impl/GetUserStoryHandler.java @@ -1,7 +1,9 @@ package icu.samnyan.aqua.sega.ongeki.handler.impl; import com.fasterxml.jackson.core.JsonProcessingException; +import icu.samnyan.aqua.sega.ongeki.dao.userdata.UserStoryRepository; import icu.samnyan.aqua.sega.ongeki.handler.BaseHandler; +import icu.samnyan.aqua.sega.ongeki.model.userdata.UserStory; import icu.samnyan.aqua.sega.util.jackson.BasicMapper; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -22,9 +24,12 @@ public class GetUserStoryHandler implements BaseHandler { private final BasicMapper mapper; + private final UserStoryRepository userStoryRepository; + @Autowired - public GetUserStoryHandler(BasicMapper mapper) { + public GetUserStoryHandler(BasicMapper mapper, UserStoryRepository userStoryRepository) { this.mapper = mapper; + this.userStoryRepository = userStoryRepository; } @@ -32,10 +37,12 @@ public class GetUserStoryHandler implements BaseHandler { public String handle(Map request) throws JsonProcessingException { Integer userId = (Integer) request.get("userId"); + List userStoryList = userStoryRepository.findByUser_Card_ExtId(userId); + Map resultMap = new LinkedHashMap<>(); resultMap.put("userId", userId); - resultMap.put("length", 0); - resultMap.put("userStoryList", new List[]{}); + resultMap.put("length", userStoryList.size()); + resultMap.put("userStoryList", userStoryList); String json = mapper.write(resultMap);