Created ResultCalculator

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

@@ -0,0 +1,49 @@
+package io.zipcoder.tc_spring_poll_application.controllers;
+
+import io.zipcoder.tc_spring_poll_application.domain.Option;
+import io.zipcoder.tc_spring_poll_application.domain.Vote;
+import io.zipcoder.tc_spring_poll_application.dtos.VoteResult;
+import io.zipcoder.tc_spring_poll_application.repositories.VoteRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+
+import java.util.Iterator;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.controllers
+ * author: https://github.com/vvmk
+ * date: 4/6/18
+ */
+@RestController
+public class ComputeResultController {
+
+    private VoteRepository voteRepository;
+
+    @Autowired
+    public ComputeResultController(VoteRepository voteRepository) {
+        this.voteRepository = voteRepository;
+    }
+
+    @RequestMapping(value = "/computeresult", method = RequestMethod.GET)
+    public ResponseEntity<?> computeResult(@RequestParam Long pollId) {
+        Iterable<Vote> allVotes = voteRepository.findVotesByPoll(pollId);
+        VoteResult voteResult = computeResults(allVotes);
+
+        return new ResponseEntity<>(voteResult, HttpStatus.OK);
+    }
+
+    protected VoteResult computeResults(Iterable<Vote> votes) {
+        Iterator<Vote> voterator = votes.iterator();
+        VoteResult result = new VoteResult();
+        while(voterator.hasNext()) {
+            Vote v = voterator.next();
+        }
+        return result;
+    }
+}

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

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

src/main/java/io/zipcoder/tc_spring_poll_application/utils/ResultCalculator.java

@@ -0,0 +1,52 @@
+package io.zipcoder.tc_spring_poll_application.utils;
+
+import io.zipcoder.tc_spring_poll_application.domain.Option;
+import io.zipcoder.tc_spring_poll_application.dtos.OptionCount;
+
+import java.util.AbstractCollection;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Optional;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.utils
+ * author: https://github.com/vvmk
+ * date: 4/6/18
+ */
+public class ResultCalculator extends AbstractCollection<OptionCount> {
+
+    private HashMap<Long, OptionCount> optionCounts;
+    private int size;
+
+    public ResultCalculator() {
+        optionCounts = new HashMap<>();
+    }
+
+    public void add(Option option) {
+        size++;
+        Long oId = option.getId();
+
+        Optional<OptionCount> check = Optional.ofNullable(getOptionCountById(oId));
+        OptionCount workingCount = check.orElse(new OptionCount(oId));
+
+        //an exercise in restraint to maintain the integrity of OptionCount as a pure DTO
+        workingCount.setCount(workingCount.getCount() + 1);
+
+        optionCounts.put(oId, workingCount);
+    }
+
+    public OptionCount getOptionCountById(Long id) {
+        return optionCounts.get(id);
+    }
+
+    @Override
+    public Iterator<OptionCount> iterator() {
+        return optionCounts.values().iterator();
+    }
+
+    @Override
+    public int size() {
+        return size;
+    }
+}

src/test/java/io/zipcoder/tc_spring_poll_application/utils/ResultCalculatorTest.java

@@ -0,0 +1,70 @@
+package io.zipcoder.tc_spring_poll_application.utils;
+
+import io.zipcoder.tc_spring_poll_application.domain.Option;
+import org.junit.Before;
+import org.junit.Test;
+
+import static junit.framework.TestCase.assertEquals;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.utils
+ * author: https://github.com/vvmk
+ * date: 4/6/18
+ */
+public class ResultCalculatorTest {
+
+    private ResultCalculator rc = new ResultCalculator();
+    private Option o1 = new Option();
+    private Option o2 = new Option();
+
+    @Before
+    public void setup() {
+        o1.setId(1L);
+        o1.setValue("Kirk");
+
+        o2.setId(2L);
+        o2.setValue("Picard");
+    }
+
+    @Test
+    public void addShouldCountOptions() {
+        int expected = rc.size() + 2;
+
+        rc.add(o1);
+        rc.add(o2);
+
+        int actual = rc.size();
+
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void addShouldCountDuplicateVotes() {
+        int expected = rc.size() + 3;
+
+        rc.add(o1);
+        rc.add(o2);
+        rc.add(o1);
+
+        int actual = rc.size();
+
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void addShouldTallyVotes() {
+        int expected = 3;
+
+        rc.add(o1);
+        rc.add(o1);
+        rc.add(o1);
+
+        rc.add(o2);
+        rc.add(o2);
+
+        int actual = rc.getOptionCountById(o1.getId()).getCount();
+
+        assertEquals(expected, actual);
+    }
+}