From 1c741aeeb78327e9ce93e6c2248e3aa779cdb030 Mon Sep 17 00:00:00 2001
From: Jinquan_Ou <Jinquan@gdut.com>
Date: 星期四, 06 四月 2023 22:16:30 +0800
Subject: [PATCH] 888

---
 zhang-content/src/main/java/com/ruoyi/service/impl/ZfEventServiceImpl.java |   48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 48 insertions(+), 0 deletions(-)

diff --git a/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEventServiceImpl.java b/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEventServiceImpl.java
index d4f9477..151a865 100644
--- a/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEventServiceImpl.java
+++ b/zhang-content/src/main/java/com/ruoyi/service/impl/ZfEventServiceImpl.java
@@ -9,13 +9,16 @@
 import com.ruoyi.common.utils.MapUtils;
 import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.StringUtils;
+import com.ruoyi.common.utils.poi.ExcelUtil;
 import com.ruoyi.domain.ZInfoUser;
+import com.ruoyi.domain.ZfEvent;
 import com.ruoyi.domain.ZfEvent;
 import com.ruoyi.mapper.ZfEventMapper;
 import com.ruoyi.service.ZInfoUserService;
 import com.ruoyi.service.ZfEventService;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.Resource;
 import java.util.HashMap;
@@ -35,6 +38,9 @@
 
     @Resource
     ZInfoUserService zInfoUserService;
+
+    @Resource
+    ZfEventService zfEventService;
     
     private String listFamilyIds(){
         SysUser user = SecurityUtils.getLoginUser().getUser();
@@ -68,6 +74,48 @@
         return list;
     }
 
+    @Override
+    public int addEvent(ZfEvent zfEvent) {
+        String familyIds = listFamilyIds();
+        boolean flag = false;//鍒ゆ柇褰撳墠鐢ㄦ埛鐨刬d鏄惁鏈夋潈鍔犲叆褰撳墠瀹跺涵id鐨勫璞�
+        if (familyIds.contains(",")) {
+            String[] familyList = familyIds.split(",");
+            for (String familyId : familyList) {
+                if (familyId.equals(zfEvent.getFamilyId())) {
+                    flag = true;
+                }
+            }
+        } else {
+            if (zfEvent.getFamilyId().equals(familyIds)) {
+                flag = true;
+            }
+        }
+        if (flag) {
+            boolean save = save(zfEvent);
+            return save ? 1 : 0;
+        } else {
+            throw new RuntimeException("浣犳病鏈夋搷浣滆瀹跺涵鍙蜂负" + zfEvent.getFamilyId() + "鏁版嵁鐨勬潈闄�");
+        }
+    }
+
+    @Override
+    public AjaxResult importExcel(MultipartFile file) {
+        ExcelUtil<ZfEvent> util = new ExcelUtil<>(ZfEvent.class);
+        List<ZfEvent> eventList = null;
+        try {
+            eventList = util.importExcel(file.getInputStream());
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+        log.info("璧勪骇鍒楄〃涓猴細{}", eventList);
+
+        for (ZfEvent zfEvent : eventList) {
+            zfEventService.addEvent(zfEvent);
+        }
+
+        return AjaxResult.success("瀵煎叆鏁版嵁鎴愬姛");
+    }
+
     private LambdaQueryWrapper<ZfEvent> buildCondition(ZfEvent zfEvent) {
         LambdaQueryWrapper<ZfEvent> lqw = new LambdaQueryWrapper<>();
         lqw.like(!StringUtils.isEmpty(zfEvent.getAddress()),ZfEvent::getAddress,zfEvent.getAddress());

--
Gitblit v1.9.1