package com.javaweb.web.controller.system;
|
|
|
import com.javaweb.common.annotation.Log;
|
import com.javaweb.common.core.controller.BaseController;
|
import com.javaweb.common.core.domain.AjaxResult;
|
import com.javaweb.common.enums.BusinessType;
|
import com.javaweb.common.utils.poiNew.ExcelUtilNew;
|
import com.javaweb.framework.util.ShiroUtils;
|
import com.javaweb.system.domain.SysDept;
|
import com.javaweb.system.domain.SysUser;
|
import com.javaweb.system.service.IExcelImportService;
|
import org.apache.shiro.authz.annotation.RequiresPermissions;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* 用户信息
|
*
|
* @author lws
|
*/
|
@Controller
|
@RequestMapping("/system/excelImport")
|
public class ExcelImportController extends BaseController {
|
private String prefix = "system/excelImport";
|
|
@Autowired
|
private IExcelImportService excelImportService;
|
|
@RequiresPermissions("system:excelImport:view")
|
@GetMapping()
|
public String user() {
|
return prefix + "/excelImport";
|
}
|
|
private static Map<String,String> excelNameMap = new HashMap<String,String>();
|
static {
|
excelNameMap.put("1","用户导入");
|
excelNameMap.put("2","部门导入");
|
}
|
@RequestMapping("/downloadExcelModel")
|
@ResponseBody
|
@Log(title = "Excel导入-模版下载", businessType = BusinessType.EXPORT)
|
public AjaxResult downloadExcelModel(String importType){
|
ExcelUtilNew excelUtil = excelUtilObj(importType);
|
return excelUtil.exportExcelModel(excelNameMap.get(importType));
|
}
|
@PostMapping("/save")
|
@ResponseBody
|
@Log(title = "Excel导入-导入", businessType = BusinessType.IMPORT)
|
public AjaxResult save(MultipartFile excelFile, String importType) {
|
try {
|
SysUser currentUser= ShiroUtils.getSysUser();
|
ExcelUtilNew excelUtil = excelUtilObj(importType);
|
List<Object> data = excelUtil.importExcel(excelFile.getInputStream());
|
return AjaxResult.success(excelImportService.saveData(data,importType,currentUser));
|
} catch (Exception e) {
|
e.printStackTrace();
|
return error("请选择正确的excel文件！");
|
}
|
}
|
|
/**
|
* 创建excel工具对象
|
* @param type
|
* @return
|
*/
|
private static ExcelUtilNew excelUtilObj(String type){
|
ExcelUtilNew excelUtil = null;
|
switch (type){
|
case "1":
|
excelUtil= new ExcelUtilNew(SysUser.class);
|
break;
|
case "2":
|
excelUtil= new ExcelUtilNew(SysDept.class);
|
break;
|
default:
|
throw new RuntimeException("不支持的导入数据类型！");
|
}
|
return excelUtil;
|
}
|
}
|
