eureka实现黑名单功能

我就是我 2023-06-12 05:22 18阅读 0赞

eureka类图结构关系如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70][]

1 重新定义注册类PeerAwareInstanceRegistryImpl

package com.cyb.config;
    
    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.stereotype.Component;
    
    import com.netflix.discovery.EurekaClient;
    import com.netflix.discovery.EurekaClientConfig;
    import com.netflix.eureka.EurekaServerConfig;
    import com.netflix.eureka.resources.ServerCodecs;
    
    @Component
    public class MyFactory implements BeanPostProcessor,ApplicationContextAware{
    
    
    	public ApplicationContext ac;
    	
    	@Override
    	public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
    		this.ac = applicationContext;
    	}
    	@Override
    	public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
    		if("peerAwareInstanceRegistry".equals(beanName)){//bean.getClass().toString().contains("InstanceRegistry")
    			System.out.println(bean+","+beanName);
    			MyInstanceRegistryDecorator mird = new MyInstanceRegistryDecorator(
    					ac.getBean(EurekaServerConfig.class), 
    					ac.getBean(EurekaClientConfig.class),  
    					ac.getBean(ServerCodecs.class), 
    					ac.getBean(EurekaClient.class));
    			mird.setSetting(ac.getBean(ClientBlackSetting.class));
    			mird.setApplicationContext(ac);
    			return mird;
    		}
    		/*if(beanName.contains("person")){
    			Person p = (Person)bean;
    			System.out.println("before: "+p.getName()+","+bean);
    			Person newBean = new Person();
    			newBean.setName("chenyuanbao");
    			bean = newBean;
    		}*/
    		return bean;
    	}
    	@Override
    	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
    		/*if(beanName.contains("person")){
    			Person p = (Person)bean;
    			System.out.println("after: "+p.getName()+","+bean);
    		}*/
    		return bean;
    	}
    }

2 定义自己的装饰模式的注册类

package com.kiiik.config;
    
    import java.util.regex.Matcher;
    import java.util.regex.Pattern;
    
    import org.springframework.cloud.netflix.eureka.server.InstanceRegistry;
    import org.springframework.core.env.Environment;
    
    import com.netflix.appinfo.InstanceInfo;
    import com.netflix.discovery.EurekaClient;
    import com.netflix.discovery.EurekaClientConfig;
    import com.netflix.eureka.EurekaServerConfig;
    import com.netflix.eureka.resources.ServerCodecs;
    
    public class MyInstanceRegistryDecorator extends InstanceRegistry {
    	private ClientBlackSetting setting;
    
    	public MyInstanceRegistryDecorator(EurekaServerConfig serverConfig, EurekaClientConfig clientConfig,
    			ServerCodecs serverCodecs, EurekaClient eurekaClient
    	/* int expectedNumberOfRenewsPerMin,int defaultOpenForTrafficCount */) {
    		super(serverConfig, clientConfig, serverCodecs, eurekaClient, 1, 1);
    	}
    
    	public void setSetting(ClientBlackSetting setting) {
    		this.setting = setting;
    	}
    
    	@Override
    	public void register(InstanceInfo info, int leaseDuration, boolean isReplication) {
    		System.out.println("============my=============");
    		if (checkRegister(info)) {
    			super.register(info, leaseDuration, isReplication);
    		}
    
    	}
    
    	@Override
    	public void register(final InstanceInfo info, final boolean isReplication) {
    		System.out.println("============my=============");
    		if (checkRegister(info)) {
    			super.register(info, isReplication);
    		}
    	}
    
    	// pub 方法
    	public boolean checkRegister(InstanceInfo clientInstance) {
    		boolean allow = true;
    		if(setting.isOpen()){//开启黑名单功能
    			if (setting.getIps().contains(clientInstance.getIPAddr())) {
    				StringBuffer sb = new StringBuffer().append("客户端[" + clientInstance.getAppName())
    						.append("," + clientInstance.getInstanceId()).append("," + clientInstance.getIPAddr())
    						.append("," + clientInstance.getPort() + "被拒绝注册！");
    				System.err.println(sb.toString());
    				allow =  false;//不允许注册
    			}
    			if(allow){//查看匹配规则
    				for(String p:setting.getPattern()){
    					Pattern pattern = Pattern.compile(p);
    					Matcher matcher = pattern.matcher(clientInstance.getIPAddr());
    					if(matcher.find()){
    						allow = false;
    						break;//中断校验
    					}
    				}
    			}else{
    			}
    		}else{
    		}
    		return allow;
    		/*
    		 * if(){//正则表达式校验
    		 * 
    		 * }
    		 */
    		// 判断配置文件是否开启eureka黑白名单的过滤机制,默认开启
    		// Boolean registerControlEnabled =
    		// isRegisterControlEnabled(environment);
    		// registration.getEurekaClient().getEurekaClientConfig().
    		/*StringBuffer sb = new StringBuffer().append("," + clientInstance.getAppName())
    				.append("," + clientInstance.getInstanceId()).append("," + clientInstance.getIPAddr())
    				.append("," + clientInstance.getPort());
    		System.err.println(sb.toString());*/
    		// return true;
    
    	}
    
    	public static Boolean isRegisterControlEnabled(Environment environment) {
    		return environment.getProperty("spring.application.register.control.enabled", Boolean.class, Boolean.TRUE);
    	}
    }

3 查看注册情况

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 1][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 2][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 3][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70]: /images/20230531/7904528622954fd5a9276757c46c59e3.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 1]: /images/20230531/061352401a7546df961f0a24d36824c7.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 2]: /images/20230531/2b260f3fd2f3428cb91c3f91cf364b92.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3p6dWNoZW55Yg_size_16_color_FFFFFF_t_70 3]: /images/20230531/d92ea010212741ff8f4f192c111cfecd.png