diff --git a/cweb/src/main/java/com/hfkj/controller/UserController.java b/cweb/src/main/java/com/hfkj/controller/UserController.java index 2006c7e..6703bb9 100644 --- a/cweb/src/main/java/com/hfkj/controller/UserController.java +++ b/cweb/src/main/java/com/hfkj/controller/UserController.java @@ -1,6 +1,7 @@ package com.hfkj.controller; import com.alibaba.fastjson.JSONObject; +import com.hfkj.common.exception.BaseException; import com.hfkj.common.exception.ErrorCode; import com.hfkj.common.exception.ErrorHelp; import com.hfkj.common.exception.SysCode; @@ -99,6 +100,9 @@ public class UserController { FileOutputStream out = null; try { String fileType = file.getOriginalFilename().substring(file.getOriginalFilename().lastIndexOf(".") + 1); + if (!fileType.endsWith("jpg") && !fileType.endsWith("png")) { + throw ErrorHelp.genException(SysCode.System, ErrorCode.COMMON_ERROR, "只能上传jpg或png格式文件"); + } String fileName = System.currentTimeMillis() + "." + fileType; String childPath = "user-head"; String destDirName = sysConfig.getFileUrl() + File.separator + childPath; @@ -114,7 +118,10 @@ public class UserController { // 修改用户头像 user.setHeadImg(CommonSysConst.getSysConfig().getDomain()+"/filesystem/"+fileNames.get(0)); userService.updateInfo(user); + return ResponseMsgUtil.success(user.getHeadImg()); + } catch (BaseException e) { + throw ErrorHelp.genException(SysCode.System, ErrorCode.COMMON_ERROR, e.getErrorMsg()); } catch (Exception e) { log.error(e.getMessage(), e); throw ErrorHelp.genException(SysCode.System, ErrorCode.COMMON_ERROR, "上传失败"); @@ -126,7 +133,7 @@ public class UserController { } } } - return ResponseMsgUtil.success(user.getHeadImg()); + throw ErrorHelp.genException(SysCode.System, ErrorCode.COMMON_ERROR, "上传失败"); } catch (Exception e) { log.error(e.getMessage(), e);