avg in sum builder

Kunagi · Apr 29, 2015 · 5f7dc15 · 5f7dc15
1 parent 625b42b
commit 5f7dc15
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/ilarkesto/core/base/BigDecimalSumBuilder.java b/src/main/java/ilarkesto/core/base/BigDecimalSumBuilder.java
@@ -22,8 +22,10 @@
 public class BigDecimalSumBuilder implements Formatable {
 
 	private BigDecimal sum;
+	private int count;
 
 	public BigDecimalSumBuilder add(Number value) {
+		count++;
 		if (value == null) return this;
 		BigDecimal bdValue = ((value instanceof BigDecimal) ? (BigDecimal) value : new BigDecimal(value.toString()));
 		if (sum == null) {
@@ -38,6 +40,10 @@ public BigDecimal getSum() {
 		return sum;
 	}
 
+	public BigDecimal getAvg(int scale) {
+		return sum.divide(new BigDecimal(count), scale, BigDecimal.ROUND_HALF_UP);
+	}
+
 	@Override
 	public String format() {
 		return sum == null ? null : Localizer.get().format(sum, true, 2);