diff --git a/flossom-modules/flossom-system/src/main/java/com/flossom/system/controller/WxInstrumentSerialController.java b/flossom-modules/flossom-system/src/main/java/com/flossom/system/controller/WxInstrumentSerialController.java index 1ff8746..1632224 100644 --- a/flossom-modules/flossom-system/src/main/java/com/flossom/system/controller/WxInstrumentSerialController.java +++ b/flossom-modules/flossom-system/src/main/java/com/flossom/system/controller/WxInstrumentSerialController.java @@ -10,6 +10,8 @@ import com.flossom.common.core.domain.entity.WxUserInstrument; import com.flossom.common.core.domain.req.WxInstrumentSerialExportVm; import com.flossom.common.core.domain.req.WxInstrumentSerialImportVm; import com.flossom.common.core.domain.req.WxUserInstrumentExportVm; +import com.flossom.common.core.exception.ServiceException; +import com.flossom.common.core.utils.StringUtils; import com.flossom.common.core.utils.poi.ExcelUtil; import com.flossom.common.core.web.controller.BaseController; import com.flossom.common.core.web.domain.AjaxResult; @@ -160,6 +162,11 @@ public class WxInstrumentSerialController extends BaseController @PostMapping("/importData") public AjaxResult importData(MultipartFile file, boolean updateSupport) throws Exception { + String fileName = file.getOriginalFilename(); + String suffix = fileName.substring(fileName.lastIndexOf(".")); + if (!StringUtils.equals(suffix, "xls") || !StringUtils.equals(suffix, "xlsx")) { + throw new ServiceException("请使用模板文件导入数据"); + } ExcelUtil util = new ExcelUtil(WxInstrumentSerialImportVm.class); List WxInstrumentSerialImportList = util.importExcel(file.getInputStream()); String message = wxInstrumentSerialService.handleImportData(WxInstrumentSerialImportList);