Sockets Handlers

Author: AlfredDev

Vote/src/main/java/com/websocket/voteApp/Vote/DTO/Response/CandidateResponse.java

@@ -16,4 +16,7 @@ public class CandidateResponse {
     private String partyName;
 
     private Long pollId;
+
+    private  Long countVote;
+
 }

Vote/src/main/java/com/websocket/voteApp/Vote/DTO/VoteResponse.java

@@ -16,6 +16,4 @@ public class VoteResponse {
     String candidateName;
     String candidatePartyName;
     LocalDateTime timestamp;
-    private long voteCount;
-
 }

Vote/src/main/java/com/websocket/voteApp/Vote/Mappers/CandidateMapper.java

@@ -9,5 +9,6 @@
 public interface CandidateMapper {
     Candidate toCandidate(CandidateResponse candidateResponse);
     @Mapping(source = "poll.id", target = "pollId")
+    @Mapping(source = "voteCount", target = "countVote")
     CandidateResponse toCandidateResponse(Candidate candidate);
 }

Vote/src/main/java/com/websocket/voteApp/Vote/Models/Candidate.java

@@ -19,7 +19,7 @@ public class Candidate {
 
     private String name;
     private String partyName;
-
+    private int voteCount;
     @ManyToOne
     @JoinColumn(name = "poll_id")
     private Poll poll;

Vote/src/main/java/com/websocket/voteApp/Vote/Service/Impl/VoteServiceImpl.java

@@ -52,9 +52,11 @@ public VoteResponse saveVote(Long userId, VoteRequest request) {
                 .timestamp(LocalDateTime.now())
                 .poll(poll)
                 .build();
-        VoteResponse votesa = voteMapper.toVoteResponse(voteRepository.save(vote));
-        votesa.setVoteCount(voteRepository.countVotesByCandidateIdAndPollId(request.getIdCandidate(), request.getPollId()));
-        return votesa;
+        voteRepository.save(vote);
+        long voteCount = voteRepository.countVotesByCandidateIdAndPollId(request.getIdCandidate(), request.getPollId());
+        candidate.setVoteCount((int) voteCount);
+        candidateRepository.save(candidate);
+        return voteMapper.toVoteResponse(vote);
     }
 
     @Override