finished up to part 5

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

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.controller;
 
 import io.zipcoder.tc_spring_poll_application.domain.Option;
+import io.zipcoder.tc_spring_poll_application.domain.Poll;
 import io.zipcoder.tc_spring_poll_application.domain.Vote;
 import io.zipcoder.tc_spring_poll_application.dtos.OptionCount;
 import io.zipcoder.tc_spring_poll_application.dtos.VoteResult;
@@ -13,9 +14,7 @@ import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.util.ArrayList;
-import java.util.Collection;
-import java.util.Iterator;
+import java.util.*;
 
 @RestController
 public class ComputeResultController {
@@ -31,21 +30,30 @@ public class ComputeResultController {
     public ResponseEntity<?> computeResult(@RequestParam Long pollId) {
         VoteResult voteResult = new VoteResult();
         Iterable<Vote> allVotes = voteRepository.findById(pollId);
-        Iterator it = allVotes.iterator();
+        Map<Long, Integer> mapOptions = new HashMap<>();
         Collection<OptionCount> optionCounts = new ArrayList<>();
+        int totalVotes = 0;
 
-//        while(it.hasNext()) {
-//            Vote vote = (Vote) it.next();
-//            Option option = vote.getOption();
-//            OptionCount optionCount = new OptionCount();
-//            try {
-//                ((ArrayList<OptionCount>) optionCounts).get(optionCount)
-//            optionCount.setOptionId(option.getId());
-//            optionCount.setCount(1);
-//                ((ArrayList<OptionCount>) optionCounts).add(optionCount);
-//            }
-//            voteResult.setResults(optionCounts);
-//        }
-        return new ResponseEntity<VoteResult>(voteResult, HttpStatus.OK);
+        for(Vote vote: allVotes) {
+            Long id = vote.getOption().getId();
+            totalVotes++;
+            if(!mapOptions.containsKey(id)) {
+                mapOptions.put(id, 1);
+            } else {
+                int count = mapOptions.get(id);
+                count = count + 1;
+                mapOptions.put(id, count);
+            }
+        }
+
+        for(Long i: mapOptions.keySet()) {
+            OptionCount optionCount = new OptionCount(i, mapOptions.get(i));
+            optionCounts.add(optionCount);
+        }
+
+        voteResult.setResults(optionCounts);
+        voteResult.setTotalVotes(totalVotes);
+
+        return new ResponseEntity<>(voteResult, HttpStatus.OK);
     }
 }

src/main/java/io/zipcoder/tc_spring_poll_application/dtos/OptionCount.java

@@ -4,6 +4,11 @@ public class OptionCount {
     private Long optionId;
     private int count;
 
+    public OptionCount(Long optionId, int count) {
+        this.optionId = optionId;
+        this.count = count;
+    }
+
     public Long getOptionId() {
         return optionId;
     }