elasticsearch 修改查询结果响应数据大小

小橘快跑

于 2024-04-30 15:12:46 发布

阅读量109

点赞数 1

分类专栏： elasticsearch 文章标签： elasticsearch

本文链接：https://blog.csdn.net/qq_50761817/article/details/138346627

版权

elasticsearch 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

/**
 * 1、通过拦截器的方式修改 RequestOptions 的大小
 * 2、通过在请求时进行修改
 */
@Configuration
public class MvcESConfig implements WebMvcConfigurer {
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        //设置ElasticsearchRestTemplate查询的最大值为300MB 默认为100MB
        registry.addInterceptor(new HandlerInterceptor() {
            private boolean isSetBuffer = false;

            @Override
            public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
                //已经设置过
                if (isSetBuffer) {
                    return true;
                }
                //设置es查询buffer大小
                RequestOptions requestOptions = RequestOptions.DEFAULT;
                Class<? extends RequestOptions> aClass = requestOptions.getClass();
                Field aDefault = aClass.getDeclaredField("httpAsyncResponseConsumerFactory");
                aDefault.setAccessible(true);
                //去除final
                Field modifiersField = Field.class.getDeclaredField("modifiers");
                modifiersField.setAccessible(true);
                modifiersField.setInt(aDefault, aDefault.getModifiers() & ~Modifier.FINAL);

                //设置默认的工厂
                aDefault.set(requestOptions, new HttpAsyncResponseConsumerFactory() {
                    @Override
                    public HttpAsyncResponseConsumer<HttpResponse> createHttpAsyncResponseConsumer() {
                        //500MB
                        return new HeapBufferedAsyncResponseConsumer(500 * 1024 * 1024);
                    }
                });
                //标记
                isSetBuffer = true;

                return true;
            }
        });
    }

    /**
     * 自定义RequestOptions 为 500M ，用这个替换 RequestOptions.DEFAULT
     * @return
     */
    private RequestOptions updateRequestOptions(){
        RequestOptions.Builder builder = RequestOptions.DEFAULT.toBuilder();
        builder.setHttpAsyncResponseConsumerFactory(
                new HttpAsyncResponseConsumerFactory
                        //修改为500MB
                        .HeapBufferedResponseConsumerFactory(500 * 1024 * 1024));
        return builder.build();
    }
}