功能完善

liuhaonan

2022-03-11 e5f711997c3247dc74d5a820673cc911bd2fbcf4

 ximon-admin/src/main/java/com/sandu/ximon/admin/service/PoleService.java

@@ -4,6 +4,7 @@
import cn.hutool.core.util.StrUtil;
import com.aliyuncs.iot.model.v20180120.BatchGetDeviceStateResponse;
import com.aliyuncs.iot.model.v20180120.QueryDeviceDetailResponse;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.github.pagehelper.PageHelper;
import com.sandu.common.execption.BusinessException;
@@ -27,9 +28,12 @@
import com.sandu.ximon.admin.manager.iot.rrpc.mainboard.MainBoardInvokeSyncService;
import com.sandu.ximon.admin.param.PoleBindingParam;
import com.sandu.ximon.admin.param.PoleParam;
import com.sandu.ximon.admin.param.PoleStatesParam;
import com.sandu.ximon.admin.redis.LightKey;
import com.sandu.ximon.admin.security.SecurityUtils;
import com.sandu.ximon.dao.domain.Pole;
import com.sandu.ximon.dao.domain.PoleBinding;
import com.sandu.ximon.dao.domain.PoleGroupRelation;
import com.sandu.ximon.dao.mapper.PoleMapper;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@@ -55,6 +59,7 @@

    private final RedisService redisService;
    private final PoleBindingService poleBindingService;
    private final PoleGroupRelationService groupRelationService;

    public boolean addPole(PoleParam param) {
        Pole pole = new Pole();
@@ -105,22 +110,287 @@
    }


    public List<Pole> queryAllStatesAndList(Integer pageNo, Integer pageSize, PoleStatesParam param) {
        // List<LampPost> list = list(Wrappers.lambdaQuery(LampPost.class).eq(LampPost::getClientId, SecurityUtils.getUserId()));
        //List<Pole> list = list(Wrappers.lambdaQuery(Pole.class));
//        PageHelper.startPage(pageNo, pageSize);
        List<Pole> list = new ArrayList<>();
        LambdaQueryWrapper<Pole> wrapper = new LambdaQueryWrapper<>();
        if (SecurityUtils.getClientId() == null) {
            wrapper = Wrappers.lambdaQuery(Pole.class);
        } else {
            wrapper = Wrappers.lambdaQuery(Pole.class).eq(Pole::getClientId, SecurityUtils.getUserId());
        }
        if (!param.getKeyword().isEmpty()) {
            wrapper.like(Pole::getPoleCode, param.getKeyword()).or(
                    wrappers -> {
                        wrappers.like(Pole::getPoleName, param.getKeyword());
                    }
            );
        }

        list = list(wrapper);
        List<Pole> bindList = new ArrayList<>();//已绑定客户
        List<Pole> unbindList = new ArrayList<>();//未绑定客户
        List<Pole> isTrue = new ArrayList<>();//实体灯杆
        List<Pole> isFalse = new ArrayList<>();//虚拟灯杆
        List<Pole> online = new ArrayList<>();//在线
        List<Pole> offline = new ArrayList<>();//离线
        if (param.getGroupid() != null) {
            //  List<Long> poleIds = new ArrayList<>();
            List<Pole> pole = new ArrayList<>();
            List<PoleGroupRelation> relations = groupRelationService.list(Wrappers.lambdaQuery(PoleGroupRelation.class)
                    .eq(PoleGroupRelation::getPoleGroupId, param.getGroupid()));
            relations.forEach(relation -> {
                        pole.add(getById(relation.getPoleId()));
                    }
            );
            list = pole;
        }

//        List<Pole> bind = isBind(list, param);
//        List<Pole> aTrue = isTrue(list, param);
//        List<Pole> onLine = isOnLine(list, param);

        if (param.getBingStates() != 2) {//是否绑定
            list = isBind(list, param);//   绑定/未绑定
            if (param.getIsTrue() != 2) {//已绑定的是否是真实灯杆
                list = isTrue(list, param);
                if (param.getOnLineStates() != 2) {//绑定的真实灯杆是否在线
                    list = isOnLine(list, param);
                }
            }else {//全部灯杆  实体/虚拟
                if (param.getOnLineStates() != 2) {  //查询全部灯杆是否在线
                    list = isOnLine(list, param);
                }
            }
        }else {//全部 绑定/未绑定
            if (param.getIsTrue() != 2) {// 判断灯杆类型
                list = isTrue(list, param);// 实体/虚拟
                if (param.getOnLineStates() != 2) {// 灯杆的在线判断
                    list = isOnLine(list, param);//在线/离线
                }
            }else {//全类型灯杆
                if (param.getOnLineStates() != 2) {//判断在线状态
                    list = isOnLine(list, param);//在线/离线
                }
            }
        }


//        list.forEach(bindwarpper -> {
////                Long clientId = bindwarpper.getClientId();
//            if (bindwarpper.getClientId() == null) {//判断是否绑定
//                unbindList.add(bindwarpper);
//                if (bindwarpper.getDeviceCode() == null) {//判断灯杆类型
//                    isFalse.add(bindwarpper);
//                } else {
//                    isTrue.add(bindwarpper);
//                }
//            } else {
//                bindList.add(bindwarpper);
//                if (bindwarpper.getDeviceCode() == null) {
//                    isFalse.add(bindwarpper);
//                } else {
//                    isTrue.add(bindwarpper);
//                }
//            }
//        });
//        if (param.getBingStates() == 1) { //未绑定
//            list = unbindList;
//        } else if (param.getBingStates() == 0) {//已绑定
//            list = bindList;
//        }// else {//全部
////            list = list(wrapper);
////        }
//
//        if (param.getIsTrue() == 1) { //虚拟灯杆
//            list = isFalse;
//        } else if (param.getIsTrue() == 0) {//实体灯杆
//            setOnline(list);
//            list = isTrue;
//            list.forEach(onLinePole -> {
//                System.out.println("online---------------------------------------------------"+("ONLINE").equals(onLinePole.getOnLineState()));
//                System.out.println("online---------------------------------------------------"+(onLinePole.getOnLineState()));
//                if (("ONLINE").equals(onLinePole.getOnLineState())) {
//                    online.add(onLinePole);
//                } else if (("OFFLINE").equals(onLinePole.getOnLineState())) {
//                    offline.add(onLinePole);
//                }
//            });
//        } else {//全部
//            setOnline(list);
//           list.forEach(onLinePole -> {
//           //    System.out.println("online-----------------------"+onLinePole.getOnLineState());
//               if (("ONLINE").equals(onLinePole.getOnLineState())) {
//                    online.add(onLinePole);
//                } else if (("OFFLINE").equals(onLinePole.getOnLineState())) {
//                    offline.add(onLinePole);
//                }
//            });
//       }
//
//        if (param.getOnLineStates() == 1) { //离线
//            list = offline;
//        } else if (param.getOnLineStates() == 0) {//在线
//            list = online;
//        }
        setCount(list);
        return list;
    }

    /**
     * 设置绑定设备数量
     *
     * @param list
     * @return
     */
    public List<Pole> setCount(List<Pole> list) {
        for (Pole post : list) {
            int size = poleBindingService.list(Wrappers.lambdaQuery(PoleBinding.class).eq(PoleBinding::getPoleId, post.getId())).size();
//                    System.out.println("size:----------------------"+size);
            post.setBindingCount(size);
        }
        return list;
    }


    /**
     * 在线状态处理
     *
     * @param list
     * @param param
     * @return
     */
    public List<Pole> isOnLine(List<Pole> list, PoleStatesParam param) {
        setOnline(list);
        List<Pole> online = new ArrayList<>();//在线
        List<Pole> offline = new ArrayList<>();//离线
        list.forEach(onLinePole -> {
            System.out.println("online---------------------------------------------------" + ("ONLINE").equals(onLinePole.getOnLineState()));
            System.out.println("online---------------------------------------------------" + (onLinePole.getOnLineState()));
            if (("ONLINE").equals(onLinePole.getOnLineState())) {
                online.add(onLinePole);
            } else if (("OFFLINE").equals(onLinePole.getOnLineState())) {
                offline.add(onLinePole);
            }
        });

        if (param.getOnLineStates() == 0) {
            return online;
        } else if (param.getOnLineStates() == 1) {
            return offline;
        }
        return list;
    }

    /**
     * 查询绑定状态
     *
     * @param list
     * @param param
     * @return
     */
    public List<Pole> isBind(List<Pole> list, PoleStatesParam param) {
        List<Pole> bindList = new ArrayList<>();//已绑定客户
        List<Pole> unbindList = new ArrayList<>();//未绑定客户
        list.forEach(bindwarpper -> {
//                Long clientId = bindwarpper.getClientId();
            if (bindwarpper.getClientId() == null) {//判断是否绑定
                unbindList.add(bindwarpper);
            } else {
                bindList.add(bindwarpper);
            }
        });
        if (param.getBingStates() == 0) {
            return bindList;
        } else if (param.getBingStates() == 1) {
            return unbindList;
        }
        return list;
    }

    public List<Pole> isTrue(List<Pole> list, PoleStatesParam param) {
        List<Pole> isTrue = new ArrayList<>();//实体灯杆
        List<Pole> isFalse = new ArrayList<>();//虚拟灯杆
        list.forEach(bindwarpper -> {
//                Long clientId = bindwarpper.getClientId();
            if (bindwarpper.getDeviceCode() == null) {
                isFalse.add(bindwarpper);
            } else {
                isTrue.add(bindwarpper);
            }
        });
        if (param.getIsTrue() == 0) {
            return isTrue;
        } else if (param.getIsTrue() == 1) {
            return isFalse;
        }
        return list;
    }


    /**
     * 设置在线状态
     *
     * @param list
     * @return
     */
    public List<Pole> setOnline(List<Pole> list) {

        List<String> MacCodes = new ArrayList<>();

        for (Pole post : list) {
            if (post.getDeviceCode() != null) {
                MacCodes.add(post.getDeviceCode());
            }
        }
//
        List<BatchGetDeviceStateResponse.DeviceStatus> deviceStatuses = MainBoardInvokeSyncService.getInstance().batchGetDeviceState(MacCodes);
        if (deviceStatuses != null) {
            for (Pole post : list) {
                for (BatchGetDeviceStateResponse.DeviceStatus deviceStatus : deviceStatuses) {
//                System.out.println("DeviceCode:---------------------"+post.getDeviceCode());
//                System.out.println("DeviceName:---------------------"+deviceStatus.getDeviceName());
                    if (post.getDeviceCode() != null && post.getDeviceCode().equals(deviceStatus.getDeviceName())) {
                        post.setOnLineState(deviceStatus.getStatus());
//                        int size = poleBindingService.list(Wrappers.lambdaQuery(PoleBinding.class).eq(PoleBinding::getPoleId, post.getId())).size();
////                    System.out.println("size:----------------------"+size);
//                        post.setBindingCount(size);
                    }
                }
                //  MacCodes.add(post.getDeviceCode());
            }
        }

        return list;
    }


    /**
     * 查询灯杆的在线状态并赋值
     *
     * @return
     */
    public List<Pole> queryStatesAndList(Integer pageNo, Integer pageSize) {
    public List<Pole> queryStatesAndList(Integer pageNo, Integer pageSize, String keyword, Long groupid) {
        // List<LampPost> list = list(Wrappers.lambdaQuery(LampPost.class).eq(LampPost::getClientId, SecurityUtils.getUserId()));
        //List<Pole> list = list(Wrappers.lambdaQuery(Pole.class));
        PageHelper.startPage(pageNo, pageSize);
        List<Pole> list = new ArrayList<>();
        LambdaQueryWrapper<Pole> wrapper = new LambdaQueryWrapper<>();
        if (SecurityUtils.getClientId() == null) {
            list = list(Wrappers.lambdaQuery(Pole.class));
            wrapper = Wrappers.lambdaQuery(Pole.class);
        } else {
            list = list(Wrappers.lambdaQuery(Pole.class).eq(Pole::getClientId, SecurityUtils.getUserId()));
            wrapper = Wrappers.lambdaQuery(Pole.class).eq(Pole::getClientId, SecurityUtils.getUserId());
        }

        if (!keyword.isEmpty()) {
            wrapper.like(Pole::getPoleCode, keyword).or(
                    wrappers -> {
                        wrappers.like(Pole::getPoleName, keyword);
                    }
            );
        }
        list = list(wrapper);
        List<String> MacCodes = new ArrayList<>();

        for (Pole post : list) {
@@ -129,8 +399,13 @@
        List<BatchGetDeviceStateResponse.DeviceStatus> deviceStatuses = MainBoardInvokeSyncService.getInstance().batchGetDeviceState(MacCodes);
        for (Pole post : list) {
            for (BatchGetDeviceStateResponse.DeviceStatus deviceStatus : deviceStatuses) {
                if (post.getDeviceCode().equals(deviceStatus.getDeviceName())) {
//                System.out.println("DeviceCode:---------------------"+post.getDeviceCode());
//                System.out.println("DeviceName:---------------------"+deviceStatus.getDeviceName());
                if (post.getDeviceCode() != null && post.getDeviceCode().equals(deviceStatus.getDeviceName())) {
                    post.setOnLineState(deviceStatus.getStatus());
                    int size = poleBindingService.list(Wrappers.lambdaQuery(PoleBinding.class).eq(PoleBinding::getPoleId, post.getId())).size();
//                    System.out.println("size:----------------------"+size);
                    post.setBindingCount(size);
                }
            }
            //  MacCodes.add(post.getDeviceCode());
@@ -161,8 +436,8 @@
     *
     * @return 是否成功
     */
    public boolean unBindPole(String deviceCode) {
        return poleBindingService.unBindPole(deviceCode);
    public boolean unBindPole(Long poleId, String deviceCode) {
        return poleBindingService.unBindPole(poleId, deviceCode);
    }