Sockets Handlers

Author: AlfredDev

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

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

Vote/src/main/java/com/websocket/voteApp/Vote/Repository/VoteRepository.java

@@ -5,6 +5,7 @@
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
@@ -16,4 +17,7 @@ public interface VoteRepository extends JpaRepository<Vote, Long> {
 
 
     Page<Vote> getVotesByPollId(Long pollId, Pageable pageable);
+
+    @Query("SELECT COUNT(v) FROM Vote v WHERE v.candidate.id = :candidateId AND v.poll.id = :pollId")
+    long countVotesByCandidateIdAndPollId(@Param("candidateId") Long candidateId, @Param("pollId") Long pollId);
 }

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

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