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<PYEveryMomentMsgVO> 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;
    }
}