List集合根据相同属性合并同类项

使用场景:

开发中遇到一个需要对导入的Excel数据进行数据校验的需求,但是同一列的数据校验提示语可能就只是行数不同,其他都相同,由于我们的数据校验提示都是放到消息盒子里的,所以为了阅读方便,需要对同类型的校验信息合并同类项。

处理逻辑:

1.定义验证信息类

需要重写equals入hashcode方法,定义merge方法(合并的逻辑:我这里是将行数拼接)

import lombok.Data;

@Data
public class VerifyMessage {
    private String sheetName;
    private String columnName;
    private String rowNum;
    private Integer verifyType;

    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((sheetName == null) ? 0 : sheetName.hashCode());
        result = prime * result + ((columnName == null) ? 0 : columnName.hashCode());
        result = prime * result + ((verifyType == null) ? 0 : verifyType.hashCode());
        return result;
    }

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

        VerifyMessage other = (VerifyMessage) obj;
        if (sheetName == null) {
            if (other.sheetName != null)
                return false;
        } else if (!sheetName.equals(other.sheetName))
            return false;

        if (columnName == null) {
            if (other.columnName != null)
                return false;
        } else if (!columnName.equals(other.columnName))
            return false;

        if (verifyType == null) {
            if (other.verifyType != null)
                return false;
        } else if (!verifyType.equals(other.verifyType))
            return false;
        return true;
    }

    static VerifyMessage merge(VerifyMessage m1, VerifyMessage m2) {
        if (!m1.equals(m2)) {
            throw new IllegalArgumentException();
        }
        return new VerifyMessage(m1.sheetName, m1.columnName, m1.rowNum + "," + m2.rowNum, m1.verifyType);
    }

    @Override
    public String toString() {
        return "VerifyMessage{" +
                "sheetName='" + sheetName + '\'' +
                ", columnName='" + columnName + '\'' +
                ", rowNum='" + rowNum + '\'' +
                ", verifyType=" + verifyType +
                '}';
    }

    public VerifyMessage(String sheetName, String columnName, String rowNum, Integer verifyType) {
        this.sheetName = sheetName;
        this.columnName = columnName;
        this.rowNum = rowNum;
        this.verifyType = verifyType;
    }
}

2.创建测试类 

public static void main(String[] args) {
    List<VerifyMessage> verifyMessageList = new ArrayList<>();
    verifyMessageList.add(new VerifyMessage("排班及目标导入","排班时长","1",SZGS.num));
    verifyMessageList.add(new VerifyMessage("排班及目标导入","排班时长","3",SZGS.num));
    verifyMessageList.add(new VerifyMessage("排班及目标导入","排班时长","7",SZGS.num));
    verifyMessageList.add(new VerifyMessage("排班及目标导入","排班时长","9",SZGS.num));
    verifyMessageList.add(new VerifyMessage("人员数据-HR","日期","5",RQGS.num));
    verifyMessageList.add(new VerifyMessage("人员数据-HR","日期","14",RQGS.num));
    verifyMessageList.add(new VerifyMessage("人员数据-HR","日期","19",RQGS.num));
    Map<VerifyMessage, VerifyMessage> map = new HashMap<>();
    for (VerifyMessage verifyMessage : verifyMessageList) {
        if (map.containsKey(verifyMessage)) {
            map.put(verifyMessage, VerifyMessage.merge(map.get(verifyMessage), verifyMessage));
        } else {
            map.put(verifyMessage, verifyMessage);
        }
    }
    for (VerifyMessage verifyMessage :map.values()){
        System.out.println(verifyMessage);
    }
}

输出:

VerifyMessage{sheetName='人员数据-HR', columnName='日期', rowNum='5,14,19', verifyType=1}
VerifyMessage{sheetName='排班及目标导入', columnName='排班时长', rowNum='1,3,7,9', verifyType=2}

总结

可以根据自己的实际需求更改合并逻辑。

猜你喜欢

转载自blog.csdn.net/wangpei930228/article/details/108867240
今日推荐