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

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

1.编写自定义注解


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


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注解,参数传值规则:项目名=项目地址。

@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.效果:控制台会打印如下日志:后面显示的替换的本地微服务
springcloud feign本地微服务之间的相互调用_第1张图片

你可能感兴趣的