Read a textbook lol

src/main/java/io/zipcoder/tc_spring_poll_application/controller/ComputeResultController.java

@@ -1,5 +1,6 @@
 package io.zipcoder.tc_spring_poll_application.controller;
 
+import dtos.OptionCount;
 import dtos.VoteResult;
 import io.zipcoder.tc_spring_poll_application.domain.Vote;
 import io.zipcoder.tc_spring_poll_application.repositories.VoteRepository;
@@ -11,6 +12,8 @@ import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
 import javax.inject.Inject;
+import java.util.HashMap;
+import java.util.Map;
 
 @RestController
 public class ComputeResultController {
@@ -19,12 +22,28 @@ public class ComputeResultController {
     private VoteRepository voteRepository;
 
     @RequestMapping(value = "/computeresult", method = RequestMethod.GET)
-    public ResponseEntity<?> computeResult(@RequestParam Long pollId) {
+    public ResponseEntity<VoteResult> computeResult(@RequestParam Long pollId) {
         VoteResult voteResult = new VoteResult();
         Iterable<Vote> allVotes = voteRepository.findVotesByPoll(pollId);
-
-        //TODO: Implement algorithm to count votes
+        countVotes(voteResult, allVotes);
         return new ResponseEntity<VoteResult>(voteResult, HttpStatus.OK);
     }
 
+    private void countVotes(VoteResult voteResult, Iterable<Vote> allVotes) {
+        int totalVotes = 0;
+        Map<Long, OptionCount> map = new HashMap<>();
+        for(Vote v : allVotes) {
+            totalVotes ++;
+            OptionCount optionCount = map.get(v.getOption().getId());
+            if(optionCount == null) {
+                optionCount = new OptionCount();
+                optionCount.setOptionId(v.getOption().getId());
+                map.put(v.getOption().getId(), optionCount);
+            }
+            optionCount.setCount(optionCount.getCount()+1);
+        }
+        voteResult.setTotalVotes(totalVotes);
+        voteResult.setResults((map.values()));
+    }
+
 }