stats

fabioesposito · Aug 18, 2017 · 3619dc4 · 3619dc4
1 parent 379e1cb
commit 3619dc4
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/survey-common/src/main/java/com/surveygorilla/common/event/AnswerSubmitted.java b/survey-common/src/main/java/com/surveygorilla/common/event/AnswerSubmitted.java
@@ -6,16 +6,20 @@
  * Created by fabio.pires on 18.08.17.
  */
 public class AnswerSubmitted {
+
     private Integer pollID;
 
     private Map<Integer, Integer> answers;
 
+    private Map<Integer, Double> stats;
+
     public AnswerSubmitted() {
     }
 
-    public AnswerSubmitted(Integer pollID, Map<Integer, Integer> answers) {
+    public AnswerSubmitted(Integer pollID, Map<Integer, Integer> answers, Map<Integer, Double> stats) {
         this.pollID = pollID;
         this.answers = answers;
+        this.stats = stats;
     }
 
     public Integer getPollID() {
@@ -33,4 +37,12 @@ public Map<Integer, Integer> getAnswers() {
     public void setAnswers(Map<Integer, Integer> answers) {
         this.answers = answers;
     }
+
+    public Map<Integer, Double> getStats() {
+        return stats;
+    }
+
+    public void setStats(Map<Integer, Double> stats) {
+        this.stats = stats;
+    }
 }
diff --git a/survey-read/src/main/java/com/surveygorilla/read/SurveyReadVerticle.java b/survey-read/src/main/java/com/surveygorilla/read/SurveyReadVerticle.java
@@ -73,10 +73,11 @@ private Handler<Message<String>> answerSubmittedEventListener() {
                     AnswerSubmitted.class);
 
             Poll p = polls.get(answerSubmitted.getPollID());
-            // Todo stats
 
             if (p != null) {
                 p.getAnswers().putAll(answerSubmitted.getAnswers());
+                p.getStats().putAll(answerSubmitted.getStats());
+
                 polls.put(p.getPollID(), p);
             }
         };

diff --git a/survey-write/src/main/java/com/surveygorilla/write/SurveyWriteVerticle.java b/survey-write/src/main/java/com/surveygorilla/write/SurveyWriteVerticle.java
@@ -18,6 +18,7 @@
 import java.util.LinkedHashMap;
 import java.util.Map;
 import java.util.concurrent.atomic.AtomicInteger;
+import java.util.stream.Stream;
 
 
 /**
@@ -90,9 +91,16 @@ private Handler<Message<Object>> submitAnswer() {
                     p.getAnswers().merge(a, 1, Integer::sum);
                 });
 
+                Double sum = p.getAnswers().values().stream().mapToDouble(s -> s).sum();
+
+                p.getAnswers().forEach((key, value) -> {
+                    Double percentage = value / sum;
+                    p.getStats().put(key, percentage);
+                });
+
                 polls.put(p.getPollID(), p);
 
-                final AnswerSubmitted answerSubmitted = new AnswerSubmitted(p.getPollID(), p.getAnswers());
+                final AnswerSubmitted answerSubmitted = new AnswerSubmitted(p.getPollID(), p.getAnswers(), p.getStats());
                 vertx.eventBus().send("answerSubmittedEvent", Json.encode(answerSubmitted));
 
                 handler.reply(Json.encodePrettily(p));