feat: 药店批量导入接口(开发自提需求~)

hnqz-upms/hnqz-upms-biz/src/main/java/com/qunzhixinxi/hnqz/admin/controller/WmDaPharmacyController.java

@@ -17,15 +17,23 @@
 
 package com.qunzhixinxi.hnqz.admin.controller;
 
+import cn.hutool.core.bean.BeanUtil;
+import cn.hutool.core.collection.CollectionUtil;
+import cn.hutool.json.JSONUtil;
 import com.baomidou.mybatisplus.core.metadata.OrderItem;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.pig4cloud.plugin.excel.annotation.RequestExcel;
 import com.qunzhixinxi.hnqz.admin.entity.*;
+import com.qunzhixinxi.hnqz.admin.entity.dto.CommonUserDTO;
+import com.qunzhixinxi.hnqz.admin.entity.model.excel.BatchPharmacyExcelModel;
+import com.qunzhixinxi.hnqz.admin.entity.model.excel.CommonUserExcelModel;
 import com.qunzhixinxi.hnqz.admin.service.WmTaskContentService;
 import com.qunzhixinxi.hnqz.admin.service.WmTaskService;
 import com.qunzhixinxi.hnqz.common.core.util.R;
 import com.qunzhixinxi.hnqz.common.log.annotation.SysLog;
 import com.qunzhixinxi.hnqz.admin.service.WmDaPharmacyService;
+import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.lang.StringUtils;
 import org.springframework.security.access.prepost.PreAuthorize;
@@ -34,8 +42,10 @@ import io.swagger.annotations.ApiOperation;
 import lombok.AllArgsConstructor;
 import org.springframework.web.bind.annotation.*;
 
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 
 /**
@@ -44,6 +54,7 @@ import java.util.Map;
  * @author pigx code generator
  * @date 2020-09-23 21:17:34
  */
+@Slf4j
 @RestController
 @AllArgsConstructor
 @RequestMapping("/wmdapharmacy")
@@ -252,4 +263,38 @@ public class WmDaPharmacyController {
 		return wmDaPharmacyService.updateByapproval(map);
 	}
 
+	/**
+	 * 药店批量导入
+	 *
+	 * @param excelModelList 药店列表
+	 * @return {@link R}<{@link ?}> 结果
+	 */
+	@SysLog("药店批量导入")
+	@PostMapping(value = "/batch")
+	public R<?> batchImport(@RequestExcel List<BatchPharmacyExcelModel> excelModelList) {
+		// 参数校验
+		if (CollectionUtils.isEmpty(excelModelList)) {
+			log.error("空报表，不能上传");
+			return R.failed("空报表，不能上传");
+		}
+
+		// 实际的处理业务
+		Map<String, String> errorMap = wmDaPharmacyService.batchByExcel(excelModelList);
+
+		// 返回上传结果
+		Map<String, Object> result = new HashMap<>(3);
+		if (CollectionUtil.isEmpty(errorMap)) {
+			result.put("code", "SUCCESS");
+			result.put("message", "上传成功");
+			result.put("data", null);
+			return R.ok(result);
+		} else {
+			log.error("上传数据错误，{}", JSONUtil.toJsonStr(errorMap));
+			result.put("code", "FAILURE");
+			result.put("message", "上传数据错误");
+			result.put("data", errorMap);
+			return R.failed(result);
+		}
+	}
+
 }

hnqz-upms/hnqz-upms-biz/src/main/java/com/qunzhixinxi/hnqz/admin/entity/model/excel/BatchPharmacyExcelModel.java

@@ -0,0 +1,35 @@
+package com.qunzhixinxi.hnqz.admin.entity.model.excel;
+
+import com.alibaba.excel.annotation.ExcelProperty;
+import lombok.Data;
+
+import java.io.Serializable;
+
+/**
+ * 药店excel批量导入model
+ *
+ * @author lixuesong
+ * @date 2023/11/01
+ */
+@Data
+public class BatchPharmacyExcelModel implements Serializable {
+	private static final long serialVersionUID = 463822341690953480L;
+
+	/**
+	 * 行编号
+	 */
+	@ExcelProperty(index = 0)
+	private String number;
+
+	/**
+	 * 名称
+	 */
+	@ExcelProperty(index = 1)
+	private String name;
+
+	/**
+	 * 地址
+	 */
+	@ExcelProperty(index = 2)
+	private String address;
+}

hnqz-upms/hnqz-upms-biz/src/main/java/com/qunzhixinxi/hnqz/admin/service/WmDaPharmacyService.java

@@ -19,6 +19,7 @@ package com.qunzhixinxi.hnqz.admin.service;
 
 import com.baomidou.mybatisplus.extension.service.IService;
 import com.qunzhixinxi.hnqz.admin.entity.WmDaPharmacy;
+import com.qunzhixinxi.hnqz.admin.entity.model.excel.BatchPharmacyExcelModel;
 import com.qunzhixinxi.hnqz.common.core.util.R;
 
 import java.util.List;
@@ -50,4 +51,12 @@ public interface WmDaPharmacyService extends IService<WmDaPharmacy> {
 	 * 批量转换经纬度
 	 */
 	void parseAddress();
+
+	/**
+	 * 批量导入
+	 *
+	 * @param excelModelList 药店列表
+	 * @return 结果
+	 */
+    Map<String, String> batchByExcel(List<BatchPharmacyExcelModel> excelModelList);
 }

hnqz-upms/hnqz-upms-biz/src/main/java/com/qunzhixinxi/hnqz/admin/service/impl/WmDaPharmacyServiceImpl.java

@@ -16,6 +16,7 @@
  */
 package com.qunzhixinxi.hnqz.admin.service.impl;
 
+import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.util.ArrayUtil;
 import cn.hutool.core.util.StrUtil;
@@ -23,11 +24,14 @@ import com.alibaba.csp.sentinel.util.StringUtil;
 import com.alibaba.fastjson.JSONObject;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.qunzhixinxi.hnqz.admin.api.dto.UserDTO;
 import com.qunzhixinxi.hnqz.admin.entity.WmDaPharmacy;
 import com.qunzhixinxi.hnqz.admin.entity.WmScorePackage;
 import com.qunzhixinxi.hnqz.admin.entity.WmScorePackageStatus;
 import com.qunzhixinxi.hnqz.admin.entity.WmTask;
 import com.qunzhixinxi.hnqz.admin.entity.WmTaskContent;
+import com.qunzhixinxi.hnqz.admin.entity.model.excel.BatchPharmacyExcelModel;
+import com.qunzhixinxi.hnqz.admin.enums.DelEnum;
 import com.qunzhixinxi.hnqz.admin.mapper.WmDaPharmacyMapper;
 import com.qunzhixinxi.hnqz.admin.mapper.WmTaskContentMapper;
 import com.qunzhixinxi.hnqz.admin.service.WmDaPharmacyService;
@@ -36,12 +40,16 @@ import com.qunzhixinxi.hnqz.admin.service.WmScorePackageStatusService;
 import com.qunzhixinxi.hnqz.admin.service.WmTaskService;
 import com.qunzhixinxi.hnqz.admin.util.MapUtil;
 import com.qunzhixinxi.hnqz.common.core.util.R;
+import com.qunzhixinxi.hnqz.common.security.service.HnqzUser;
 import com.qunzhixinxi.hnqz.common.security.util.SecurityUtils;
 import lombok.AllArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import java.time.LocalDateTime;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -52,6 +60,7 @@ import java.util.Map;
  * @author pigx code generator
  * @date 2020-09-23 21:17:34
  */
+@Slf4j
 @Service
 @AllArgsConstructor
 public class WmDaPharmacyServiceImpl extends ServiceImpl<WmDaPharmacyMapper, WmDaPharmacy> implements WmDaPharmacyService {
@@ -435,6 +444,74 @@ public class WmDaPharmacyServiceImpl extends ServiceImpl<WmDaPharmacyMapper, WmD
 		}
 	}
 
+	/**
+	 * 批量导入
+	 *
+	 * @param excelModelList 药店列表
+	 * @return 结果
+	 */
+	@Override
+	public Map<String, String> batchByExcel(List<BatchPharmacyExcelModel> excelModelList) {
+
+		// 变量
+		Map<String, String> errorMap = new HashMap<>();
+		List<WmDaPharmacy> currentList = new ArrayList<>();
+
+		// 获取操作员
+		HnqzUser creator = SecurityUtils.getUser();
+
+		log.info("开始校验数据：{}", excelModelList);
+		// 业务值校验
+		for (BatchPharmacyExcelModel excelModel : excelModelList) {
+			WmDaPharmacy pharmacy = new WmDaPharmacy();
+			// 校验名称
+			String name = StrUtil.cleanBlank(excelModel.getName());
+			if (StrUtil.isBlank(name)) {
+				log.error("姓名为空");
+				errorMap(errorMap, "名称为空", excelModel.getNumber());
+			} else {
+				pharmacy.setPharmacyName(name);
+			}
+
+			// 校验地址
+			String address = StrUtil.cleanBlank(excelModel.getName());
+			if (StrUtil.isBlank(address)) {
+				log.error("地址为空");
+				errorMap(errorMap, "地址为空", excelModel.getNumber());
+			} else {
+				pharmacy.setAddress(address);
+			}
+
+			// 重复校验
+			int count = this.count(Wrappers.<WmDaPharmacy>lambdaQuery()
+					.eq(WmDaPharmacy::getPharmacyName, name)
+					.eq(WmDaPharmacy::getDelFlag, DelEnum.NOT_DEL.val()));
+			if (count > 0) {
+				log.error("{} 已存在", name);
+				errorMap(errorMap, "名称已存在", excelModel.getNumber());
+			}
+
+			// 记录正确的结果
+			pharmacy.setProvince("");
+			pharmacy.setCity("");
+			pharmacy.setArea("");
+			//更新审核信息
+			LocalDateTime now = LocalDateTime.now();
+			pharmacy.setCreateTime(now);
+			pharmacy.setCreateUser(creator.getId());
+			// 根据地址转换经纬度
+			this.convertToLatAndLong(pharmacy);
+			currentList.add(pharmacy);
+		}
+
+		// 上传结果
+		if (CollUtil.isEmpty(errorMap)) {
+			currentList.forEach(this::save);
+		}
+
+		return errorMap;
+	}
+
 	/**
 	 * 根据地址转换经纬度
 	 *
@@ -456,5 +533,19 @@ public class WmDaPharmacyServiceImpl extends ServiceImpl<WmDaPharmacyMapper, WmD
 		}
 	}
 
+	/**
+	 * 封装错误信息
+	 *
+	 * @param errorMap 记录错误信息map
+	 * @param key      原因
+	 * @param idStr    错误id
+	 */
+	private void errorMap(Map<String, String> errorMap, String key, String idStr) {
+		if (errorMap.containsKey(key)) {
+			String val = errorMap.get(key);
+			idStr = String.join(";", val, idStr);
+		}
+		errorMap.put(key, idStr);
+	}
 
 }

hnqz-upms/hnqz-upms-biz/src/test/java/com/qunzhixinxi/hnqz/admin/ManualImportSurveyInfoTest.java

@@ -0,0 +1,103 @@
+package com.qunzhixinxi.hnqz.admin;
+
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.read.builder.ExcelReaderBuilder;
+import com.alibaba.fastjson.JSONObject;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.qunzhixinxi.hnqz.admin.entity.WmDaPharmacy;
+import com.qunzhixinxi.hnqz.admin.enums.DelEnum;
+import com.qunzhixinxi.hnqz.admin.service.WmDaPharmacyService;
+import com.qunzhixinxi.hnqz.admin.util.MapUtil;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+import java.time.LocalDateTime;
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.List;
+
+/**
+ * 手动导入调查信息测试
+ *
+ * @author lixuesong
+ * @date 2023/09/27
+ */
+@RunWith(SpringJUnit4ClassRunner.class)
+@SpringBootTest(classes = {HnqzAdminApplication.class})
+public class ManualImportSurveyInfoTest {
+
+    @Autowired
+    private WmDaPharmacyService wmDaPharmacyService;
+
+//    @Test
+    public void testImport() {
+        ExcelReaderBuilder readerBuilder = EasyExcel.read("D:\\li312\\Desktop\\零售终端门店资料1027.xlsx");
+        List<Object> objects = readerBuilder.doReadAllSync();
+        System.out.println(objects);
+
+		List<WmDaPharmacy> pharmacyList = new ArrayList<>();
+        for (int i = 1; i < objects.size(); i++) {
+
+//            try {
+//                Thread.sleep(200);
+//            } catch (InterruptedException e) {
+//                throw new RuntimeException(e);
+//            }
+
+            LinkedHashMap<Integer, String> map = (LinkedHashMap) objects.get(i);
+            String name = map.get(2);
+            String address = map.get(4);
+
+			int count = wmDaPharmacyService.count(Wrappers.<WmDaPharmacy>lambdaQuery()
+					.eq(WmDaPharmacy::getPharmacyName, name)
+					.eq(WmDaPharmacy::getDelFlag, DelEnum.NOT_DEL.val()));
+			if (count > 0) {
+				System.out.println(name + " 已存在,自动跳过");
+			}
+
+			WmDaPharmacy wmDaPharmacy = new WmDaPharmacy();
+            wmDaPharmacy.setPharmacyName(name);
+            wmDaPharmacy.setAddress(address);
+            wmDaPharmacy.setProvince("");
+            wmDaPharmacy.setCity("");
+            wmDaPharmacy.setArea("");
+            //更新审核信息
+            LocalDateTime now = LocalDateTime.now();
+            wmDaPharmacy.setCreateTime(now);
+            wmDaPharmacy.setCreateUser(-1);
+            // 根据地址转换经纬度
+            this.convertToLatAndLong(wmDaPharmacy);
+
+            System.out.println(String.format("第%s条,%s-%s-%s", i, name, address, wmDaPharmacy));
+
+			pharmacyList.add(wmDaPharmacy);
+        }
+		wmDaPharmacyService.saveBatch(pharmacyList);
+
+
+    }
+
+    /**
+     * 根据地址转换经纬度
+     *
+     * @param wmDaPharmacy
+     */
+    private void convertToLatAndLong(WmDaPharmacy wmDaPharmacy) {
+        String fullAddress = wmDaPharmacy.getProvince() + wmDaPharmacy.getCity() + wmDaPharmacy.getArea()
+                + wmDaPharmacy.getAddress();
+        JSONObject jsonObject = MapUtil.getLatAndLogByName(fullAddress);
+        if (jsonObject != null) {
+            wmDaPharmacy.setProvince(jsonObject.getString("province"));
+            wmDaPharmacy.setCity(jsonObject.getString("city"));
+            wmDaPharmacy.setArea(jsonObject.getString("district"));
+            String location = jsonObject.getString("location");
+            if (null != location && location.split(",").length > 0) {
+                wmDaPharmacy.setLongitude(location.split(",")[0]);
+                wmDaPharmacy.setLatitude(location.split(",")[1]);
+            }
+        }
+    }
+}