From b3cef9d27013afee054bdd467defd0b6be218526 Mon Sep 17 00:00:00 2001
From: mitao <2763622819@qq.com>
Date: 星期五, 13 九月 2024 18:28:02 +0800
Subject: [PATCH] 修改bug

---
 ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/controller/management/MgtSensitiveWordsController.java |   50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 50 insertions(+), 0 deletions(-)

diff --git a/ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/controller/management/MgtSensitiveWordsController.java b/ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/controller/management/MgtSensitiveWordsController.java
index dfa802e..aee1db6 100644
--- a/ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/controller/management/MgtSensitiveWordsController.java
+++ b/ruoyi-modules/ruoyi-article/src/main/java/com/ruoyi/article/controller/management/MgtSensitiveWordsController.java
@@ -1,23 +1,31 @@
 package com.ruoyi.article.controller.management;
 
 
+import com.alibaba.excel.EasyExcel;
 import com.ruoyi.article.controller.management.dto.MgtSensitiveWordsDTO;
 import com.ruoyi.article.controller.management.dto.MgtSensitiveWordsQuery;
 import com.ruoyi.article.controller.management.vo.MgtSensitiveWordsVO;
 import com.ruoyi.article.service.ISensitiveWordsService;
 import com.ruoyi.common.core.domain.R;
+import com.ruoyi.common.core.utils.page.CollUtils;
 import com.ruoyi.common.core.utils.page.PageDTO;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import io.swagger.annotations.ApiParam;
+import java.io.IOException;
+import java.net.URLEncoder;
+import javax.servlet.http.HttpServletResponse;
 import lombok.RequiredArgsConstructor;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestPart;
 import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
 
 /**
  * <p>
@@ -72,4 +80,46 @@
         sensitiveWordsService.removeById(id);
         return R.ok();
     }
+
+    /**
+     * 下载导入模板
+     *
+     * @param response
+     */
+    @ApiOperation("下载导入模板")
+    @GetMapping("/exportTemplate")
+    public void exportSensitiveWordsTemplate(HttpServletResponse response) {
+        // 这里注意 有同学反应使用swagger 会导致各种问题,请直接用浏览器或者用postman
+        response.setContentType(
+                "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
+        response.setCharacterEncoding("utf-8");
+        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
+        try {
+            String fileName = URLEncoder.encode("敏感词导入模板", "UTF-8").replaceAll("\\+", "%20");
+            response.setHeader("Content-disposition",
+                    "attachment;filename*=utf-8''" + fileName + ".xlsx");
+            EasyExcel.write(response.getOutputStream(), MgtSensitiveWordsDTO.class)
+                    .sheet("敏感词导入模板")
+                    .doWrite(CollUtils.emptyList());
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    /**
+     * 敏感词导入
+     *
+     * @param file 导入文件
+     * @return
+     */
+    @ApiOperation("敏感词导入")
+    @PostMapping("/importSensitiveWords")
+    public R<?> importSensitiveWords(@RequestPart("file") MultipartFile file) {
+        try {
+            sensitiveWordsService.importSensitiveWords(file);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+        return R.ok();
+    }
 }

--
Gitblit v1.7.1