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();
    }
}