spring boot 自定義Feign調(diào)用loadbalance

1:創(chuàng)建Feign接口定義


@FeignClient(name = "user-service",configuration = UserServiceFeignClientConfiguration.class)
public interface UserRegistrationService {

    @PostMapping(value = "/user/register", produces = "application/json;v=3") // V3
    Boolean registerUser(@RequestBody @Validated @Valid User user);

}

2:創(chuàng)建項(xiàng)目bootstrap啟動(dòng)類

@EnableAutoConfiguration
@EnableFeignClients(clients = UserRegistrationService.class )
@LoadBalancerClient(name = "user-service",configuration = UserServiceLoadBalanceConfiguration.class)
public class FeignClientSpringBootstrap {

    public static void main(String[] args) {
        ConfigurableApplicationContext context
                = new SpringApplicationBuilder(FeignClientSpringBootstrap.class).web(WebApplicationType.NONE)
                .build().
                run(args);


        UserRegistrationService userRegistrationService = context.getBean(UserRegistrationService.class);

        User user = new User();
        user.setId(2L);
        user.setName("test");
        Boolean res = userRegistrationService.registerUser(user);

        System.out.println("res==="+res);
        context.close();
    }
}

3:創(chuàng)建loadbalance配置類UserServiceLoadBalanceConfiguration


@Configuration
public class UserServiceLoadBalanceConfiguration {


    @Bean
    public ReactorLoadBalancer<ServiceInstance> randomLoadBalancer(Environment environment, LoadBalancerClientFactory loadBalancerClientFactory){
        String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
        return new RandomLoadBalancer(loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class),name);
    }


    @Bean
    public ServiceInstanceListSupplier serviceInstanceListSupplier(){
        return new UserServiceServiceInstanceListSupplier();
    }

}

參考官方文檔鏈接

image.png

4:創(chuàng)建UserServiceServiceInstanceListSupplier類返回loadbalance調(diào)用那個(gè)服務(wù)實(shí)列

public class UserServiceServiceInstanceListSupplier implements ServiceInstanceListSupplier {
    @Override
    public String getServiceId() {
        return "user-service";
    }

    @Override
    public Flux<List<ServiceInstance>> get() {
        return Flux.just(Arrays.asList(getLocalServiceInstance()));
    }

    private ServiceInstance getLocalServiceInstance(){
        DefaultServiceInstance instance = new DefaultServiceInstance();

        instance.setInstanceId("user-service");
        instance.setHost("127.0.0.1");
        instance.setPort(8091);
        return instance;
    }
}


5:創(chuàng)建Feign處理Response類ApiResponseDecoder

public class ApiResponseDecoder implements Decoder {

    private final  Decoder decoder;

    public ApiResponseDecoder(Decoder decoder) {
        this.decoder = decoder;
    }

    @Override
    public Object decode(Response response, Type type) throws IOException, DecodeException, FeignException {
       String contentType = getContentType(response);

       MediaType mediaType = MediaType.parseMediaType(contentType);

       String version = mediaType.getParameter("v");

       if(version == null){

           Object object = decoder.decode(response, ApiResponse.class);

           if(object instanceof ApiResponse){
               return ApiResponse.class.cast(object).getBody();
           }

       }
        return decoder.decode(response,type);
    }

    private String getContentType(Response response){
        Collection<String> types=response.headers().getOrDefault("Content-Type", Arrays.asList(""));
        return types.iterator().next();
    }
}

6:配置ApiResponseDecoder



@Configuration
public class UserServiceFeignClientConfiguration {


    @Autowired
    private  ObjectFactory<HttpMessageConverters> messageConverters;


    @Bean
    public Decoder feignDecoder(){
        return new ApiResponseDecoder(new ResponseEntityDecoder(new SpringDecoder(messageConverters)));
    }

}

7:運(yùn)行結(jié)果

image.png

8:代碼執(zhí)行邏輯

EnableFeignClients:啟動(dòng)Feign配置,生成代理對(duì)象UserRegistrationService
LoadBalancerClient(name = "user-service",configuration = UserServiceLoadBalanceConfiguration.class):
配置Feign遠(yuǎn)程調(diào)用時(shí)候loadbalance配置,如果沒有配置,走默認(rèn)配置

@FeignClient(name = "user-service",configuration = UserServiceFeignClientConfiguration.class):
配置Feign返回處理Response信息

github代碼

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容