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.效果:控制台会打印如下日志:后面显示的替换的本地微服务

相关推荐
南 阳1 分钟前
从微服务到AI服务:Nacos 3.0如何重构下一代动态治理体系?
人工智能·微服务·云原生·重构
pjx9871 小时前
质量的“试金石”:精通Spring Boot单元测试与集成测试
spring boot·spring·单元测试·集成测试
幼儿园口算大王2 小时前
Spring反射机制
java·spring·反射
MrWho不迷糊3 小时前
企业级权限系统怎么设计四 —— ABAC模型统一功能权限与数据权限
后端·微服务
Howard_Stark3 小时前
Spring生命周期
spring
计算机毕设定制辅导-无忧学长4 小时前
Spring 与 ActiveMQ 的深度集成实践(二)
spring·activemq·java-activemq
溪i4 小时前
react-spring/web + children not defined
前端·spring·react.js
带刺的坐椅5 小时前
qwen3 惊喜发布,用 ollama + solon ai (java) 尝个鲜
java·spring·solon·solon-ai·qwen3
程序媛学姐6 小时前
Spring Cloud Kubernetes:在K8s环境中的服务发现与配置
spring cloud·kubernetes·服务发现
kill bert6 小时前
第33周JavaSpringCloud微服务 分布式综合应用
微服务·云原生·架构