package com.zhujizheng.IHome.everymomentmsg.vo; import lombok.Data; import lombok.extern.slf4j.Slf4j; import java.util.ArrayList; import java.util.List; @Slf4j @Data public class PYEveryMomentUnreadMsgVO { /** 未读消息总数 **/ private Integer msgCount; /** 动态未读消息 **/ private ArrayList emArr; /** 评论未读消息 **/ private ArrayList commentArr; /** 点赞未读消息 **/ private ArrayList thumbupArr; public static PYEveryMomentUnreadMsgVO createVO(List list) { PYEveryMomentUnreadMsgVO unreadMsgVO = new PYEveryMomentUnreadMsgVO(); unreadMsgVO.msgCount = list.size(); ArrayList emArr = new ArrayList(); ArrayList commentArr = new ArrayList(); ArrayList thumbupArr = new ArrayList(); for (PYEveryMomentMsgVO msgVO : list) { if (msgVO.getMsgType() == 1) { // 发布动态 emArr.add(msgVO); } else if (msgVO.getMsgType() == 2) { // 评论 commentArr.add(msgVO); } else if (msgVO.getMsgType() == 3) { // 点赞 thumbupArr.add(msgVO); } } unreadMsgVO.emArr = emArr; unreadMsgVO.commentArr = commentArr; unreadMsgVO.thumbupArr = thumbupArr; log.info("unreadMsgVO = " + unreadMsgVO); return unreadMsgVO; } }