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