package com.example.erp.service.sd; import com.alibaba.fastjson.JSON; import com.baomidou.dynamic.datasource.annotation.DS; import com.example.erp.entity.sd.BasicGlassType; import com.example.erp.mapper.sd.BasicGlassTypeMapper; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.*; @Service @DS("erp_sd") public class BasicGlassTypeServise { @Autowired BasicGlassTypeMapper basicGlassTypeMapper; public List getOneLevel() { return basicGlassTypeMapper.getOneLevel(); } public List getTwoLevel() { return basicGlassTypeMapper.getTwoLevel(); } public List> getAll() { List oneGlassType = basicGlassTypeMapper.getOneLevel(); List twoGlassType = basicGlassTypeMapper.getTwoLevel(); List> list = new ArrayList<>(); for (BasicGlassType glassType : oneGlassType) { Map map = new HashMap<>(); map.put("value", glassType.getTypeID()); map.put("label", glassType.getTypeName()); map.put("children", ""); List equalList = new ArrayList<>(); for (BasicGlassType basicGlassType : twoGlassType) { if (Objects.equals(basicGlassType.getBelong(), glassType.getTypeID())) { Map getMap = JSON.parseObject(JSON.toJSONString(basicGlassType), Map.class); getMap.put("value",getMap.get("typeID")); getMap.put("label",getMap.get("typeName")); equalList.add(JSON.toJSONString(getMap) ); } } map.replace("children",equalList.toString()); list.add(map); } return list; } }