Stream APIを使い、グループ化して合計を求める

ソース

public class Product {

	/** 商品のID*/
	private String id;

	/** 商品の価格 */
	private int price;

	public Product(String id, int price) {
		super();
		this.id = id;
		this.price = price;
	}

	public String getId() {
		return id;
	}

	public int getPrice() {
		return price;
	}

	@Override
	public String toString() {
		return "Product [id=" + id + ", price=" + price + "]";
	}

}

Baseクラス

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class Base {

	public void test() {
		System.out.println("-----------------------------");
		System.out.println("実行するクラス");
		System.out.println(getClass().getSimpleName());
		System.out.println("-----------------------------");

		// データ取得
		List<Product> list = getData();

		// データ出力
		for (Product product : list) {
			System.out.println(product);
		}

		// グループ化
		Map<String, Integer> sums = calc(list);

		// 結果出力
		System.out.println("-----------------------");
		System.out.println("グループごとの合計");
		System.out.println("-----------------------");
		System.out.println(sums);
	}

	protected Map<String, Integer> calc(List<Product> list) {

		// グループ化の結果を格納するマップ
		Map<String, Integer> sums = new HashMap<>();

		// -----------------------------------
		// TODO 以下、グループ化して合計を求める処理
		// -----------------------------------

		return sums;
	}

	private List<Product> getData() {
		List<Product> list = new ArrayList<>();

		// データ生成
		list.add(new Product("A001", 10));
		list.add(new Product("B001", 100));
		list.add(new Product("B002", 200));
		list.add(new Product("C001", 1000));
		list.add(new Product("C002", 2000));
		list.add(new Product("C003", 3000));

		// シャッフル
		Collections.shuffle(list);

		return list;
	}

}

上記サンプルのテスト

public class BaseTest {

	public static void main(String[] args) {
		Base base = new Base();
		base.test();
	}

}

結果

-----------------------------
実行するクラス
Base
-----------------------------
Product [id=A001, price=10]
Product [id=C003, price=3000]
Product [id=B002, price=200]
Product [id=B001, price=100]
Product [id=C001, price=1000]
Product [id=C002, price=2000]
-----------------------
グループごとの合計
-----------------------
{}

SampleAクラス

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

public class SampleA extends Base {

	@Override
	protected Map<String, Integer> calc(List<Product> list) {
		// グループ化の結果を格納するマップ
		Map<String, Integer> sums = new HashMap<>();

		// (A-1) グループ化のキーにする文字を用意
		Set<String> targetSigns = new HashSet<>();
		for (Product product : list) {
			targetSigns.add(product.getId().substring(0, 1));
		}

		// (A-2) 商品をグループ化する。
		Map<String, List<Product>> group = new HashMap<>();

		Iterator<String> targetSignsIterator = targetSigns.iterator();

		while (targetSignsIterator.hasNext()) {
			String targetSign = targetSignsIterator.next();

			List<Product> products = new ArrayList<>();

			for (Product product : list) {
				String sign = product.getId().substring(0, 1);

				if (targetSign.equals(sign)) {
					products.add(product);
				}
			}

			group.put(targetSign, products);
		}

		// (A-3) グループ各々について、価格を合計する。
		Iterator<String> groupKeysIterator = group.keySet().iterator();

		while (groupKeysIterator.hasNext()) {
			String sign = groupKeysIterator.next();

			// (A-4) 指定したグループの合計を求める。
			int sum = 0;
			List<Product> products = group.get(sign);

			for (Product product : products) {
				sum += product.getPrice();
			}

			sums.put(sign, sum);
		}

		return sums;
	}
}

SampleAクラスのテスト

public class SampleATest {

	public static void main(String[] args) {
		SampleA sampleA = new SampleA();
		sampleA.test();
	}

}

結果

-----------------------------
実行するクラス
SampleA
-----------------------------
Product [id=C002, price=2000]
Product [id=B001, price=100]
Product [id=A001, price=10]
Product [id=C003, price=3000]
Product [id=B002, price=200]
Product [id=C001, price=1000]
-----------------------
グループごとの合計
-----------------------
{A=10, B=300, C=6000}

ソース

import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class SampleB extends Base {

	@Override
	protected Map<String, Integer> calc(List<Product> list) {
		// グループ化の結果を格納するマップ
		Map<String, Integer> sums = new HashMap<>();

		// (B-1) グループ化のキーにする文字を用意
		// 後のStream APIの処理に含まれるので、処理なし。

		// (B-2) 商品をグループ化する。
		Map<String, List<Product>> group = list
				.stream()
				.collect(Collectors.groupingBy(
						p -> p.getId().substring(0, 1)));

		// (B-3) グループ各々について、価格を合計する。
		Iterator<String> groupKeysIterator = group.keySet().iterator();

		while (groupKeysIterator.hasNext()) {
			String sign = groupKeysIterator.next();

			// (B-4) 指定したグループの合計を求める。
			int sum = group.get(sign)
					.stream()
					.mapToInt(product -> product.getPrice())
					.sum();

			sums.put(sign, sum);
		}

		return sums;
	}
}

SampleBクラスのテスト

public class SampleBTest {

	public static void main(String[] args) {
		SampleB sampleB = new SampleB();
		sampleB.test();
	}

}

結果

-----------------------------
実行するクラス
SampleB
-----------------------------
Product [id=C001, price=1000]
Product [id=B001, price=100]
Product [id=C002, price=2000]
Product [id=C003, price=3000]
Product [id=B002, price=200]
Product [id=A001, price=10]
-----------------------
グループごとの合計
-----------------------
{A=10, B=300, C=6000}

ProductWithKey

public class ProductWithKey extends Product {

	/** グループ化のキー */
	private String key;

	public ProductWithKey(String id, int price) {
		super(id, price);
		this.key = id.substring(0, 1);
	}

	public String getKey() {
		return key;
	}

}

SampleC

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class SampleC extends Base {

	@Override
	protected Map<String, Integer> calc(List<Product> list) {
		return list
				.stream()
				.map(p -> new ProductWithKey(p.getId(), p.getPrice()))
				.collect(Collectors.groupingBy(
						ProductWithKey::getKey,
						Collectors.summingInt(ProductWithKey::getPrice)));

	}

}

SampleCクラスのテスト

public class SampleCTest {

	public static void main(String[] args) {
		SampleC sampleC = new SampleC();
		sampleC.test();
	}

}

結果

-----------------------------
実行するクラス
SampleC
-----------------------------
Product [id=B001, price=100]
Product [id=C003, price=3000]
Product [id=B002, price=200]
Product [id=A001, price=10]
Product [id=C002, price=2000]
Product [id=C001, price=1000]
-----------------------
グループごとの合計
-----------------------
{A=10, B=300, C=6000}