From c31bf5a8902485b750b5814a0d1c54ecc65d2c96 Mon Sep 17 00:00:00 2001
From: zjj <3272660260@qq.com>
Date: 星期六, 31 八月 2024 10:12:27 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/asrs/controller/LocOwnerController.java |   59 ++++++++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 52 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/zy/asrs/controller/LocOwnerController.java b/src/main/java/com/zy/asrs/controller/LocOwnerController.java
index 7b60ee8..b977446 100644
--- a/src/main/java/com/zy/asrs/controller/LocOwnerController.java
+++ b/src/main/java/com/zy/asrs/controller/LocOwnerController.java
@@ -1,22 +1,34 @@
 package com.zy.asrs.controller;
 
-import com.alibaba.fastjson.JSONArray;
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.baomidou.mybatisplus.mapper.Wrapper;
 import com.baomidou.mybatisplus.plugins.Page;
-import com.core.common.DateUtils;
-import com.zy.asrs.entity.LocOwner;
-import com.zy.asrs.service.LocOwnerService;
 import com.core.annotations.ManagerAuth;
 import com.core.common.BaseRes;
 import com.core.common.Cools;
+import com.core.common.DateUtils;
 import com.core.common.R;
+import com.core.exception.CoolException;
+import com.zy.asrs.entity.LocOwner;
+import com.zy.asrs.service.LocOwnerService;
+import com.zy.asrs.utils.LocOwnerExcelListener;
+import com.zy.common.entity.LocOwnerExcel;
 import com.zy.common.web.BaseController;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
-import java.util.*;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.net.URLEncoder;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 @RestController
 public class LocOwnerController extends BaseController {
@@ -112,13 +124,16 @@
     @ManagerAuth
     public R query(String condition) {
         EntityWrapper<LocOwner> wrapper = new EntityWrapper<>();
-        wrapper.like("id", condition);
+        wrapper.like("owner", condition);
         Page<LocOwner> page = locOwnerService.selectPage(new Page<>(0, 10), wrapper);
         List<Map<String, Object>> result = new ArrayList<>();
         for (LocOwner locOwner : page.getRecords()){
             Map<String, Object> map = new HashMap<>();
             map.put("id", locOwner.getId());
-            map.put("value", locOwner.getId());
+            map.put("value", locOwner.getOwner());
+            map.put("phone", locOwner.getPhone());
+            map.put("addr", locOwner.getAddr());
+            map.put("type", "locOwner");
             result.add(map);
         }
         return R.ok(result);
@@ -134,4 +149,34 @@
         return R.ok();
     }
 
+    /**
+     * excel瀵煎叆妯℃澘涓嬭浇
+     */
+    @RequestMapping(value = "/locOwner/excel/import/mould")
+    public void plaExcelImportMould(HttpServletResponse response) throws IOException {
+        List<LocOwnerExcel> excels = new ArrayList<>();
+        response.setContentType("application/vnd.ms-excel");
+        response.setCharacterEncoding("utf-8");
+        String fileName = URLEncoder.encode("璐т富妗f瀵煎叆妯℃澘", "UTF-8");
+        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
+        EasyExcel.write(response.getOutputStream(), LocOwnerExcel.class)
+                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
+                .sheet("sheet1")
+                .doWrite(excels);
+    }
+
+    // excel瀵煎叆
+    @PostMapping(value = "/locOwner/excel/import/auth")
+    @ManagerAuth(memo = "璐т富妗f鏁版嵁瀵煎叆")
+    @Transactional
+    public R matExcelImport(MultipartFile file) throws IOException {
+        LocOwnerExcelListener listener = new LocOwnerExcelListener(getUserId());
+        try {
+            EasyExcel.read(file.getInputStream(), LocOwnerExcel.class, listener).sheet().doRead();
+        }catch (Exception e){
+            throw new CoolException("瀵煎叆鐨勬暟鎹牸寮忛敊璇紝鍘熷洜锛�" + e.getMessage());
+        }
+        return R.ok("鎴愬姛鍚屾"+listener.getTotal()+"鏉la鐗╂枡鏁版嵁");
+    }
+
 }

--
Gitblit v1.9.1