package com.moral.api.pojo.vo.groupMenu; import com.fasterxml.jackson.annotation.JsonInclude; import com.moral.api.entity.GroupMenu; import com.moral.api.pojo.dto.groupMenu.GroupMenuDTO; import com.moral.api.pojo.dto.groupMenu.GroupMenuQueryDTO; import com.moral.constant.ResponseCodeEnum; import lombok.Data; import java.util.ArrayList; import java.util.List; /** * @ClassName GroupMenuQueryVO * @Description TODO * @Author 陈凯裕 * @Date 2021/5/11 14:42 * @Version TODO **/ @Data @JsonInclude(JsonInclude.Include.NON_EMPTY) public class GroupMenuQueryVO { List menus; public static GroupMenuQueryVO convert(GroupMenuQueryDTO dto) { if (dto.getCode() != ResponseCodeEnum.SUCCESS.getCode()) return null; GroupMenuQueryVO queryVo = new GroupMenuQueryVO(); List menus = new ArrayList<>(); List menuDtos = dto.getDtos(); for (GroupMenuDTO menuDto : menuDtos) { GroupMenuVO vo = convertToQueryPage(menuDto); menus.add(vo); } queryVo.setMenus(menus); return queryVo; } private static GroupMenuVO convertToQueryPage(GroupMenuDTO dto){ GroupMenuVO vo = new GroupMenuVO(); GroupMenu groupMenu = dto.getGroupMenu(); vo.setId(groupMenu.getId()); vo.setChannelKey(groupMenu.getChannelKey()); vo.setMenuId(groupMenu.getMenuId()); return vo; } }