iss提交7：

1、添加使用personnel服务的controller，把原来的controller放入legacy包中 2、在dubbo消费者配置文件中配置personnel服务

iss提交7：
1、添加使用personnel服务的controller，把原来的controller放入legacy包中 2、在dubbo消费者配置文件中配置personnel服务
徐烜
1 parent 8f5c3b7e
Showing 3 changed files with 183 additions and 4 deletions
src/main/java/com/bsth/controller/schedule/basicinfo/EmployeeController_facade.java
src/main/java/com/bsth/controller/schedule/basicinfo/EmployeeController.java → src/main/java/com/bsth/controller/schedule/basicinfo/legacy/EmployeeController.java
src/main/resources/dubbo/applicationContext_dubbo_consumer.xml
+package com.bsth.controller.schedule.basicinfo;
+
+import com.alibaba.dubbo.config.annotation.Reference;
+import com.bsth.common.ResponseCode;
+import com.bsth.control_v2.plan_module.common.dto.baseinfo.PersonnelDto;
+import com.bsth.control_v2.plan_module.common.dto.sys.UserDto;
+import com.bsth.control_v2.plan_module.common.service.BServiceFacade;
+import com.bsth.control_v2.plan_module.common.service.baseinfo.PersonnelServiceFacade;
+import com.bsth.controller.schedule.BController_facade;
+import com.bsth.service.schedule.EmployeeService;
+import com.bsth.service.schedule.utils.DataToolsFile;
+import org.apache.commons.lang3.StringUtils;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RequestMethod;
+import org.springframework.web.bind.annotation.RequestParam;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.multipart.MultipartFile;
+
+import javax.servlet.http.HttpServletResponse;
+import java.io.*;
+import java.util.*;
+
+/**
+ * 人员基础信息Controller_facade
+ */
+@RestController
+@RequestMapping("ee")
+public class EmployeeController_facade extends BController_facade<Integer, PersonnelDto> {
+    @Override
+    public void setCreateUserInfo(PersonnelDto personnelDto, Integer userId, Date createDate) {
+        UserDto userDto = UserDto.getBuilder()
+                .setId(userId)
+                .build();
+        personnelDto.setCreateBy(userDto);
+        personnelDto.setCreateDate(createDate);
+    }
+
+    @Override
+    public void setUpdateUserInfo(PersonnelDto personnelDto, Integer userId, Date updateDate) {
+        UserDto userDto = UserDto.getBuilder()
+                .setId(userId)
+                .build();
+        personnelDto.setUpdateBy(userDto);
+        personnelDto.setUpdateDate(updateDate);
+    }
+
+    @Reference
+    private PersonnelServiceFacade personnelServiceFacade;
+    @Override
+    protected BServiceFacade<Integer, PersonnelDto> getBServiceFacade() {
+        return personnelServiceFacade;
+    }
+    @RequestMapping(value = "/validate_gh", method = RequestMethod.GET)
+    public Map<String, Object> validate_gh(@RequestParam Map<String, Object> param) {
+        Map<String, Object> rtn = new HashMap<>();
+        try {
+            // 工号验证
+            PersonnelDto personnelDto = PersonnelDto.getBuilder()
+                    .setId(param.get("id_eq") == null ?
+                            null : Integer.valueOf(param.get("id_eq").toString()))
+                    .setCompanyCode(param.get("companyCode_eq") == null ?
+                            null : param.get("companyCode_eq").toString())
+                    .setJobCode(param.get("jobCode_eq") == null ?
+                            null : param.get("jobCode_eq").toString())
+                    .build();
+            this.personnelServiceFacade.validate_gh(personnelDto);
+            rtn.put("status", ResponseCode.SUCCESS);
+        } catch (Exception exp) {
+            rtn.put("status", ResponseCode.ERROR);
+            rtn.put("msg", exp.getMessage());
+        }
+
+        return rtn;
+    }
+
+    //------------------ TODO：导入导出逻辑，之后会服务化 -----------------//
+    @Autowired
+    private EmployeeService employeeService;
+    // 上传excel文件
+    @RequestMapping(value = "/uploadFile", method = RequestMethod.POST)
+    public Map<String, Object> uploadFile(MultipartFile file) {
+        Map<String, Object> rtn = new HashMap<>();
+        try {
+            DataToolsFile dataToolsFile = employeeService.uploadFile(file.getOriginalFilename(), file.getBytes());
+            // excel文件名
+            String fileName = dataToolsFile.getFile().getAbsolutePath();
+            Workbook wb = dataToolsFile.getFileType().getWorkBook(dataToolsFile.getFile());
+
+            // excel文件sheet
+            List<String> sheetnames = new ArrayList<>();
+            for (int i = 0; i < wb.getNumberOfSheets(); i ++) {
+                sheetnames.add(wb.getSheetAt(i).getSheetName());
+            }
+
+            wb.close();
+
+            rtn.put("status", ResponseCode.SUCCESS);
+            rtn.put("filename", fileName);
+            rtn.put("sheetnames", StringUtils.join(sheetnames, ","));
+        } catch (Exception exp) {
+            exp.printStackTrace();
+            rtn.put("status", ResponseCode.ERROR);
+            rtn.put("msg", exp.getMessage());
+        }
+        return rtn;
+    }
+    // 导入excel文件
+    @RequestMapping(value = "/importFile", method = RequestMethod.POST)
+    public Map<String, Object> importFile(@RequestParam Map<String, Object> params) {
+        Map<String, Object> rtn = new HashMap<>();
+
+        try {
+            File file = new File(String.valueOf(params.get("filename")));
+            if (!file.exists()) {
+                throw new Exception("导入文件不存在！");
+            }
+            employeeService.importData(file, params);
+
+            rtn.put("status", ResponseCode.SUCCESS);
+            rtn.put("msg", "导入文件成功");
+        } catch (Exception exp) {
+            rtn.put("status", ResponseCode.ERROR);
+            rtn.put("msg", exp.getMessage());
+        }
+
+        return rtn;
+    }
+    // 上传并导入excel文件
+    @RequestMapping(value = "/uploadAndImportFile", method = RequestMethod.POST)
+    public Map<String, Object> uploadAndImportFile(MultipartFile file) {
+        Map<String, Object> rtn = new HashMap<>();
+
+        try {
+            DataToolsFile dataToolsFile = employeeService.uploadFile(file.getOriginalFilename(), file.getBytes());
+            Map<String, Object> params = new HashMap<>();
+            employeeService.importData(dataToolsFile.getFile(), params);
+
+            rtn.put("status", ResponseCode.SUCCESS);
+            rtn.put("msg", "上传&导入文件成功");
+        } catch (Exception exp) {
+            exp.printStackTrace();
+            rtn.put("status", ResponseCode.ERROR);
+            rtn.put("msg", exp.getMessage());
+        }
+
+        return rtn;
+    }
+
+    // 导出数据到xls文件
+    @RequestMapping(value = "/exportFile", method = RequestMethod.GET)
+    public void exportFile(HttpServletResponse response,
+                           @RequestParam Map<String, Object> params) throws Exception {
+        DataToolsFile dataToolsFile = employeeService.exportData(params);
+        // 流输出导出文件
+        response.setHeader("content-type", "application/octet-stream");
+        response.setHeader("Content-Disposition", "attachment; filename=" + dataToolsFile.getFile().getName());
+        response.setContentType("application/octet-stream");
+
+        OutputStream os = response.getOutputStream();
+        BufferedOutputStream bos = new BufferedOutputStream(os);
+
+        InputStream is = new FileInputStream(dataToolsFile.getFile());
+        BufferedInputStream bis = new BufferedInputStream(is);
+
+        int length = 0;
+        byte[] temp = new byte[1 * 1024 * 10];
+        while ((length = bis.read(temp)) != -1) {
+            bos.write(temp, 0, length);
+        }
+        bos.flush();
+        bis.close();
+        bos.close();
+        is.close();
+    }
+
+}
-package com.bsth.controller.schedule.basicinfo;
+package com.bsth.controller.schedule.basicinfo.legacy;
 import com.bsth.common.ResponseCode;
 import com.bsth.controller.schedule.BController;
@@ -9,7 +9,6 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RequestParam;
-import org.springframework.web.bind.annotation.RestController;
 import java.util.HashMap;
 import java.util.Map;
@@ -17,8 +16,8 @@ import java.util.Map;
 /**
  * 人员基础信息Controller
  */
-@RestController
-@RequestMapping("ee")
+//@RestController
+//@RequestMapping("ee")
 public class EmployeeController extends BController<Personnel, Integer> {
     @Autowired
     private EmployeeService employeeService;
@@ -12,6 +12,8 @@
     <!-- 注册中心地址 -->
     <dubbo:registry protocol="zookeeper" address="127.0.0.1:2181" check="false" />
+    <!-- 人员信息服务 -->
+    <dubbo:reference interface="com.bsth.control_v2.plan_module.common.service.baseinfo.PersonnelServiceFacade" id="personnelServiceFacadeImpl" check="false" />
     <!-- 车辆信息服务 -->
     <dubbo:reference interface="com.bsth.control_v2.plan_module.common.service.baseinfo.CarServiceFacade" id="carServiceFacadeImpl" check="false" />
     <!-- 车辆设备信息服务 -->