Spring RestTemplate调用HTTP接口上传文件

柔情只为你懂 2023-10-10 10:59 15阅读 0赞

#### 功能需求 ####

APP端上传图片到后端，后端需要调用第三方接口发送得到的图片，然后获取返回结果响应给APP。

#### 解决思路 ####

先接收到图片在服务地址临时缓存，然后把图片取出调用第三方接口发送，获取到返回结果后删除本地临时缓存的图片完成操作。

#### 实现 ####

@Slf4j
    public Object updateUserFavicon(@RequestParam("uploadFile") MultipartFile multipartFile,
                                        HttpServletRequest req) {
            //请求头参数
            String authorization = req.getHeader("Authorization");
            //请求参数
            String ucUid = req.getParameter("ucUid");
            SimpleDateFormat sdf = new SimpleDateFormat("yyyy/MM");
            //获取工程目录 （项目在容器中的实际发布运行的根路径）
            String realPath = req.getSession().getServletContext().getRealPath("/uploadFile/");
            String format = sdf.format(new Date());
            //保存路径 /uploadFile/yyyy/MM/
            File folder = new File(realPath + format);
            if (!folder.isDirectory()) {
                folder.mkdirs();
            }
            String newName = UUID.randomUUID().toString();
            try {
                //获取文件扩展名
                String extName = FileUtil.extName(multipartFile.getOriginalFilename());
                File filePath = File.createTempFile(newName, "." + extName,folder);
                //文件保存操作
                multipartFile.transferTo(filePath);
                log.info("============文件保存路径==============  : {}", filePath);
                //设置请求头
                HttpHeaders headers = new HttpHeaders();
                MediaType type = MediaType.parseMediaType("multipart/form-data");
                headers.add("Authorization", authorization);
                headers.setContentType(type);
                //转为FileSystemResource 用于发送文件
                FileSystemResource fileSystemResource = new FileSystemResource(filePath);
                //设置请求体，注意是LinkedMultiValueMap
                MultiValueMap<String, Object> form = new LinkedMultiValueMap<>();
                form.add("ucUid", ucUid);
                form.add("file", fileSystemResource);
                form.add("filename", newName);
                //用HttpEntity封装整个请求报文
                HttpEntity<MultiValueMap<String, Object>> files = new HttpEntity<>(form, headers);
                ResponseEntity<Map> entity = restTemplate.postForEntity(AppUserMsgRequestEnum.updateUserFavicon.getAllUrl(), files, Map.class);
                //access_token过期返回状态码401
                if (entity.getStatusCode().value() == 401) {
                    return R.ok("401", "用户access_token过期");
                }
                Map map1 = entity.getBody();
                int status = (int) map1.get("status");
                if (status == 0) {
                    File del = new File(filePath.toURI());
                    del.delete();
                    return R.ok().data(map1);
                 } else {
                    return R.ok(ErrorCodeEnum.P_ERROR.getCode(), "操作失败");
                }
            } catch (IOException e) {
                e.printStackTrace();
                return R.ok(ErrorCodeEnum.P_ERROR.getCode(), "系统异常");
            }
        }

注意：

21行中的File.createTempFile(String prefix, String suffix, File directory)在调用时，directory目录如果不存在的话，临时文件会创建失败，所以要确定传入的directory目录值是存在的。

\--欢迎评论讨论