@Override public List<CatalogItem> listCatalogItems() { List<Catalog> catalogs = findAll(); if (CollectionUtils.isEmpty(catalogs)) return new ArrayList<CatalogItem>(); Map<Long, List<CatalogItem>> thirdChildrenMap = new LinkedHashMap<Long, List<CatalogItem>>(); for (Catalog c : catalogs) { if (c.getLevel().ordinal() == CatalogLevel.THIRD.ordinal()) { if (!thirdChildrenMap.containsKey(c.getPid())) thirdChildrenMap.put(c.getPid(), new ArrayList<CatalogItem>()); CatalogItem ct = new CatalogItem(); ct.setCatalog(c); thirdChildrenMap.get(c.getPid()).add(ct); } } Map<Long, List<CatalogItem>> secondChildrenMap = new LinkedHashMap<Long, List<CatalogItem>>(); for (Catalog c : catalogs) { if (c.getLevel().ordinal() == CatalogLevel.SECOND.ordinal()) { if (!secondChildrenMap.containsKey(c.getPid())) secondChildrenMap.put(c.getPid(), new ArrayList<CatalogItem>()); CatalogItem ct = new CatalogItem(); ct.setCatalog(c); ct.setSubCatalog(thirdChildrenMap.get(c.getId())); secondChildrenMap.get(c.getPid()).add(ct); } } List<CatalogItem> list = new ArrayList<CatalogItem>(); for (Catalog c : catalogs) { if (c.getLevel().ordinal() == CatalogLevel.FIRST.ordinal()) { CatalogItem ct = new CatalogItem(); ct.setCatalog(c); ct.setSubCatalog(secondChildrenMap.get(c.getId())); list.add(ct); } } return list; }
@Override public Catalog saveCatalog(Catalog catalog) { Catalog sameAliasCatalog = findByAlias(catalog.getModule(), catalog.getAlias()); if ((catalog.getId() == null && sameAliasCatalog != null) || (catalog.getId() != null && sameAliasCatalog != null && !sameAliasCatalog.getId().equals(catalog.getId()))) throw new AppException(ErrorCode.ENTITY_ALREADY_EXIST, "访问别名已被使用,请换一个!"); Date now = new Date(); Catalog entity = null; if (catalog.getId() == null) { entity = new Catalog(); entity.setUpdateAt(now); entity.setCreateAt(now); } else { entity = catalogDao.findOne(catalog.getId()); if (entity == null) throw new AppException(ErrorCode.ENTITY_NOT_FOUND, "要更新的栏目没有找到!"); entity.setUpdateAt(now); } entity.setAlias(catalog.getAlias()); entity.setClickDisplay(catalog.getClickDisplay()); entity.setLevel(getSetLevel(catalog)); entity.setModule(catalog.getModule()); entity.setNavDisplay(catalog.getNavDisplay()); entity.setNewWindowOpen( catalog.getNewWindowOpen() != null ? catalog.getNewWindowOpen() : false); entity.setPid( catalog.getPid() == null ? null : (catalog.getPid() == 0 ? null : catalog.getPid())); entity.setRank(catalog.getRank() != null ? catalog.getRank() : 0); entity.setSubTitle(catalog.getSubTitle()); entity.setTitle(catalog.getTitle()); if (entity.getModule().ordinal() == Module.outlink.ordinal()) { if (entity.getAttributes() == null) entity.setAttributes(new HashMap<String, String>()); entity.getAttributes().put(Catalog.ATTR_KEY_OUTLINK_URL, catalog.getOutlinkUrl()); } return catalogDao.save(entity); }