fix: entity mapping

Congress-TF · Feb 4, 2024 · 3576df2 · 3576df2
1 parent 01c848a
commit 3576df2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...odule/src/main/java/com/congress/coremodule/mypage/domain/service/MyPageQueryService.java b/...odule/src/main/java/com/congress/coremodule/mypage/domain/service/MyPageQueryService.java
@@ -1,6 +1,7 @@
 package com.congress.coremodule.mypage.domain.service;
 
 import com.congress.coremodule.law.domain.entity.Law;
+import com.congress.coremodule.law.domain.entity.LegislateLaw;
 import com.congress.coremodule.law.domain.repository.LawRepository;
 import com.congress.coremodule.member.domain.entity.Member;
 import com.congress.coremodule.member.domain.repository.MemberRepository;
@@ -46,8 +47,12 @@ public List<Long> getLawIds() {
 
     public List<Long> getLegislatorIds(Long memberId) {
 
-        List<LegislateVote> vote = legislateVoteRepository.findLegislateVotesByMemberId(memberId);
-        return vote.stream().map(LegislateVote::getId).collect(Collectors.toList());
+        return legislateVoteRepository
+                .findLegislateVotesByMemberId(memberId)
+                .stream()
+                .map(LegislateVote::getLegislateLaw)
+                .map(LegislateLaw::getId)
+                .toList();
     }
 
     public List<String> getHashTagNames(Long memberId) {