Spring RestTemplate調用第三方接口上傳文件

功能需求

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目錄值是存在的。

--歡迎評論討論