Trinh Tong преди 7 години
родител
ревизия
b04cbbb049
променени са 1 файла, в които са добавени 6 реда и са изтрити 8 реда
  1. 6
    8
      src/main/java/io/zipcoder/tc_spring_poll_application/controller/ComputeResultController.java

+ 6
- 8
src/main/java/io/zipcoder/tc_spring_poll_application/controller/ComputeResultController.java Целия файл

13
 import org.springframework.web.bind.annotation.RequestParam;
13
 import org.springframework.web.bind.annotation.RequestParam;
14
 import org.springframework.web.bind.annotation.RestController;
14
 import org.springframework.web.bind.annotation.RestController;
15
 
15
 
16
-import java.util.Collection;
17
 import java.util.HashMap;
16
 import java.util.HashMap;
18
-import java.util.HashSet;
19
 
17
 
20
 @RestController
18
 @RestController
21
 public class ComputeResultController {
19
 public class ComputeResultController {
42
         for (Vote v: allVotes
40
         for (Vote v: allVotes
43
              ) {
41
              ) {
44
             voteCount++;
42
             voteCount++;
45
-            Long name = (v.getOption().getId());
46
-            if (!optionCountsMap.keySet().contains(name)) {
47
-                optionCountsMap.put(name, new OptionCount());
43
+            Long optionID = (v.getOption().getId());
44
+            if (!optionCountsMap.keySet().contains(optionID)) {
45
+                optionCountsMap.put(optionID, new OptionCount());
46
+//                optionCountsMap.get(optionID).setCount(1);
48
             } else {
47
             } else {
49
-                OptionCount temp = optionCountsMap.get(name);
48
+                OptionCount temp = optionCountsMap.get(optionID);
50
                 temp.setCount(temp.getCount()+1);
49
                 temp.setCount(temp.getCount()+1);
51
-                optionCountsMap.replace(name, temp);
50
+                optionCountsMap.replace(optionID, temp);
52
             }
51
             }
53
         }
52
         }
54
 
53
 
55
         voteResult.setTotalVotes(voteCount);
54
         voteResult.setTotalVotes(voteCount);
56
-
57
         voteResult.setResults(optionCountsMap.values());
55
         voteResult.setResults(optionCountsMap.values());
58
 
56
 
59
         //TODO: Implement algorithm to count votes
57
         //TODO: Implement algorithm to count votes