diff --git a/src/main/java/icu/samnyan/aqua/sega/maimai2/Maimai2ServletController.kt b/src/main/java/icu/samnyan/aqua/sega/maimai2/Maimai2ServletController.kt index ab3f17d8..fa69d3e9 100644 --- a/src/main/java/icu/samnyan/aqua/sega/maimai2/Maimai2ServletController.kt +++ b/src/main/java/icu/samnyan/aqua/sega/maimai2/Maimai2ServletController.kt @@ -97,9 +97,9 @@ class Maimai2ServletController( "userCourseList" to repos.userCourse.findByUser_Card_ExtId(userId) ) } - val getUserFavorite = UserReqHandler { _, userId -> mapOf( + val getUserFavorite = UserReqHandler { req, userId -> mapOf( "userId" to userId, - "userFavoriteData" to repos.userFavorite.findByUser_Card_ExtId(userId) + "userFavorite" to repos.userFavorite.findByUser_Card_ExtIdAndItemKind(userId, req["itemKind"] as Int) ) } val getUserActivity = UserReqHandler { _, userId -> diff --git a/src/main/java/icu/samnyan/aqua/sega/maimai2/model/Repos.kt b/src/main/java/icu/samnyan/aqua/sega/maimai2/model/Repos.kt index 77c748b3..aebf8a65 100644 --- a/src/main/java/icu/samnyan/aqua/sega/maimai2/model/Repos.kt +++ b/src/main/java/icu/samnyan/aqua/sega/maimai2/model/Repos.kt @@ -61,6 +61,8 @@ interface Mai2UserFavoriteRepo : Mai2UserLinked { fun findByUserAndItemKind(user: Mai2UserDetail, kind: Int): Optional fun findByUserIdAndItemKind(userId: Long, kind: Int): List + + fun findByUser_Card_ExtIdAndItemKind(userId: Long, kind: Int): Optional } interface Mai2UserFriendSeasonRankingRepo : Mai2UserLinked {