From 2039385fa4b4ac688907134ee7c7025fb2aea1fb Mon Sep 17 00:00:00 2001
From: whywhyo <1511349576@qq.com>
Date: 星期日, 16 七月 2023 12:59:17 +0800
Subject: [PATCH] 666

---
 zhang-content/src/main/java/com/ruoyi/service/impl/ZfEquipmentServiceImpl.java |  523 +++++++++++++++++++++++++++++++++++++---------------------
 1 files changed, 333 insertions(+), 190 deletions(-)

diff --git a/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEquipmentServiceImpl.java b/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEquipmentServiceImpl.java
index 8b1a5bf..2d99122 100644
--- a/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEquipmentServiceImpl.java
+++ b/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEquipmentServiceImpl.java
@@ -13,6 +13,7 @@
 import com.ruoyi.domain.*;
 import com.ruoyi.domain.ZfEquipment;
 import com.ruoyi.mapper.ZfEquipmentMapper;
+import com.ruoyi.service.ZAuthorityService;
 import com.ruoyi.service.ZInfoUserService;
 import com.ruoyi.service.ZfEquipmentService;
 import com.ruoyi.util.ArraysUtil;
@@ -28,8 +29,7 @@
 import java.util.List;
 import java.util.stream.Collectors;
 
-import static com.ruoyi.constant.MenuAuthority.EQUIPMENT_LIST;
-import static com.ruoyi.constant.MenuAuthority.EVENT_LIST;
+import static com.ruoyi.constant.MenuAuthority.*;
 
 /**
  * <p>
@@ -48,14 +48,19 @@
     @Resource
     ZfEquipmentService zfEquipmentService;
 
-    private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment) {
+    @Resource
+    ZAuthorityService zAuthorityService;
+
+    private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment,List<Long> familyIdList) {
         LambdaQueryWrapper<ZfEquipment> lqw = new LambdaQueryWrapper<>();
+        lqw.orderByDesc(ZfEquipment::getCreateTime);
         lqw.like(StringUtils.isNotEmpty(zfEquipment.getName()), ZfEquipment::getName, zfEquipment.getName());
         lqw.like(StringUtils.isNotEmpty(zfEquipment.getBuyer()), ZfEquipment::getBuyer, zfEquipment.getBuyer());
         lqw.like(StringUtils.isNotEmpty(zfEquipment.getContent()), ZfEquipment::getContent, zfEquipment.getContent());
         lqw.like(StringUtils.isNotEmpty(zfEquipment.getLocation()), ZfEquipment::getLocation, zfEquipment.getLocation());
         lqw.like(StringUtils.isNotEmpty(zfEquipment.getRemark()), ZfEquipment::getRemark, zfEquipment.getRemark());
         lqw.eq(zfEquipment.getHappenTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenTime());
+        lqw.in(ZfEquipment::getFamilyId,familyIdList);
         lqw.between(zfEquipment.getHappenStartTime() != null && zfEquipment.getHappenEndTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenStartTime(), zfEquipment.getHappenEndTime());
         return lqw;
     }
@@ -73,136 +78,150 @@
 
 
     
-    //------------------------------------------------------------------------------------------
-    private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment, String familyIds, String secondFamilyAuthority) {
-        //妯$硦鏌ヨ鐨勬潯浠�
-        LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment);
-
-        //瀹氫箟涓�涓竷灏旀爣璁帮紝鏍囪绗竴瀹跺涵鍙锋槸鍚︽湁閫楀彿
-        boolean flag = false;
-
-        StringBuilder secondFamilyIdsBuffer = new StringBuilder();
-
-        //澶勭悊浼犲叆鐨剆econdFamilyAuthority   3{2007 2018 2015},4{2007 2019}
-
-        if (StringUtils.isNotEmpty(secondFamilyAuthority)) {
-
-            if (secondFamilyAuthority.contains(",")) {
-                //閫楀彿闅斿紑寰楀埌濡備笅涓诧細3{2007 2018 2015}
-                String[] authorityList = secondFamilyAuthority.split(",");
-                for (String authorityAndId : authorityList) {
-                    setSecondFamilyAuthority(secondFamilyIdsBuffer, authorityAndId);
-                }
-
-                //鎶婂悗闈㈢殑閫楀彿鍘绘帀
-                if (secondFamilyIdsBuffer.length() > 1) {
-                    secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1);
-                }
-
-            } else {
-                setSecondFamilyAuthority(secondFamilyIdsBuffer, secondFamilyAuthority);
-                //鎶婂悗闈㈢殑閫楀彿鍘绘帀
-                if (secondFamilyIdsBuffer.length() > 1) {
-                    secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1);
-                }
-            }
-        }
-        //灏哠tringBuffer杞垚String
-        String secondFamilyIds = secondFamilyIdsBuffer.toString();
-
-        //绗竴瀹跺涵鍙风殑鏉′欢
-        if (familyIds.contains(",")) {
-            flag = true;
-        }
-
-        //绗簩瀹跺涵鍙风殑鏉′欢
-        if (StringUtils.isNotEmpty(secondFamilyIds)) {
-            if (secondFamilyIds.contains(",")) {
-                String[] secondFamilyIdList = secondFamilyIds.split(",");
-                if (flag) {
-                    //濡傛灉绗竴瀹跺涵鍙峰拰绗簩瀹跺涵鍙烽兘鏈夐�楀彿
-                    String[] familyList = familyIds.split(",");
-                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIdList);
-                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
-                } else {
-                    //濡傛灉绗竴瀹跺涵鍙锋病鏈夐�楀彿浣嗘槸绗簩瀹跺涵鍙锋湁閫楀彿
-                    String[] totalFamilyIdList = ArraysUtil.appendToArray(secondFamilyIdList, familyIds);
-                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
-                }
-
-            } else {
-                if (flag) {
-                    //濡傛灉绗竴瀹跺涵鍙锋湁閫楀彿浣嗙浜屽搴彿娌℃湁閫楀彿
-                    String[] familyList = familyIds.split(",");
-                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIds);
-                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
-                } else {
-                    //濡傛灉绗竴瀹跺涵鍙峰拰绗簩瀹跺涵鍙烽兘娌℃湁閫楀彿
-                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyIds, secondFamilyIds);
-                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
-                }
-            }
-        } else {
-            //濡傛灉娌℃湁绗簩瀹跺涵鍙�
-            if (flag) {
-                String[] familyList = familyIds.split(",");
-                lqw.in(ZfEquipment::getFamilyId, familyList);
-            } else {
-                lqw.eq(ZfEquipment::getFamilyId, familyIds);
-            }
-
-        }
-
-        lqw.orderByDesc(ZfEquipment::getCreateTime);
-        return lqw;
-    }
-
-    private void setSecondFamilyAuthority(StringBuilder secondFamilyIdsBuffer, String authorityAndId) {
-        //寰楀埌鏉冮檺涓�
-        String authorities = authorityAndId.substring(2, authorityAndId.length() - 1);
-
-        if (authorities.contains(" ")) {
-            //绌烘牸闅斿紑寰楀埌鑿滃崟id锛岀湅鏄惁鏈夊搴旂殑鏉冮檺
-            String[] authorityIdList = authorities.split(" ");
-            for (String s : authorityIdList) {
-                if (s.equals(EQUIPMENT_LIST)) {
-                    secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(",");
-                    break;
-                }
-            }
-        } else {
-            if (authorities.equals(EQUIPMENT_LIST)) {
-                secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(",");
-            }
-        }
-    }
-
-    private ZInfoUser getMySelf() {
-        SysUser user = SecurityUtils.getLoginUser().getUser();
-        Long userId = user.getUserId();
-        LambdaQueryWrapper<ZInfoUser> zInfoUserLambdaQueryWrapper = new LambdaQueryWrapper<>();
-        zInfoUserLambdaQueryWrapper.eq(ZInfoUser::getUserId, userId);
-        return zInfoUserService.getOne(zInfoUserLambdaQueryWrapper);
-    }
-
-    private String listFamilyIds() {
-        ZInfoUser mySelf = getMySelf();
-        return mySelf.getFamilyId();
-    }
-
-    private String listSecondFamilyIds() {
-        ZInfoUser mySelf = getMySelf();
-        return mySelf.getSecondFamilyId();
-    }
+//    //------------------------------------------------------------------------------------------
+//    private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment, String familyIds, String secondFamilyAuthority) {
+//        //妯$硦鏌ヨ鐨勬潯浠�
+//        LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment);
+//
+//        //瀹氫箟涓�涓竷灏旀爣璁帮紝鏍囪绗竴瀹跺涵鍙锋槸鍚︽湁閫楀彿
+//        boolean flag = false;
+//
+//        StringBuilder secondFamilyIdsBuffer = new StringBuilder();
+//
+//        //澶勭悊浼犲叆鐨剆econdFamilyAuthority   3{2007 2018 2015},4{2007 2019}
+//
+//        if (StringUtils.isNotEmpty(secondFamilyAuthority)) {
+//
+//            if (secondFamilyAuthority.contains(",")) {
+//                //閫楀彿闅斿紑寰楀埌濡備笅涓诧細3{2007 2018 2015}
+//                String[] authorityList = secondFamilyAuthority.split(",");
+//                for (String authorityAndId : authorityList) {
+//                    setSecondFamilyAuthority(secondFamilyIdsBuffer, authorityAndId);
+//                }
+//
+//                //鎶婂悗闈㈢殑閫楀彿鍘绘帀
+//                if (secondFamilyIdsBuffer.length() > 1) {
+//                    secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1);
+//                }
+//
+//            } else {
+//                setSecondFamilyAuthority(secondFamilyIdsBuffer, secondFamilyAuthority);
+//                //鎶婂悗闈㈢殑閫楀彿鍘绘帀
+//                if (secondFamilyIdsBuffer.length() > 1) {
+//                    secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1);
+//                }
+//            }
+//        }
+//        //灏哠tringBuffer杞垚String
+//        String secondFamilyIds = secondFamilyIdsBuffer.toString();
+//
+//        //绗竴瀹跺涵鍙风殑鏉′欢
+//        if (familyIds.contains(",")) {
+//            flag = true;
+//        }
+//
+//        //绗簩瀹跺涵鍙风殑鏉′欢
+//        if (StringUtils.isNotEmpty(secondFamilyIds)) {
+//            if (secondFamilyIds.contains(",")) {
+//                String[] secondFamilyIdList = secondFamilyIds.split(",");
+//                if (flag) {
+//                    //濡傛灉绗竴瀹跺涵鍙峰拰绗簩瀹跺涵鍙烽兘鏈夐�楀彿
+//                    String[] familyList = familyIds.split(",");
+//                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIdList);
+//                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
+//                } else {
+//                    //濡傛灉绗竴瀹跺涵鍙锋病鏈夐�楀彿浣嗘槸绗簩瀹跺涵鍙锋湁閫楀彿
+//                    String[] totalFamilyIdList = ArraysUtil.appendToArray(secondFamilyIdList, familyIds);
+//                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
+//                }
+//
+//            } else {
+//                if (flag) {
+//                    //濡傛灉绗竴瀹跺涵鍙锋湁閫楀彿浣嗙浜屽搴彿娌℃湁閫楀彿
+//                    String[] familyList = familyIds.split(",");
+//                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIds);
+//                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
+//                } else {
+//                    //濡傛灉绗竴瀹跺涵鍙峰拰绗簩瀹跺涵鍙烽兘娌℃湁閫楀彿
+//                    String[] totalFamilyIdList = ArraysUtil.appendToArray(familyIds, secondFamilyIds);
+//                    lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList);
+//                }
+//            }
+//        } else {
+//            //濡傛灉娌℃湁绗簩瀹跺涵鍙�
+//            if (flag) {
+//                String[] familyList = familyIds.split(",");
+//                lqw.in(ZfEquipment::getFamilyId, familyList);
+//            } else {
+//                lqw.eq(ZfEquipment::getFamilyId, familyIds);
+//            }
+//
+//        }
+//
+//        lqw.orderByDesc(ZfEquipment::getCreateTime);
+//        return lqw;
+//    }
+//
+//    private void setSecondFamilyAuthority(StringBuilder secondFamilyIdsBuffer, String authorityAndId) {
+//        //寰楀埌鏉冮檺涓�
+//        String authorities = authorityAndId.substring(2, authorityAndId.length() - 1);
+//
+//        if (authorities.contains(" ")) {
+//            //绌烘牸闅斿紑寰楀埌鑿滃崟id锛岀湅鏄惁鏈夊搴旂殑鏉冮檺
+//            String[] authorityIdList = authorities.split(" ");
+//            for (String s : authorityIdList) {
+//                if (s.equals(EQUIPMENT_LIST)) {
+//                    secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(",");
+//                    break;
+//                }
+//            }
+//        } else {
+//            if (authorities.equals(EQUIPMENT_LIST)) {
+//                secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(",");
+//            }
+//        }
+//    }
+//
+//    private ZInfoUser getMySelf() {
+//        SysUser user = SecurityUtils.getLoginUser().getUser();
+//        Long userId = user.getUserId();
+//        LambdaQueryWrapper<ZInfoUser> zInfoUserLambdaQueryWrapper = new LambdaQueryWrapper<>();
+//        zInfoUserLambdaQueryWrapper.eq(ZInfoUser::getUserId, userId);
+//        return zInfoUserService.getOne(zInfoUserLambdaQueryWrapper);
+//    }
+//
+//    private String listFamilyIds() {
+//        ZInfoUser mySelf = getMySelf();
+//        return mySelf.getFamilyId();
+//    }
+//
+//    private String listSecondFamilyIds() {
+//        ZInfoUser mySelf = getMySelf();
+//        return mySelf.getSecondFamilyId();
+//    }
 
     @Override
     public List<ZfEquipment> selectByCondition(ZfEquipment zfEquipment) {
-        String familyIds = listFamilyIds();
-        String secondFamilyIds = listSecondFamilyIds();
-        LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, familyIds, secondFamilyIds);
+//        String familyIds = listFamilyIds();
+//        String secondFamilyIds = listSecondFamilyIds();
+//        LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, familyIds, secondFamilyIds);
+//        List<ZfEquipment> beanRecords = list(lambdaQueryWrapper);
+//        log.info("浠庢暟鎹簱涓煡鍒扮殑涓�:{}", beanRecords);
+//        return markOwnData(familyIds, beanRecords);
+
+        //瑕佹煡鑷繁瀹跺涵鐨�
+        ZInfoUser myself = zInfoUserService.getMyself();
+        Long familyId = myself.getFamilyId();
+        //涔熻鏌ュ埆浜烘巿鏉冪殑
+        List<ZAuthority> authority = zAuthorityService.getAuthority();
+        List<Long> idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST)).map(ZAuthority::getFid).collect(Collectors.toList());
+        //鍔犱笂鑷繁瀹跺涵鐨刬d
+        idList.add(familyId);
+
+        LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, idList);
         List<ZfEquipment> beanRecords = list(lambdaQueryWrapper);
         log.info("浠庢暟鎹簱涓煡鍒扮殑涓�:{}", beanRecords);
-        return markOwnData(familyIds, beanRecords);
+        return markOwnData(familyId, beanRecords);
     }
 
     @Override
@@ -214,94 +233,103 @@
             list = list();
         return list;    }
 
-    private static List<ZfEquipment> markOwnData(String familyIds, List<ZfEquipment> beanRecords) {
-        return beanRecords.stream().peek(record -> {
-            String recordFamilyId = record.getFamilyId();
-            if (familyIds.contains(",")) {
-                String[] familyIdList = familyIds.split(",");
-                for (String familyId : familyIdList) {
-                    if (familyId.equals(recordFamilyId)) {
-                        record.setOwnData(1);
+//    private static List<ZfEquipment> markOwnData(String familyIds, List<ZfEquipment> beanRecords) {
+//        return beanRecords.stream().peek(record -> {
+//            String recordFamilyId = record.getFamilyId();
+//            if (familyIds.contains(",")) {
+//                String[] familyIdList = familyIds.split(",");
+//                for (String familyId : familyIdList) {
+//                    if (familyId.equals(recordFamilyId)) {
+//                        record.setOwnData(1);
+//                    }
+//                }
+//            } else {
+//                if (familyIds.equals(recordFamilyId)) {
+//                    record.setOwnData(1);
+//                }
+//            }
+//        }).collect(Collectors.toList());
+//    }
+
+    public List<ZfEquipment> markOwnData(Long familyId,List<ZfEquipment> beanRecords){
+        return beanRecords.stream().peek(
+                bean -> {
+                    if (bean.getFamilyId() == familyId) {
+                        bean.setOwnData(1);
+                    } else {
+                        bean.setOwnData(0);
                     }
                 }
-            } else {
-                if (familyIds.equals(recordFamilyId)) {
-                    record.setOwnData(1);
-                }
-            }
-        }).collect(Collectors.toList());
+        ).collect(Collectors.toList());
     }
 
     @Override
     public AjaxResult selectDataList(ZfEquipment zfEquipment, Integer pageNum, Integer pageSize) {
-        String familyIds = listFamilyIds();
-        String secondFamilyAuthority = listSecondFamilyIds();
-        LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, familyIds, secondFamilyAuthority);
+//        String familyIds = listFamilyIds();
+//        String secondFamilyAuthority = listSecondFamilyIds();
+//        LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, familyIds, secondFamilyAuthority);
+//
+//        Page<ZfEquipment> zfEquipmentPage = new Page<>(pageNum, pageSize);
+//        Page<ZfEquipment> pageResult = page(zfEquipmentPage, lqw);
+//
+//        List<ZfEquipment> beanRecords = pageResult.getRecords();//寰楀埌鏌ヨ鍑烘潵鐨勬暟鎹�
+//
+//        List<ZfEquipment> dtoResult = markOwnData(familyIds, beanRecords);
+//
+//        HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult);
+//        return AjaxResult.success(data);
+
+        //瑕佹煡鑷繁瀹跺涵鐨�
+        ZInfoUser myself = zInfoUserService.getMyself();
+        Long familyId = myself.getFamilyId();
+        //涔熻鏌ュ埆浜烘巿鏉冪殑
+        List<ZAuthority> authority = zAuthorityService.getAuthority();
+        List<Long> idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST)).map(ZAuthority::getFid).collect(Collectors.toList());
+        //鍔犱笂鑷繁瀹跺涵鐨刬d
+        idList.add(familyId);
+//        String familyIds = listFamilyIds();
+//        String secondFamilyAuthority = listSecondFamilyIds();
+        LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, idList);
+
 
         Page<ZfEquipment> zfEquipmentPage = new Page<>(pageNum, pageSize);
         Page<ZfEquipment> pageResult = page(zfEquipmentPage, lqw);
 
         List<ZfEquipment> beanRecords = pageResult.getRecords();//寰楀埌鏌ヨ鍑烘潵鐨勬暟鎹�
 
-        List<ZfEquipment> dtoResult = markOwnData(familyIds, beanRecords);
+        List<ZfEquipment> dtoResult = markOwnData(familyId, beanRecords);
 
         HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult);
         return AjaxResult.success(data);
     }
 
     @Override
-    public AjaxResult addData(ZfEquipment zfEquipment) {
+    public AjaxResult addData(ZfEquipment zfEquipment){
+        ZInfoUser myself = zInfoUserService.getMyself();
+        Long familyId = myself.getFamilyId();
 
-        String familyIds = listFamilyIds();
-        boolean flag = false;//鍒ゆ柇褰撳墠鐢ㄦ埛鐨刬d鏄惁鏈夋潈鍔犲叆褰撳墠瀹跺涵id鐨勫璞�
-        if (familyIds.contains(",")) {
-            String[] familyList = familyIds.split(",");
-            for (String familyId : familyList) {
-                if (familyId.equals(zfEquipment.getFamilyId())) {
-                    flag = true;
-                }
-            }
-        } else {
-            if (zfEquipment.getFamilyId().equals(familyIds)) {
-                flag = true;
-            }
-        }
-        if (flag) {
-            if (save(zfEquipment)) {
-                return AjaxResult.success();
-            } else {
-                return AjaxResult.error();
-            }
-
-        } else {
-            throw new RuntimeException("浣犳病鏈夋搷浣滆瀹跺涵鍙蜂负" + zfEquipment.getFamilyId() + "鏁版嵁鐨勬潈闄�");
-        }
-    }
-
-    @Override
-    public AjaxResult addData2(ZfEquipment zfEquipment) {
-        ZInfoUser mySelf = getMySelf();
-        String myFamilyId = mySelf.getFamilyId();
-
-        if (StringUtils.isNotEmpty(zfEquipment.getFamilyId())) {
-            throw new RuntimeException("瀵煎叆鏁版嵁鏃舵偍涓嶈兘鎸囧畾瀹跺涵鍙�");
-        }
-
-        if (StringUtils.isEmpty(myFamilyId)) {
+        if(familyId == null){
             throw new RuntimeException("鎮ㄨ繕鏈姞鍏ヤ换浣曞搴�");
         }
 
-        if (myFamilyId.contains(",")) {
-            String[] myFamilyIds = myFamilyId.split(",");
-            //榛樿娣诲姞绗竴涓搴殑鏁版嵁
-            zfEquipment.setFamilyId(myFamilyIds[0]);
-        } else {
-            zfEquipment.setFamilyId(myFamilyId);
+        List<ZAuthority> authority = zAuthorityService.getAuthority();
+        List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_ADD)).map(ZAuthority::getFid).collect(Collectors.toList());
+        familyIdList.add(familyId);
+
+        if (zfEquipment.getFamilyId()!=null && !familyIdList.contains(zfEquipment.getFamilyId())) {
+            throw new RuntimeException("浣犳病鏈夋潈闄愭搷浣滄瀹跺涵鐨勬暟鎹�");
+        }
+
+
+        if(zfEquipment.getFamilyId() == null){
+            //榛樿娣诲姞鑷繁瀹跺涵鐨勬暟鎹�
+            zfEquipment.setFamilyId(familyId);
         }
 
         //鍒ゆ柇鏄惁鏈夐噸澶嶆暟鎹�
         LambdaQueryWrapper<ZfEquipment> lqw = uniqueCondition(zfEquipment);
         List<ZfEquipment> list = list(lqw);
+
         if(list.size()>0){
             throw new RuntimeException("璇峰嬁鏂板閲嶅鏁版嵁");
         }
@@ -311,7 +339,72 @@
         } else {
             return AjaxResult.error();
         }
+
     }
+
+//    @Override
+//    public AjaxResult addData(ZfEquipment zfEquipment) {
+//
+//        String familyIds = listFamilyIds();
+//        boolean flag = false;//鍒ゆ柇褰撳墠鐢ㄦ埛鐨刬d鏄惁鏈夋潈鍔犲叆褰撳墠瀹跺涵id鐨勫璞�
+//        if (familyIds.contains(",")) {
+//            String[] familyList = familyIds.split(",");
+//            for (String familyId : familyList) {
+//                if (familyId.equals(zfEquipment.getFamilyId())) {
+//                    flag = true;
+//                }
+//            }
+//        } else {
+//            if (zfEquipment.getFamilyId().equals(familyIds)) {
+//                flag = true;
+//            }
+//        }
+//        if (flag) {
+//            if (save(zfEquipment)) {
+//                return AjaxResult.success();
+//            } else {
+//                return AjaxResult.error();
+//            }
+//
+//        } else {
+//            throw new RuntimeException("浣犳病鏈夋搷浣滆瀹跺涵鍙蜂负" + zfEquipment.getFamilyId() + "鏁版嵁鐨勬潈闄�");
+//        }
+//    }
+
+//    @Override
+//    public AjaxResult addData2(ZfEquipment zfEquipment) {
+//        ZInfoUser mySelf = getMySelf();
+//        String myFamilyId = mySelf.getFamilyId();
+//
+//        if (StringUtils.isNotEmpty(zfEquipment.getFamilyId())) {
+//            throw new RuntimeException("瀵煎叆鏁版嵁鏃舵偍涓嶈兘鎸囧畾瀹跺涵鍙�");
+//        }
+//
+//        if (StringUtils.isEmpty(myFamilyId)) {
+//            throw new RuntimeException("鎮ㄨ繕鏈姞鍏ヤ换浣曞搴�");
+//        }
+//
+//        if (myFamilyId.contains(",")) {
+//            String[] myFamilyIds = myFamilyId.split(",");
+//            //榛樿娣诲姞绗竴涓搴殑鏁版嵁
+//            zfEquipment.setFamilyId(myFamilyIds[0]);
+//        } else {
+//            zfEquipment.setFamilyId(myFamilyId);
+//        }
+//
+//        //鍒ゆ柇鏄惁鏈夐噸澶嶆暟鎹�
+//        LambdaQueryWrapper<ZfEquipment> lqw = uniqueCondition(zfEquipment);
+//        List<ZfEquipment> list = list(lqw);
+//        if(list.size()>0){
+//            throw new RuntimeException("璇峰嬁鏂板閲嶅鏁版嵁");
+//        }
+//
+//        if (save(zfEquipment)) {
+//            return AjaxResult.success();
+//        } else {
+//            return AjaxResult.error();
+//        }
+//    }
 
 
 
@@ -329,13 +422,63 @@
         log.info("璧勪骇鍒楄〃涓猴細{}", propertyList);
 
         for (ZfEquipment zfEquipment : propertyList) {
-            zfEquipmentService.addData2(zfEquipment);
+            zfEquipmentService.addData(zfEquipment);
         }
 
         return AjaxResult.success("瀵煎叆鏁版嵁鎴愬姛");
 
     }
 
+    @Override
+    public AjaxResult updateData(ZfEquipment zfEquipment) {
+
+        ZInfoUser myself = zInfoUserService.getMyself();
+        Long familyId = myself.getFamilyId();
+
+        //鍏堟牴鎹甶d鏌ヨ鍑烘暟鎹殑familyId锛岀湅鐪嬫湁娌℃湁鏉冮檺鎿嶄綔
+        Long dataFamilyId = getById(zfEquipment.getId()).getFamilyId();
+
+        List<ZAuthority> authority = zAuthorityService.getAuthority();
+        List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_UPDATE)).map(ZAuthority::getFid).collect(Collectors.toList());
+        familyIdList.add(familyId);
+
+        if (dataFamilyId!=null && !familyIdList.contains(dataFamilyId)) {
+            throw new RuntimeException("浣犳病鏈夋潈闄愭搷浣滄瀹跺涵鐨勬暟鎹�");
+        }
+
+        if(updateById(zfEquipment)){
+            return AjaxResult.success();
+        }else {
+            return AjaxResult.error();
+        }
+
+
+    }
+
+    @Override
+    public AjaxResult deleteData(Long[] ids) {
+        List<ZfEquipment> dataList = zfEquipmentService.listByIds(Arrays.asList(ids));
+
+        ZInfoUser myself = zInfoUserService.getMyself();
+        Long familyId = myself.getFamilyId();
+
+        List<ZAuthority> authority = zAuthorityService.getAuthority();
+        List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_REMOVE)).map(ZAuthority::getFid).collect(Collectors.toList());
+        familyIdList.add(familyId);
+
+        for (ZfEquipment data : dataList) {
+            if (!familyIdList.contains(data.getFamilyId())){
+                throw new RuntimeException("浣犳病鏈夋潈闄愭搷浣滄瀹跺涵鐨勬暟鎹�");
+            }
+        }
+
+        if (zfEquipmentService.removeByIds(Arrays.asList(ids))) {
+            return AjaxResult.success();
+        }else {
+            return AjaxResult.error();
+        }
+        
+    }
 
 
 }

--
Gitblit v1.9.1