package com.moral.controller;
|
|
import com.moral.common.bean.ResultBean;
|
import com.moral.entity.OrganizationSensorUnit;
|
import com.moral.entity.SensorUnit;
|
import com.moral.service.OrganizationSensorUnitService;
|
import com.moral.service.SensorUnitService;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.annotation.Resource;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
@RestController
|
@RequestMapping("org-sunit")
|
public class OrganizationSensorUnitController {
|
@Resource
|
OrganizationSensorUnitService osuService;
|
@Resource
|
SensorUnitService sensorUnitService;
|
/**
|
* 获取组织和单位对应关系实体组
|
* @param orgId
|
* @return
|
*/
|
@GetMapping("gets-byoid")
|
public ResultBean getListByorgId(Integer orgId) {
|
Map resultData = new HashMap();
|
List<OrganizationSensorUnit> osuList = osuService.queryAllByOrgId(orgId);
|
if(osuList!=null && osuList.size()>0) {
|
List<Integer> sensorIdList = osuList.stream().map(item -> {
|
return item.getSensor().getId();
|
}).collect(Collectors.toList());
|
Map sensorUnitMap = sensorUnitService.queryGroupSensorBySids(sensorIdList);
|
// 只输出有选项的
|
List resultOsuList = osuList.stream().filter(item -> {
|
Integer id = item.getSensor().getId();
|
return sensorUnitMap.get(id)!=null;
|
}).collect(Collectors.toList());
|
resultData.put("osuList",resultOsuList);
|
resultData.put("sensorUnitMap",sensorUnitMap);
|
}else {
|
resultData.put("osuList",new ArrayList<>()); // 组织和待选单位
|
resultData.put("sensorUnitMap",new HashMap<>()); // 待选单位
|
}
|
return new ResultBean(resultData);
|
}
|
@PostMapping("saves")
|
public ResultBean saves(@RequestBody List<OrganizationSensorUnit> osuList){
|
osuService.addsOrModifys(osuList);
|
return ResultBean.success();
|
}
|
}
|
