springcloud feign本地微服务之间的相互调用

一、随着项目微服务数量越来越多,本地调试的时候很容易把请求打到别人的机器上或者云上环境,所以用自定义注解的方式实现微服务本地之间调用,使用起来也非常简单,只需要在启动类上加一个注解即可。

1.编写自定义注解

java 复制代码
import org.springframework.context.annotation.Import;

import java.lang.annotation.*;

@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.TYPE})
@Documented
@Import(FeignClientsServiceNameAppendBeanPostProcessor.class)
public @interface EnableLocalHostFeign {
    String[] serverNameAndUrl() default {};

}

2:加一个规则配置类,在bean交给spring管理之前更改url

java 复制代码
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.scheduling.annotation.Async;
import org.springframework.util.ReflectionUtils;

import java.lang.reflect.Field;
import java.util.Objects;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.regex.Pattern;

@Slf4j
public class FeignClientsServiceNameAppendBeanPostProcessor implements BeanPostProcessor, ApplicationContextAware {

    private ApplicationContext applicationContext;


    private AtomicInteger atomicInteger = new AtomicInteger();
    private static final Pattern pattern = Pattern
            .compile("^([hH][tT]{2}[pP]://|[hH][tT]{2}[pP][sS]://)(([A-Za-z0-9-~]+).)+([A-Za-z0-9-~\\/])+$");

    @SneakyThrows
    @Override
    @Async
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (atomicInteger.getAndIncrement() == 0) {
            String[] serverNameAndUrl = null;
            String[] beanNamesForAnnotation = applicationContext.getBeanNamesForAnnotation(EnableLocalHostFeign.class);
            for (String s : beanNamesForAnnotation) {
                if (s.contains(".")) {
                    continue;
                }
                Class<?> aClass1 = applicationContext.getBean(s).getClass();
                EnableLocalHostFeign annotation1 = AnnotationUtils.findAnnotation(aClass1, EnableLocalHostFeign.class);
                serverNameAndUrl = annotation1.serverNameAndUrl();
            }
            if (serverNameAndUrl == null || serverNameAndUrl.length == 0) {
                log.info("Annotation not found EnableLocalHostFeign");
                return null;
            }
            String beanNameOfFeignClientFactoryBean = "org.springframework.cloud.openfeign.FeignClientFactoryBean";
            Class beanNameClz = Class.forName(beanNameOfFeignClientFactoryBean);
            String[] finalserverNameAndUrl = serverNameAndUrl;
            applicationContext.getBeansOfType(beanNameClz).forEach((feignBeanName, beanOfFeignClientFactoryBean) -> {
                try {
                    for (String server : finalserverNameAndUrl) {
                        if (!server.contains("=")) {
                            log.info("EnableLocalHostFeign serverNameAndUrl error,skip");
                            continue;
                        }
                        int i = server.indexOf("=");
                        String finalServerName = server.substring(0, i).replaceAll("\\s*", "");
                        String finalServerUrl = server.substring(i + 1, server.length()).replaceAll("\\s*", "");

                        if (!pattern.matcher(finalServerUrl).matches()) {
                            log.info("EnableLocalHostFeign url error,skip");
                            continue;
                        }
                        String serverName = getFiled(beanNameClz, "name", beanOfFeignClientFactoryBean);
                        if (serverName.equals(finalServerName)) {
                            setField(beanNameClz, "url", beanOfFeignClientFactoryBean, finalServerUrl);
                            log.info("EnableLocalHostFeign : " + feignBeanName + "-->" + beanOfFeignClientFactoryBean);
                        }
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
        }
        return null;
    }

    private String getFiled(Class clazz, String fieldName, Object obj) throws Exception {
        Field field = ReflectionUtils.findField(clazz, fieldName);
        if (Objects.nonNull(field)) {
            ReflectionUtils.makeAccessible(field);
            Object value = field.get(obj);
            if (Objects.nonNull(value)) {
                return value.toString();
            }
        }
        return null;
    }

    private void setField(Class clazz, String fieldName, Object obj, String url) throws Exception {
        Field field = ReflectionUtils.findField(clazz, fieldName);
        if (Objects.nonNull(field)) {
            ReflectionUtils.makeAccessible(field);
            ReflectionUtils.setField(field, obj, url);
        }
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }

    public static void main(String[] args) {
        String url1 = "http://www.xx.com";
        String url2 = "w.xx.com";
        String url3 = "http://w.xx.com";
        String url4 = "ssss";
        String url5 = "http://localhost:9012";
        Pattern pattern = Pattern
                .compile("^([hH][tT]{2}[pP]://|[hH][tT]{2}[pP][sS]://)(([A-Za-z0-9-~]+).)+([A-Za-z0-9-~\\/])+$");
        System.out.println(pattern.matcher(url1).matches());
        System.out.println(pattern.matcher(url2).matches());
        System.out.println(pattern.matcher(url3).matches());
        System.out.println(pattern.matcher(url4).matches());
        System.out.println(pattern.matcher(url5).matches());

    }
}

3.使用方法:springboot项目启动类添加@EnableLocalHostFeign注解,参数传值规则:项目名=项目地址。

java 复制代码
@EnableDiscoveryClient
@EnableFeignClients(basePackages = {"xxx"})
@SpringBootApplication(scanBasePackages = {"xxx"},
        exclude= {DataSourceAutoConfiguration.class})
@EnableLocalHostFeign(serverNameAndUrl ={ "项目名=项目地址",
        "hello-world-server = http://localhost:9006","hello-world-server2=http://192.168.1.8:9001"})
public class xxxxxApplication {
    public static void main(String[] args) {
        SpringApplication.run(xxxxxApplication.class, args);
    }

}

4.效果:控制台会打印如下日志:后面显示的替换的本地微服务

相关推荐
Yharim1 小时前
Spring框架 - SpringCloud
spring cloud·面试
Foyo Designer2 小时前
【 <二> 丹方改良:Spring 时代的 JavaWeb】之 Spring Boot 的未来:从微服务到云原生的演进
java·spring boot·redis·spring·微服务·云原生·restful
Java水解3 小时前
Spring 核心技术解析【纯干货版】- XVI:Spring 网络模块 Spring-WebMvc 模块精讲
后端·spring
阿里云云原生3 小时前
Spring项目开发的智能助手:通义灵码使用指南
spring
快乐的木子李4 小时前
Java spring mybatis面试题(200道),八股文
java·开发语言·spring·mybatis
爱的叹息7 小时前
Spring MVC 逻辑视图(JSP、Thymeleaf、FreeMarker)与非逻辑视图(JSON、Excel、PDF、XML)详解及示例
java·spring·mvc
Dmmmmmmm7 小时前
基于 React + Umi + Qiankun 的微前端架构实践
微服务
我要学编程(ಥ_ಥ)8 小时前
初始JavaEE篇 —— SpringBoot 统一功能处理
java·spring boot·后端·spring·java-ee
都叫我大帅哥8 小时前
Spring IoC源码解析:原来你是这样的"快递仓库管理员"
java·spring·源码阅读
流浪书生gzs8 小时前
docker镜像导出导入
java·spring cloud·eureka