From 80b6cbfc9c861469146318d0b3dd5f8b8b525b8a Mon Sep 17 00:00:00 2001 From: xiejun <xiejun@vci-tech.com> Date: 星期五, 01 十一月 2024 15:11:19 +0800 Subject: [PATCH] Revert "集成获取mdm分发通用数据格式接口集成" --- Source/BladeX-Tool/blade-starter-http/src/main/java/org/springblade/core/http/RetryInterceptor.java | 74 +++++++++++++++++++++++++++++++++++++ 1 files changed, 74 insertions(+), 0 deletions(-) diff --git a/Source/BladeX-Tool/blade-starter-http/src/main/java/org/springblade/core/http/RetryInterceptor.java b/Source/BladeX-Tool/blade-starter-http/src/main/java/org/springblade/core/http/RetryInterceptor.java new file mode 100644 index 0000000..a6ff367 --- /dev/null +++ b/Source/BladeX-Tool/blade-starter-http/src/main/java/org/springblade/core/http/RetryInterceptor.java @@ -0,0 +1,74 @@ +/* + * Copyright (c) 2018-2028, DreamLu All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * Neither the name of the dreamlu.net developer nor the names of its + * contributors may be used to endorse or promote products derived from + * this software without specific prior written permission. + * Author: DreamLu 鍗㈡槬姊� (596392912@qq.com) + */ +package org.springblade.core.http; + +import lombok.RequiredArgsConstructor; +import okhttp3.Interceptor; +import okhttp3.Request; +import okhttp3.Response; +import okhttp3.ResponseBody; +import org.springframework.retry.backoff.FixedBackOffPolicy; +import org.springframework.retry.policy.SimpleRetryPolicy; +import org.springframework.retry.support.RetryTemplate; + +import java.io.IOException; +import java.util.function.Predicate; + +/** + * 閲嶈瘯鎷︽埅鍣紝搴斿浠g悊闂 + * + * @author L.cm + */ +@RequiredArgsConstructor +public class RetryInterceptor implements Interceptor { + private final RetryPolicy retryPolicy; + + @Override + public Response intercept(Chain chain) throws IOException { + Request request = chain.request(); + RetryTemplate template = createRetryTemplate(retryPolicy); + return template.execute(context -> { + Response response = chain.proceed(request); + // 缁撴灉闆嗘牎楠� + Predicate<ResponseSpec> respPredicate = retryPolicy.getRespPredicate(); + if (respPredicate == null) { + return response; + } + // copy 涓�浠� body + ResponseBody body = response.peekBody(Long.MAX_VALUE); + try (HttpResponse httpResponse = new HttpResponse(response)) { + if (respPredicate.test(httpResponse)) { + throw new IOException("Http Retry ResponsePredicate test Failure."); + } + } + return response.newBuilder().body(body).build(); + }); + } + + private static RetryTemplate createRetryTemplate(RetryPolicy policy) { + RetryTemplate template = new RetryTemplate(); + // 閲嶈瘯绛栫暐 + SimpleRetryPolicy retryPolicy = new SimpleRetryPolicy(); + retryPolicy.setMaxAttempts(policy.getMaxAttempts()); + // 璁剧疆闂撮殧绛栫暐 + FixedBackOffPolicy backOffPolicy = new FixedBackOffPolicy(); + backOffPolicy.setBackOffPeriod(policy.getSleepMillis()); + template.setRetryPolicy(retryPolicy); + template.setBackOffPolicy(backOffPolicy); + return template; + } +} -- Gitblit v1.9.3