package com.java110.common.cmd.area;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.java110.core.annotation.Java110Cmd;
|
import com.java110.core.context.ICmdDataFlowContext;
|
import com.java110.core.event.cmd.Cmd;
|
import com.java110.core.event.cmd.CmdEvent;
|
import com.java110.dto.area.AreaDto;
|
import com.java110.intf.common.IAreaInnerServiceSMO;
|
import com.java110.utils.exception.CmdException;
|
import com.java110.utils.util.ListUtil;
|
import com.java110.vo.ResultVo;
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import java.util.List;
|
|
/**
|
* 查询地区树形结构
|
*/
|
@Java110Cmd(serviceCode = "area.queryAreaTree")
|
public class QueryAreaTreeCmd extends Cmd {
|
|
@Autowired
|
private IAreaInnerServiceSMO areaInnerServiceSMOImpl;
|
|
@Override
|
public void validate(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
|
|
}
|
|
@Override
|
public void doCmd(CmdEvent event, ICmdDataFlowContext context, JSONObject reqJson) throws CmdException {
|
|
AreaDto areaDto = new AreaDto();
|
List<AreaDto> areaDtos = areaInnerServiceSMOImpl.getArea(areaDto);
|
|
JSONObject areaData = new JSONObject();
|
areaData.put("id", "0");
|
areaData.put("areaCode", "0");
|
|
areaData.put("text", "地区");
|
areaData.put("icon", "/img/org.png");
|
areaData.put("children", new JSONArray());
|
JSONArray areas = areaData.getJSONArray("children");
|
if (ListUtil.isNull(areaDtos)) {
|
context.setResponseEntity(ResultVo.createResponseEntity(areas));
|
return;
|
}
|
|
JSONObject areaInfo = null;
|
for (AreaDto tmpAreaDto : areaDtos) {
|
if (AreaDto.AREA_LEVEL_ONE.equals(tmpAreaDto.getAreaLevel())) {
|
areaInfo = new JSONObject();
|
areaInfo.put("id", tmpAreaDto.getId());
|
areaInfo.put("areaCode", tmpAreaDto.getAreaCode());
|
areaInfo.put("areaLevel", tmpAreaDto.getAreaLevel());
|
areaInfo.put("text", tmpAreaDto.getAreaName());
|
areaInfo.put("icon", "/img/org.png");
|
areaInfo.put("children", new JSONArray());
|
areas.add(areaInfo);
|
}
|
}
|
|
JSONObject area = null;
|
for (int cIndex = 0; cIndex < areas.size(); cIndex++) {
|
area = areas.getJSONObject(cIndex);
|
// find floor data in unitDtos
|
findTwoLevel(area, areaDtos);
|
}
|
context.setResponseEntity(ResultVo.createResponseEntity(areaData));
|
|
}
|
|
private void findTwoLevel(JSONObject area, List<AreaDto> areaDtos) {
|
|
JSONArray childrens = area.getJSONArray("children");
|
JSONObject childrenInfo = null;
|
for (AreaDto tmpAreaDto : areaDtos) {
|
if (!AreaDto.AREA_LEVEL_TWO.equals(tmpAreaDto.getAreaLevel())) {
|
continue;
|
}
|
if (area.getString("areaCode").equals(tmpAreaDto.getParentAreaCode())) {
|
childrenInfo = new JSONObject();
|
childrenInfo.put("id", tmpAreaDto.getId());
|
childrenInfo.put("areaCode", tmpAreaDto.getAreaCode());
|
childrenInfo.put("areaLevel", tmpAreaDto.getAreaLevel());
|
childrenInfo.put("text", tmpAreaDto.getAreaName());
|
childrenInfo.put("children", new JSONArray());
|
childrenInfo.put("icon", "/img/floor.png");
|
childrens.add(childrenInfo);
|
}
|
}
|
|
|
JSONObject floor = null;
|
for (int cIndex = 0; cIndex < childrens.size(); cIndex++) {
|
floor = childrens.getJSONObject(cIndex);
|
// find floor data in unitDtos
|
findThreeLevel(floor, areaDtos);
|
}
|
}
|
|
private void findThreeLevel(JSONObject twoArea, List<AreaDto> areaDtos) {
|
JSONArray childrens = twoArea.getJSONArray("children");
|
JSONObject childrenInfo = null;
|
for (AreaDto tmpAreaDto : areaDtos) {
|
if (!AreaDto.AREA_LEVEL_THREE.equals(tmpAreaDto.getAreaLevel())) {
|
continue;
|
}
|
if (twoArea.getString("areaCode").equals(tmpAreaDto.getParentAreaCode())) {
|
childrenInfo = new JSONObject();
|
childrenInfo.put("id", tmpAreaDto.getId());
|
childrenInfo.put("areaCode", tmpAreaDto.getAreaCode());
|
childrenInfo.put("areaLevel", tmpAreaDto.getAreaLevel());
|
childrenInfo.put("text", tmpAreaDto.getAreaName());
|
childrenInfo.put("children", new JSONArray());
|
childrenInfo.put("icon", "/img/unit.png");
|
childrens.add(childrenInfo);
|
}
|
}
|
}
|
}
|
