java - ArrayList 'squash' 对其元素的操作

Question

我有一个看似微不足道的问题，但我正在寻找解决该问题的最佳方法。

假设我有一个类:

产品类

public class Product {
    private int id;
    private String code;
    private String price;
    private String quantity;

    public Product() {}

    //getters and setters

    @Override
    public boolean equals(Object obj) {
        boolean result = false;
        if (obj instanceof ProductOnStockDto) {
            ProductOnStockDto product = (ProductOnStockDto) obj;
            result = (this.code.equals(product.getCode()) && this.price.equals(product.getPrice()));
        }
        return result;
    }
}

假设我有一个对象列表:

List<Product> products;

其中填充了类似 this 的数据(1 行是一个列表元素):

我需要的是遍历此列表并压缩具有相同代码和相同价格的元素。例如:

在结果列表中，我想为 code 3 提供两个产品:

code 3 | 1.22 | 11
code 3 | 2.20 | 6

所以我会这样做:

for(Product product : products) {
    for(Product productToSquash : products) {
        if(product.getId() != productToSquash.getId() && product.equals(productToSquash)) {
            //Here I would like to squash these two rows
            //Data conversion is ommited
            product.setQuantity(product.getQuantity() + productToSquash.getQuantity());
            //after that I would like to remove productToSquash object from the collection
        }
    }
}

但据我所知，不允许修改我正在迭代的集合。那么根据示例压缩所有产品列表的最佳方法是什么？

Answer 1

首先，您的 equals()方法引用ProductOnStockDto .那应该是Product .

要在迭代期间删除元素，请使用 Iterator直接，即使用“旧式”for 循环，并使用 Map<Product, Product>跟踪以前看到的产品。这要求您还实现 hashCode() :

@Override
public int hashCode()
{
    return this.code.hashCode() * 37 + this.price.hashCode();
}

Map<Product, Product> map = new HashMap<Product, Product>();
for (Iterator<Product> productIter = products.iterator(); productIter.hasNext(); ) {
    Product product = productIter.next();
    Product productToKeep = map.get(product);
    if (productToKeep == null)
        map.put(product, product);
    else {
        productToKeep.setQuantity(productToKeep.getQuantity() + product.getQuantity());
        productIter.remove();
    }
}

你真的不应该这样做，因为 equals()方法正在返回 true对于不是真正相等的对象。

你应该有一个只有 code 的键类和 price ，关键类是需要实现的类 equals()和 hashCode() , 不是 Product .