相关推荐recommended
Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题
作者:mmseoamin日期:2023-12-05

一、问题描述

使用Spring Cloud Gateway开发内部API网关时,当业务的Http请求体大小超过256K时,会出现如下报错:Exceeded limit on max bytes to buffer : 262144。

Spring Boot框架给了两种方式来修改这个大小的方式:

方式一:使用修改配置参数值,spring.max-in-memory-size: 1024 * 1024 的方式

spring:
  application:
    name: gateway
  codec:
    max-in-memory-size: 1024 * 1024

方式二:使用WebFluxConfigurer,通过set方法设置max-in-memory-size

	@Override
    public void configureHttpMessageCodecs(ServerCodecConfigurer configurer) {
        configurer.defaultCodecs().maxInMemorySize((int) maxInMemorySize.toBytes());
    }

但在实际上线以后发现并为解决256K的限制,这是为什么呢?

二、问题定位

首先排查了SpringBoot是如何使用spring.max-in-memory-size这个参数的,当业务在启动时候设置的参数到底是怎么执行的。

第一步:SpringBoot首先会自动加载autoconfigure包下面的CodecsAutoConfiguration类:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第1张这个类主要是加载Encoders和Decoders,如果我们使用的application/json的请求格式,并且项目里面也存在jackson2的包,则SpringBoot会加载,Jackson2JsonDecoder 和Jackson2JsonEncoder:

org.springframework.boot.autoconfigure.http.codec.CodecsAutoConfiguration
	@Configuration(proxyBeanMethods = false)
	@ConditionalOnClass(ObjectMapper.class)
	static class JacksonCodecConfiguration {
		@Bean
		@Order(0)
		@ConditionalOnBean(ObjectMapper.class)
		CodecCustomizer jacksonCodecCustomizer(ObjectMapper objectMapper) {
			return (configurer) -> {
				CodecConfigurer.DefaultCodecs defaults = configurer.defaultCodecs();
				// 这里会new一个Jackson2JsonDecoder
				defaults.jackson2JsonDecoder(new Jackson2JsonDecoder(objectMapper, EMPTY_MIME_TYPES));
				defaults.jackson2JsonEncoder(new Jackson2JsonEncoder(objectMapper, EMPTY_MIME_TYPES));
			};
		}
	}

第二步:将设置的Jackson2JsonDecoder交给org.springframework.http.codec.support.BaseDefaultCodecs类去做初始化的工作:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第2张

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第3张通过追踪到这里,我们可以看出来是执行了setMaxInMemorySize方法的,那么我们只需要知道这个set之前的值是从哪里来的。

第三步:setMaxInMemorySize修改参数

如果通过上述的两个方式来修改setMaxInMemorySize,会调用BaseDefaultCodecs的maxInMemorySize方法来设置的:

org.springframework.http.codec.support.BaseDefaultCodecs
 @Override
	public void maxInMemorySize(int byteCount) {
		if (!ObjectUtils.nullSafeEquals(this.maxInMemorySize, byteCount)) {
			this.maxInMemorySize = byteCount;
			initReaders();
		}
	}

到这里可以确定的逻辑是,不管是通过哪种set maxInMemorySize的方式,都会执行Jackson2JsonDecoder父类AbstractJackson2Decoder的setMaxInMemorySize方法来修改maxInMemorySize这个私有属性:Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第4张

而这个maxInMemorySize默认的256 * 1024 就是控制256K请求体的大小。具体可以查看org.springframework.http.codec.json.AbstractJackson2Decoder的decodeToMono方法,这个方法负责将http请求的body转换成json格式:

@Override
	public Mono decodeToMono(Publisher input, ResolvableType elementType,
			@Nullable MimeType mimeType, @Nullable Map hints) {
		return DataBufferUtils.join(input, this.maxInMemorySize)
				.flatMap(dataBuffer -> Mono.justOrEmpty(decode(dataBuffer, elementType, mimeType, hints)));
	}
 

DataBufferUtils.join(input, this.maxInMemorySize)方法里面使用了LimitedDataBufferList:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第5张

如果超过了maxByteCount则会抛出raiseLimitException,这里就是异常具体抛出的地方:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第6张

第四步:通过工具排查执行setMaxInMemorySize以后maxInMemorySize

既然代码流程里面执行了setMaxInMemorySize,但是却没生效,那么这里到底执行以后maxInMemorySize的值是多少呢?

通过arthas工具查看了AbstractJackson2Decoder值:

// sc 获取classLoadHash:
sc -d org.springframework.http.codec.json.AbstractJackson2Decoder
// 获取value
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[0].maxInMemorySize'
@Integer[1048576]
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[1].maxInMemorySize'
@Integer[262144]
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[2].maxInMemorySize'
@Integer[262144]
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[3].maxInMemorySize'
@Integer[262144]
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[4].maxInMemorySize'
@Integer[262144]
[arthas@48]$ vmtool -c 18b4aac2 -a getInstances --className org.springframework.http.codec.json.AbstractJackson2Decoder --express '#val=instances[5].maxInMemorySize'

通过获取多个AbstractJackson2Decoder实例的maxInMemorySize属性发现只有一个设置了1048576=1M,其他都是262144=256K,看到这里已经可以得出结论,不是setMaxInMemorySize没有生效,而是只生效了一个实例,因为初始化的时候只new了一个Jackson2JsonDecoder。

那其他的Jackson2JsonDecoder是从哪里来的?

第五步:查看Spring Cloud Gateway源码:源码地址

查看了Spring Cloud Gateway源码,发现了它给的代码中将Http Request Body 转换成Json的方式如下:

org.springframework.cloud.gateway.filter.factory.JsonToGrpcGatewayFilterFactory

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第7张

在witeWith方法里面定义一个将Reques转换成Json的方法deserializeJSONRequest:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第8张

这里我们会发现如果你按照官方给的这种写法,每次请求进来会new Jackson2JsonDecoder,每次new出来的新对象是没有执行setMaxInMemorySize操作的,所以后期new的对象都默认使用了256 * 1024。

到此问题已经得到结论,不是SpringBoot框架setMaxInMemorySize没有生效,而是后期new的对象没有重新setMaxInMemorySize。

三、解决方案

如果要解决上述问题,我们需要保证每个新new的Jackson2JsonDecoder都能够使用设置的spring.max-in-memory-size,方法可以参考如下几种解决方案:

方案一:重写org.springframework.http.codec.json.AbstractJackson2Decoder

这种方法简单粗暴,直接将AbstractJackson2Decoder的maxInMemorySize改成spring.max-in-memory-size,通过自己写的AbstractJackson2Decoder覆盖SpringBoot框架里的AbstractJackson2Decoder,亲测有效。

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第9张

方案二:将maxInMemorySize修改成static的私有属性

将maxInMemorySize改成static以后,则SpringBoot框架在初始化的时候执行一次setMaxInMemorySize则也能解决其他new的Jackson2JsonDecoder 对象maxInMemorySize不生效问题:

Spring Cloud Gateway 彻底解决Exceeded limit on max bytes to buffer : 262144报错问题,在这里插入图片描述,第10张

方案三:每次new Jackson2JsonDecoder对像时执行setMaxInMemorySize

private Flux deserializeJSONRequest(ServerWebExchange exchange) {
        return exchange.getRequest().getBody().map(dataBufferBody -> {
            ResolvableType targetType = ResolvableType.forType(JsonNode.class);
            Jackson2JsonDecoder Jackson2JsonDecoder = new Jackson2JsonDecoder();
            // 每次创建对象以后需要设置setMaxInMemorySize
            Jackson2JsonDecoder.setMaxInMemorySize(1024 * 1024);
            return Jackson2JsonDecoder.decode(dataBufferBody, targetType, null, null);
        }).cast(JsonNode.class);
    }