Refactor : TripMemberMsg에 NumberOfPeople 추가

WePlanPlans · Jan 11, 2024 · 2b7b339 · 2b7b339
1 parent 10fdaa1
commit 2b7b339
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/main/java/org/tenten/tentenstomp/domain/trip/dto/response/TripMemberMsg.java b/src/main/java/org/tenten/tentenstomp/domain/trip/dto/response/TripMemberMsg.java
@@ -5,6 +5,7 @@
 public record TripMemberMsg(
     Long tripId,
     List<TripMemberInfoMsg> connectedMembers,
-    List<TripMemberInfoMsg> tripMembers
+    List<TripMemberInfoMsg> tripMembers,
+    Long numberOfPeople
 ) {
 }
diff --git a/src/main/java/org/tenten/tentenstomp/domain/trip/service/TripService.java b/src/main/java/org/tenten/tentenstomp/domain/trip/service/TripService.java
@@ -45,11 +45,12 @@ public class TripService {
     @Transactional
     public void connectMember(String tripId, MemberConnectMsg memberConnectMsg) {
         HashMap<Long, TripMemberInfoMsg> connectedMemberMap = tripConnectedMemberMap.getOrDefault(tripId, new HashMap<>());
+        Trip trip = tripRepository.getReferenceById(Long.parseLong(tripId));
         Optional<TripMemberInfoMsg> tripMemberInfoByMemberId = memberRepository.findTripMemberInfoByMemberId(memberConnectMsg.memberId());
         tripMemberInfoByMemberId.ifPresent(tripMemberInfoMsg -> connectedMemberMap.put(memberConnectMsg.memberId(), tripMemberInfoMsg));
 
         TripMemberMsg tripMemberMsg = new TripMemberMsg(
-            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId))
+            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId)), trip.getNumberOfPeople()
         );
         tripConnectedMemberMap.put(tripId, connectedMemberMap);
         sendToKafkaAndSave(tripMemberMsg);
@@ -59,10 +60,11 @@ public void connectMember(String tripId, MemberConnectMsg memberConnectMsg) {
     @Transactional
     public void disconnectMember(String tripId, MemberDisconnectMsg memberDisconnectMsg) {
         HashMap<Long, TripMemberInfoMsg> connectedMemberMap = tripConnectedMemberMap.getOrDefault(tripId, new HashMap<>());
+        Trip trip = tripRepository.getReferenceById(Long.parseLong(tripId));
         connectedMemberMap.remove(memberDisconnectMsg.memberId());
 
         TripMemberMsg tripMemberMsg = new TripMemberMsg(
-            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId))
+            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId)), trip.getNumberOfPeople()
         );
         tripConnectedMemberMap.put(tripId, connectedMemberMap);
         sendToKafkaAndSave(tripMemberMsg);
@@ -185,8 +187,9 @@ private TripMemberMsg getTripMemberMsg(String tripId) {
             return objectMapper.convertValue(cached, TripMemberMsg.class);
         }
         HashMap<Long, TripMemberInfoMsg> connectedMemberMap = tripConnectedMemberMap.getOrDefault(tripId, new HashMap<>());
+        Trip trip = tripRepository.getReferenceById(Long.parseLong(tripId));
         TripMemberMsg tripMemberMsg = new TripMemberMsg(
-            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId))
+            Long.parseLong(tripId), connectedMemberMap.values().stream().toList(), memberRepository.findTripMemberInfoByTripId(Long.parseLong(tripId)), trip.getNumberOfPeople()
         );
         redisCache.save(MEMBER, tripId, tripMemberMsg);
         return tripMemberMsg;