eureka实现黑名单功能

eureka类图结构关系如下：

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 查看注册情况