Dubbo2.6.2实现文件上传

Myth丶恋晨 2022-01-22 04:59 320阅读 0赞

前端：使用formData

后端：

消费者

/**
         * 修改用户头像
         *
         * @param headPic 用户头像
         * @return
         */
        @PostMapping(value = "alterUserHeadImgAction", headers = {"content-type=multipart/form-data"})
        public JSONObject alterUserHeadImg(@RequestParam("headPic") MultipartFile headPic) {
            responseBean = new JSONObject();
    
            try {
    
                //修改头像
                CustomUsers loginedUser = usersService.alterUserHeadImgService(userId,
                //因为MultipartFile对象未序列化，所以我们就转换为字节数组再传         
                headPic.getBytes(),headPic.getOriginalFilename());
    
            
            } catch (Exception e) {
                //处理错误信息
                e.printStackTrace();
            }
            return responseBean;
        }

提供者

/**
         * 修改用户头像
         *
         * @param userId  用户id
         * @param bytes 上传文件流
         * @param fileName 上传文件名
         * @return 修改后的用户对象
         * @throws Exception 抛出异常
         */
        @Override
        public CustomUsers alterUserHeadImgService(Integer userId, byte[] bytes,String fileName) throws Exception {
    
    
            //限制只能是png,jpg,jpeg
            String suffix = fileName.substring(fileName.lastIndexOf("."));
            if(!".jpg".equalsIgnoreCase(suffix) && !".png".equalsIgnoreCase(suffix) && !".jpeg".equalsIgnoreCase(suffix)) {
                throw new FileUnknowException("上传图片格式错误，只能是png,jpg,jpeg格式的文件");
            }
            //生成UUID,使文件名不会重复
            String uuid = UUID.randomUUID().toString();
            uuid = uuid.replaceAll("-", "");
            try {
                //上传
                //文件名
                String newFileName = uuid + suffix;
    
                //存入磁盘的路径
                String diskPath =  "路径" + newFileName;
    
        
                FileOutputStream fileOutputStream = new FileOutputStream(new File(diskPath));
                //写入磁盘
                fileOutputStream.write(bytes);
    
                //修改数据库头像路径
                usersMapper.updateUserHead(userId,diskPath);
                //查看修改后的用户新信息
                List<CustomUsers> users = usersMapper.selectById(userId);
    
    
                if(users == null || users.size() != 1) {
                    throw new UserUnknowException("账号有误,请重试");
                }
                return users.get(0);
            } catch (Exception e) {
                e.printStackTrace();
            }
            return null;
        }