HashSet 对象去重复处理

背景

1. 需要请求两个批量接口，一个需要的数据是String类型的数组，一个需要的对象类型的数组
2. 为了防止重复调用接口中的数据，需要把请求的数据进行去重复。

选择使用集合

HashSet类（散列存放）

Java.util.HashSet类实现了Java.util.Set接口。

- 它不允许出现重复元素；

- 不保证集合中元素的顺序

- 允许包含值为null的元素，但最多只能有一个null元素。

public class BrandInfo {
    private String brandId;
    private String storeId;

    public String getBrandId() {
        return brandId;
    }

    public void setBrandId(String brandId) {
        this.brandId = brandId;
    }

    public String getStoreId() {
        return storeId;
    }

    public void setStoreId(String storeId) {
        this.storeId = storeId;
    }
}


           

public class SetTest {

    public static void main(String[] args) {
        //String类型
        HashSet<String> stringHashSet = new HashSet<>();
        stringHashSet.add("wangjie");
        stringHashSet.add("ABC");
        stringHashSet.add("BC");
        stringHashSet.add("ABC");
        System.out.println("stringHashSet size "+stringHashSet.size());

        //对象类型
        HashSet<BrandInfo> brandInfoHashSet = new HashSet<>();
        BrandInfo brandInfo1 = new BrandInfo();
        brandInfo1.setBrandId("brand1");
        brandInfo1.setStoreId("store1");

        BrandInfo brandInfo2 = new BrandInfo();
        brandInfo2.setBrandId("brand2");
        brandInfo2.setStoreId("store2");

        BrandInfo brandInfo3 = new BrandInfo();
        brandInfo3.setBrandId("brand1");
        brandInfo3.setStoreId("store1");

        brandInfoHashSet.add(brandInfo1);
        brandInfoHashSet.add(brandInfo2);
        brandInfoHashSet.add(brandInfo3);
        System.out.println("brandInfoHashSet  size "+brandInfoHashSet.size());
    }
}
           

stringHashSet size 
brandInfoHashSet  size 
           

brandInfoHashSet 中包含两个相同的对象 brandInfo1 和 brandInfo3

重写类的 equals 和 hashCode 方法

public class BrandInfo {
    private String brandId;
    private String storeId;

    public String getBrandId() {
        return brandId;
    }

    public void setBrandId(String brandId) {
        this.brandId = brandId;
    }

    public String getStoreId() {
        return storeId;
    }

    public void setStoreId(String storeId) {
        this.storeId = storeId;
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;

        BrandInfo brandInfo = (BrandInfo) o;

        if (!brandId.equals(brandInfo.brandId)) return false;
        return storeId.equals(brandInfo.storeId);
    }

    @Override
    public int hashCode() {
        int result = brandId.hashCode();
        result =  * result + storeId.hashCode();
        return result;
    }
}
           

再运行 SetTest 类

stringHashSet size 
brandInfoHashSet  size