SpringCloud调用第三方接口实现文件下载——接口穿透

10 篇文章 0 订阅
4 篇文章 0 订阅

需求:

  1. 现在有两个系统,一个是SSM架构系统,一个是SpringCloud架构系统
  2. 现在SpringCloud系统要求下载SSM系统的文件,采取调用SSM系统接口的方式
  3. 通过调用SpringCloud接口实现下载

主要操作可以分成三步:

  1. 获取token
  2. 请求SSM系统获取文件的字节流和文件信息
  3. 将请求的字节流重新装配在SpringCloud系统的返回参数里

这样就完成了接口的穿透转载了。

token的获取就不说了,不同的系统获取的方式也不同,下面说一下获取SSM系统的字节流:

    @Override
    public void downloadFile(HttpServletResponse response, String fileId) {
        HttpHeaders headers = new HttpHeaders();
        String url = "http://xxxxxxxx/file/download?fileId=" + fileId;
        String token = "bearer " + this.sendToken();
        headers.add("Authorization", token);
        //请求SSM系统获取要下载的文件流
        ResponseEntity<byte[]> responseNew = Singleton.INSTANCE.getRestTemplate().exchange(url, HttpMethod.GET, new HttpEntity<byte[]>(headers), byte[].class);
        logger.info("=================状态码=================");
        logger.info(CommonConstant.LOG_LOG_PARAM, responseNew.getStatusCodeValue());
        logger.info("================返回信息================");
        logger.info(CommonConstant.LOG_LOG_PARAM, responseNew.getHeaders().getContentType());
        logger.info(CommonConstant.LOG_LOG_PARAM, responseNew.getHeaders().getContentType().getSubtype());
        logger.info("=======================================");
        // 下载文件
        this.sendFile(response, responseNew);
    }

this.sendToken()是获取token的方法;

this.sendFile(response, responseNew)是重新装配字节流的方法:

    private void sendFile(HttpServletResponse response, ResponseEntity<byte[]> responseNew) {
        //获取请求SSM系统返回报文中文件字节码
        byte[] result = responseNew.getBody();

        //将数据回写到SpringCloud系统返回参数
        response.setHeader("content-type", String.valueOf(responseNew.getHeaders().getContentType()));
        response.setHeader("Content-Disposition", String.valueOf(responseNew.getHeaders().getContentDisposition()));
        response.setHeader("Accept-Ranges", "bytes");
        response.setContentType(String.valueOf(responseNew.getHeaders().getContentType()));
        try (InputStream inputStream = new ByteArrayInputStream(result); OutputStream outputStream = response.getOutputStream()) {
            byte[] buff = new byte[1024];
            // 读取字节流
            int read = inputStream.read(buff);
            //通过while循环写入到指定了的对象中
            while (read != -1) {
                outputStream.write(buff, 0, buff.length);
                outputStream.flush();
                read = inputStream.read(buff);
            }
        } catch (IOException e) {
            e.printStackTrace();
            throw new CommonException(MessagesConstant.PTP_COMMON_FILE_DOWNLOAD_ERROR);
        }
    }

这时候就完成了文件接口的穿透了。我这个接口只要一个文件的id就可以下载了。

    @ApiOperation(value = "通用PTP文件下载")
    @GetMapping("/all/download/{fileId}")
    public ResponseEntity downloadPtpFile(@ApiParam(value = "文件ID", required = true, name = "fileId") @PathVariable String fileId, HttpServletResponse response) {
        fileService.downloadFile(response, fileId);
        return Results.success();
    }

调用这个接口,就可以下载SSM系统的文件了。

这里要注意一点,在springCloud中RestTemplate原来的配置可能用做Feign调用了,这是要重新创建一个RestTemplate来使用。这里给一个单例的写法

public enum RestTemplateSingleton {

    INSTANCE;

    public RestTemplate getRestTemplate() {
        // 长连接保持时长30秒
        PoolingHttpClientConnectionManager pollingConnectionManager = new PoolingHttpClientConnectionManager(
                30, TimeUnit.SECONDS);
        // 最大连接数
        pollingConnectionManager.setMaxTotal(3000);
        // 单路由的并发数
        pollingConnectionManager.setDefaultMaxPerRoute(100);
        HttpClientBuilder httpClientBuilder = HttpClients.custom();
        httpClientBuilder.setConnectionManager(pollingConnectionManager);
        // 重试次数2次,并开启
        httpClientBuilder.setRetryHandler(new DefaultHttpRequestRetryHandler(2, true));
        // 保持长连接配置,需要在头添加Keep-Alive
        httpClientBuilder
                .setKeepAliveStrategy(new DefaultConnectionKeepAliveStrategy());
        HttpClient httpClient = httpClientBuilder.build();
        // httpClient连接底层配置clientHttpRequestFactory
        HttpComponentsClientHttpRequestFactory clientHttpRequestFactory =
                new HttpComponentsClientHttpRequestFactory(httpClient);
        clientHttpRequestFactory.setReadTimeout(15000);
        clientHttpRequestFactory.setConnectTimeout(15000);

        return new RestTemplate(clientHttpRequestFactory);
    }

}

这样就创建一个可以使用的RestTemplate对象了。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值