From e613dc170308423742325db4a8bd9a3e01659add Mon Sep 17 00:00:00 2001
From: luliqiang <kidgrow>
Date: Fri, 19 Feb 2021 10:28:11 +0800
Subject: [PATCH] 增加合作商信息管理功能

---
 kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/ScreeningOrganizationInfoController.java |  140 +++++++++++++++++++++++++++++++++++++++++-----
 1 files changed, 125 insertions(+), 15 deletions(-)

diff --git a/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/ScreeningOrganizationInfoController.java b/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/ScreeningOrganizationInfoController.java
index 9c856e7..32e9fb3 100644
--- a/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/ScreeningOrganizationInfoController.java
+++ b/kidgrow-business/kidgrow-opration-center/kidgrow-opration-center-server/src/main/java/com/kidgrow/oprationcenter/controller/ScreeningOrganizationInfoController.java
@@ -1,11 +1,14 @@
 package com.kidgrow.oprationcenter.controller;
 
+import cn.hutool.core.bean.BeanUtil;
 import cn.hutool.http.HttpRequest;
 import com.alibaba.fastjson.JSONObject;
 import com.kidgrow.common.controller.BaseController;
 import com.kidgrow.common.model.PageResult;
 import com.kidgrow.common.model.ResultBody;
+import com.kidgrow.common.utils.ExcelUtil;
 import com.kidgrow.common.utils.QRCodeUtil;
+import com.kidgrow.oprationcenter.model.ScreeningOrganizationExcel;
 import com.kidgrow.oprationcenter.model.ScreeningOrganizationInfo;
 import com.kidgrow.oprationcenter.service.IScreeningOrganizationInfoService;
 import com.kidgrow.redis.util.RedisUtils;
@@ -17,14 +20,17 @@
 import org.apache.commons.collections4.MapUtils;
 import org.apache.commons.lang.StringUtils;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.io.ClassPathResource;
 import org.springframework.validation.BindingResult;
 import org.springframework.validation.ObjectError;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 import sun.misc.BASE64Decoder;
 
 import javax.servlet.http.HttpServletResponse;
 import javax.validation.Valid;
 import java.io.ByteArrayInputStream;
+import java.io.File;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.util.ArrayList;
@@ -104,15 +110,16 @@
             }
             return ResultBody.failed().msg(errMsg.toString());
         } else {
-            if (organizationInfo.getId() == null) {
-                //新增机构增加关注二维码
-                String url = createQr(String.valueOf(organizationInfo.getId()));
-                if (StringUtils.isNotBlank(url)) {
-                    organizationInfo.setQrCode(url);
-                }
-            }
             boolean v = organizationInfoService.saveOrUpdate(organizationInfo);
             if (v) {
+                if (StringUtils.isBlank(organizationInfo.getQrCode())) {
+                    //新增机构增加关注二维码
+                    String url = createQr(String.valueOf(organizationInfo.getId()));
+                    if (StringUtils.isNotBlank(url)) {
+                        organizationInfo.setQrCode(url);
+                        organizationInfoService.updateById(organizationInfo);
+                    }
+                }
                 return ResultBody.ok().data(organizationInfo).msg("保存成功");
             } else {
                 return ResultBody.failed().msg("保存失败");
@@ -122,7 +129,7 @@
 
     private String createQr(String id) {
         String url = "";
-        String str = "{\"action_name\": \"QR_LIMIT_STR_SCENE\", \"action_info\": {\"scene\": {\"scene_str\": \"kagscreening_\" " + id + "}}}";
+        String str = "{\"action_name\": \"QR_LIMIT_STR_SCENE\", \"action_info\": {\"scene\": {\"scene_str\": \"kagscreening_" + id + "\"}}}";
 
         String token = redisUtils.get("Wechat:Token:KagReceiveToken").toString();
         String result = HttpRequest.post("https://api.weixin.qq.com/cgi-bin/qrcode/create?access_token=" + token)
@@ -180,22 +187,125 @@
             out = response.getOutputStream();
             fileInputStream = new ByteArrayInputStream(arr);
             int len = 0;
-            byte[] by = new byte[1024*10];
-            while ((len = fileInputStream.read(by))>0){
-                out.write(by,0,len);
+            byte[] by = new byte[1024 * 10];
+            while ((len = fileInputStream.read(by)) > 0) {
+                out.write(by, 0, len);
             }
 
-        }catch (Exception w){
-            log.error("",w);
-        }finally {
+        } catch (Exception w) {
+            log.error("", w);
+        } finally {
             try {
                 out.close();
                 fileInputStream.close();
-            }catch(Exception w){
+            } catch (Exception w) {
 
             }
 
         }
 
     }
+
+    /**
+     * 导入文件
+     */
+    @ApiOperation(value = "导入")
+    @PostMapping("/import")
+    public ResultBody downLoad(@RequestParam("file") MultipartFile excel,
+                               @RequestParam("hospitalId") String hospitalId,
+                               @RequestParam("areaCode") String areaCode,
+                               @RequestParam("areaName") String areaName,
+                               @RequestParam("hospitalName") String hospitalName) {
+
+        int rowNum = 0;
+
+        try {
+            if (!excel.isEmpty()) {
+                List<ScreeningOrganizationExcel> list = ExcelUtil.importExcel(excel, 0, 1, ScreeningOrganizationExcel.class);
+                rowNum = list.size();
+                if (rowNum > 0) {
+                    String[] area = areaCode.split("#");
+
+                    list.forEach(u -> {
+                        ScreeningOrganizationInfo organizationInfo = new ScreeningOrganizationInfo();
+                        BeanUtil.copyProperties(u, organizationInfo);
+                        organizationInfo.setSourceHospitalName(hospitalName);
+                        organizationInfo.setSourceHospitalId(Long.parseLong(hospitalId));
+                        organizationInfo.setProvince(area[0]);
+                        organizationInfo.setCity(area[1]);
+                        organizationInfo.setArea(area[2]);
+                        organizationInfo.setAreaCode(areaName);
+                        organizationInfoService.saveOrUpdate(organizationInfo);
+                        if (StringUtils.isBlank(organizationInfo.getQrCode())) {
+                            organizationInfo.setQrCode(createQr(String.valueOf(organizationInfo.getId())));
+                            organizationInfoService.updateById(organizationInfo);
+                        }
+                    });
+
+                }
+            }
+        } catch (Exception e) {
+            log.error("导入失败", e);
+        }
+        return ResultBody.ok().msg("导入数据成功,一共【" + rowNum + "】行");
+    }
+
+    /**
+     * 下载模板
+     */
+    @ApiOperation(value = "下载模板")
+    @PostMapping("/export")
+    public void export(HttpServletResponse response) {
+        try {
+            InputStream inStream = null;
+            ClassPathResource resource = new ClassPathResource(File.separator + "template" + File.separator + "机构导入模板.xlsx");
+            if (resource.exists()) {
+                inStream = resource.getInputStream();
+            }
+            // 设置输出的格式
+            response.reset();
+            response.setContentType("bin");
+            response.addHeader("Content-Disposition", "attachment; filename=机构导入模板.xlsx");
+            // 循环取出流中的数据
+            byte[] b = new byte[100];
+            int len;
+
+            while ((len = inStream.read(b)) > 0) {
+                response.getOutputStream().write(b, 0, len);
+            }
+            inStream.close();
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    /**
+     * 通过医院ID查询机构列表
+     */
+    @ApiOperation(value = "通过医院ID查询机构列表")
+    @PostMapping("/getOrganizationListByHospitalId")
+    public ResultBody getOrganizationListByHospitalId(@RequestBody String id) {
+        List<ScreeningOrganizationInfo> byHospitalId = organizationInfoService.findByHospitalId(Long.valueOf(id));
+        if (byHospitalId != null && byHospitalId.size() > 0) {
+            return ResultBody.ok().data(byHospitalId).msg("查询成功");
+        }
+        return ResultBody.failed().msg("查询失败");
+    }
+
+    /**
+     * 关键字查询列表
+     */
+    @ApiOperation(value = "关键字查询列表")
+    @ApiImplicitParams({
+            @ApiImplicitParam(name = "page", value = "分页起始位置", required = true, dataType = "Integer"),
+            @ApiImplicitParam(name = "limit", value = "分页结束位置", required = true, dataType = "Integer")
+    })
+    @PostMapping(value = "/listByKeyWord")
+    public PageResult<ScreeningOrganizationInfo> listByKeyWord(@RequestParam Map<String, Object> params) {
+        if (params.size() == 0) {
+            params.put("page", 1);
+            params.put("limit", 10);
+        }
+        return organizationInfoService.findByKeyWord(params);
+    }
 }

--
Gitblit v1.8.0