[#23] refactor: 요금 차트를 위한 요금 카운트 기능 추가

codesquad-member-2020 · May 28, 2020 · dcb25e9 · dcb25e9
1 parent 529d124
commit dcb25e9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/BE/src/main/java/com/codesquad/airbnb/domain/dto/Statistics.java b/BE/src/main/java/com/codesquad/airbnb/domain/dto/Statistics.java
@@ -20,4 +20,6 @@ public class Statistics {
     private int averagePrice;
 
     private List<Integer> prices;
+
+    private int[] counts;
 }
diff --git a/BE/src/main/java/com/codesquad/airbnb/infra/dao/ViewDAO.java b/BE/src/main/java/com/codesquad/airbnb/infra/dao/ViewDAO.java
@@ -107,7 +107,18 @@ public Integer mapRow(ResultSet rs, int rowNum) throws SQLException {
             averagePrice = calculateAverage(prices);
         }
 
-        return new Statistics(lowestPrice, highestPrice, averagePrice, prices);
+        int endPrice = 1000000;
+        int divide = 20000;
+        int[] counts = new int[endPrice/divide];
+
+        for (int price : prices) {
+            if((price/divide) >= (endPrice/divide)) {
+                counts[(endPrice/divide)-1]++;
+            }
+            counts[price/divide] ++;
+        }
+
+        return new Statistics(lowestPrice, highestPrice, averagePrice, prices, counts);
     }
 
     private Integer calculateAverage(List<Integer> prices) {