危化品全生命周期管理后端
kongzy
2024-09-14 ed36af4d4cc5feac72a384d85f9032fc6dc1223a
hazmat-system/src/main/java/com/gkhy/hazmat/system/service/impl/HzHazmatFlowServiceImpl.java
@@ -53,6 +53,7 @@
        //设置分表id
        IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        HzHazmatFlow hazmatFlow = baseMapper.selectById(hazmatFlowId);
        IdTableNameHandler.removeCurrentId();
        if (!hazmatFlow.getCompanyId().equals(currentUser.getCompanyId())) {
            throw new ApiException("无权限查看其它企业数据");
        }
@@ -68,6 +69,7 @@
        //设置分表id
        IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        int row = baseMapper.insert(hazmatFlow);
        IdTableNameHandler.removeCurrentId();
        if (row < 1) {
            throw new ApiException("新增危化品流向失败");
        }
@@ -81,6 +83,7 @@
        //设置分表id
        IdTableNameHandler.setCurrentId(currentUser.getCompanyId());
        HzHazmatFlow hazmatFlow=baseMapper.selectById(hazmatFlowId);
        IdTableNameHandler.removeCurrentId();
        if(hazmatFlow==null){
            throw new ApiException("流向信息不存在");
        }