package com.zbooksoft.gdmis.controller;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.ruili.wcp.common.EncryptUtil;
|
import com.ruili.wcp.data.entity.management.Dept;
|
import com.ruili.wcp.data.entity.management.Role;
|
import com.ruili.wcp.data.entity.management.User;
|
import com.ruili.wcp.service.config.ImportLogService;
|
import com.ruili.wcp.service.management.DeptService;
|
import com.ruili.wcp.service.management.RoleService;
|
import com.ruili.wcp.service.management.UserService;
|
import com.ruili.wcp.web.common.LicenseException;
|
import com.ruili.wcp.web.model.AjaxResponse;
|
import com.ruili.wcp.web.model.ErrorInfo;
|
import org.apache.poi.ss.usermodel.Cell;
|
import org.apache.poi.ss.usermodel.Row;
|
import org.apache.poi.ss.usermodel.Sheet;
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
import org.apache.shiro.authz.annotation.RequiresUser;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RequestParam;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.servlet.ModelAndView;
|
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.ArrayList;
|
import java.util.Date;
|
import java.util.List;
|
|
import static com.ruili.wcp.web.controller.general.DataImportController.getExcelValue;
|
|
/**
|
* @Description:
|
* @Author: zhai
|
* @Date: 2025/11/25
|
**/
|
@Controller
|
@RequestMapping("/ghUser")
|
public class GhUserController {
|
@Autowired
|
DeptService deptService;
|
|
@Autowired
|
UserService userService;
|
|
@Autowired
|
RoleService roleService;
|
private static final Logger logger = LoggerFactory.getLogger(GhUserController.class);
|
|
/**
|
* xml导入页面
|
*
|
* @return
|
*/
|
@RequestMapping(value = "/userIndex")
|
@RequiresUser
|
public ModelAndView wasm() {
|
ModelAndView mv = new ModelAndView("gh/user/userIndex");
|
return mv;
|
}
|
|
/**
|
* @param file 0 导入转孔信息 1 导入图幅信息 2 导入实物案卷信息
|
* @return
|
* @throws LicenseException
|
*/
|
|
@RequestMapping({"/importExcel"})
|
@ResponseBody
|
@RequiresUser
|
public Object importExcel(@RequestParam(value = "file", required = false) MultipartFile file) throws LicenseException {
|
try {
|
readExcel(file);
|
return new AjaxResponse(true);
|
} catch (Exception e) {
|
logger.error(e.getMessage(), e);
|
return new AjaxResponse(new ErrorInfo("导入失败"), false);
|
}
|
}
|
|
private void readExcel(MultipartFile file) throws IOException {
|
|
String originalFilename = file.getOriginalFilename();
|
|
InputStream inputStream = file.getInputStream();
|
if (originalFilename.matches("^.+\\.(?i)(xls)$")) {
|
//创建工作簿对象
|
// HSSFWorkbook hssfWorkbook = new HSSFWorkbook(inputStream);
|
// HashMap<String, Object> map = readProject2003(hssfWorkbook);
|
// readExcel2003(hssfWorkbook, map);
|
|
}
|
if (originalFilename.matches("^.+\\.(?i)(xlsx)$")) {
|
//创建工作簿对象
|
XSSFWorkbook xssfWorkbook = new XSSFWorkbook(inputStream);
|
insertDept(xssfWorkbook.getSheetAt(1));
|
insertUser(xssfWorkbook.getSheetAt(0));
|
|
}
|
}
|
|
|
private void insertDept(Sheet sheet) {
|
int maxRow = sheet.getLastRowNum();
|
for (int j = 1; j <= maxRow; j++) {
|
try {
|
Row newRow = sheet.getRow(j);
|
|
Dept dept = new Dept();
|
dept.setIcon("fa fa-address-card");
|
dept.setDeptType(0);
|
dept.setSortNum(j);
|
|
Cell deptName = newRow.getCell(1);
|
String deptNameString = getExcelValue(deptName);
|
String[] split1 = deptNameString.split("\\.");
|
dept.setDeptName(split1[split1.length - 1]);
|
dept.setDeptExtendStringField5(deptNameString);
|
|
Cell deptNum = newRow.getCell(2);
|
dept.setDeptNum(getExcelValue(deptNum));
|
|
Cell deptCode = newRow.getCell(3);
|
String deptCodeString = getExcelValue(deptCode);
|
dept.setDeptCode(deptCodeString);
|
|
String[] split = deptCodeString.split("\\.");
|
String parentDeptCode = "";
|
if (split.length > 1) {
|
if (split.length == 2) {
|
parentDeptCode = split[0];
|
}
|
if (split.length == 3) {
|
parentDeptCode = split[0] + "." + split[1];
|
}
|
QueryWrapper<Dept> deptQueryWrapper = new QueryWrapper<>();
|
deptQueryWrapper.eq("dept_code", parentDeptCode);
|
List<Dept> list = deptService.list(deptQueryWrapper);
|
if (list.size() > 0) {
|
dept.setParentId(list.get(0).getDeptId());
|
}
|
} else {
|
dept.setParentId(0L);
|
}
|
Cell oaDeptId = newRow.getCell(0);
|
dept.setDeptExtendStringField1(getExcelValue(oaDeptId));
|
deptService.saveOrUpdate(dept);
|
} catch (Exception e) {
|
logger.error(e.getMessage(), e);
|
}
|
}
|
}
|
|
private void insertUser(Sheet sheet) {
|
int maxRow = sheet.getLastRowNum();
|
for (int j = 1; j <= maxRow; j++) {
|
Row newRow = sheet.getRow(j);
|
User user = new User();
|
Cell userName = newRow.getCell(1);
|
user.setUserName(getExcelValue(userName));
|
Cell mobile = newRow.getCell(4);
|
user.setMobilePhone(getExcelValue(mobile));
|
String mobileString = getExcelValue(mobile);
|
if (mobileString.length() >= 4) {
|
String pwd = mobileString.substring(mobileString.length() - 4, mobileString.length());
|
String newPwd = "gh@" + pwd;
|
String md5Pwd = EncryptUtil.getInstance().MD5(newPwd);
|
user.setPassword(md5Pwd);
|
} else {
|
String newPwd = "gh@0000";
|
String md5Pwd = EncryptUtil.getInstance().MD5(newPwd);
|
user.setPassword(md5Pwd);
|
}
|
Cell trueName = newRow.getCell(2);
|
user.setTrueName(getExcelValue(trueName));
|
Byte enableLogin = 0;
|
user.setEnableLogin(enableLogin);
|
user.setUserType(1);
|
|
user.setCreateTime(new Date());
|
Cell deptName = newRow.getCell(3);
|
String deptNameString = getExcelValue(deptName);
|
// String[] split = deptNameString.split("\\.");
|
// String deptCode = split[split.length - 1];
|
LambdaQueryWrapper<Dept> wrapper = new LambdaQueryWrapper<>();
|
wrapper.eq(Dept::getDeptExtendStringField5, deptNameString);
|
Dept dept = deptService.getOne(wrapper);
|
user.setDeptId(dept.getDeptId());
|
user.setDeptName(dept.getDeptName());
|
Cell oldId = newRow.getCell(0);
|
user.setUserExtendStringField1(getExcelValue(oldId));
|
//插入系统用户表
|
userService.saveUser(user);
|
//初始化用户默认角色
|
List<Role> roles = roleService.addUserToDefaultRole(user.getUserId());
|
}
|
}
|
}
|
