springBoot接口中直接返回图片数据

发表于 | 分类于

起因

最近在做涉及到分享推广的业务，需要由业务员分享二维码进入推广页面，由于是新项目，前期预算和用量都有限，没有搭建对象存储服务，所以决定使用后台服务动态生成二维码图片直接图片数据并返回。

首先是二维码的生成，决定使用google的zxing，毕竟google的东西还是不错的，maven添加依赖如下：

1
2
3
4
5
6
7
8
9
10
11
12
<!-- https://mvnrepository.com/artifact/com.google.zxing/core -->
        <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>core</artifactId>
            <version>3.3.3</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/com.google.zxing/javase -->
        <dependency>
            <groupId>com.google.zxing</groupId>
            <artifactId>javase</artifactId>
            <version>3.3.3</version>
        </dependency>

继续查zxing的使用方法，发现大多数都是生成二维码然后写成图片文件的，不太适合我现在的情况。

类似这种

1
2
3
4
5
6
7
Map hints = new HashMap();
      hints.put(EncodeHintType.CHARACTER_SET, "utf-8");
      hints.put(EncodeHintType.ERROR_CORRECTION, ErrorCorrectionLevel.M);
      hints.put(EncodeHintType.MARGIN, 2);
      BitMatrix qrcode = new QRCodeWriter().encode(href, BarcodeFormat.QR_CODE, 300, 300);
      //网上的方案大多数都是通过io流写到文件系统，
      MatrixToImageWriter.writeToStream(qrcode,"png",response.getOutputStream());

于是企图用response的输出流返回，但是返回的数据浏览器看到的全是乱码，这种方案并没有成功

但是根据个人经验，一般这种开源方案既然二维码数据BitMatrix对象都生成了，肯定有获取原始数据的方法，点进MatrixToImageWriter类搜索方法，果然，找到了能直接返回BufferedImage对象的方法

在这里插入图片描述
现在，BufferedImage对象已经有了，只差把它扔回前端了，继续百度，发现可以直接返回该对象，类似以下配置

1
2
3
4
5
@GetMapping(value = "/qrcode", produces = MediaType.IMAGE_JPEG_VALUE)
@ResponseBody
public BufferedImage generateQRCode() {
        //返回BufferedImage的对象
    }

以为问题即将解决，然而浏览器访问返回406，上网一查，原来是没有对应消息类型的转换器导致的，有博主提到需要如下配置

1
2
3
4
@Bean
    public BufferedImageHttpMessageConverter addConverter(){
        return new BufferedImageHttpMessageConverter();
    }

加了上面的配置后发现问题仍没有解决，报错仍是406，怀疑配置没有生效，于是决定走源码查看原因。debug源码时发现messageConverters的list中确实没有我配置的，说明的确是配置问题，查找messageConverters的set操作，查到如图的地方

在这里插入图片描述
发现springMVC是在配置RequestMappingHandlerAdapter设置的HttpMessageConverter，进入getMessageConverters()方法
在这里插入图片描述
根据我的工地英语8级，extendMessageConverters这个方法应该是在添加自定义的HttpMessageConverter，进入该方法
在这里插入图片描述
空实现，很明显估计是模板模式，需要自己去扩展，于是自己写了一个配置类继承WebMvcConfigurationSupport,重写extendMessageConverters方法

1
2
3
4
@Override
protected void extendMessageConverters(List<HttpMessageConverter<?>> converters) {
    converters.add(new BufferedImageHttpMessageConverter());
}

浏览器再访问，二维码图片展示，问题解决

总结

实现一个方案的过程中碰到了各种各样的奇怪问题，最好的方式是先网上找资料快速解决问题，如果无法解决，再通过自己走源码的方式从根本原因上寻找出现问题的原因，解决问题最复杂的地方是定位问题，问题定位了，解决便不再是难题