最近项目需要自定义Ribbon路由规则。需要获取 服务在 注册中心的注册列表

一种是继承 AbstractLoadBalancerRule 类 实现自定义规则类。其中可以得到LoadBalancer负载均衡器,从而得到服务列表

public class MyRule extends AbstractLoadBalancerRule{
	
	private static final Logger LOGGER = LoggerFactory.getLogger(MyRule.class);

	
	public Server choose( Object arg0) {
		ILoadBalancer lb = this.getLoadBalancer();
		
		List<Server> upServerList = lb.getReachableServers(); //可用服务列表
		List<Server> allServerList = lb.getAllServers(); 
		
		return null;
	}
	

	@Override
	public void initWithNiwsConfig(IClientConfig arg0) {
		// TODO Auto-generated method stub
		
	}

}

在配置文件中指定某一服务

SERVICE-HI.ribbon.NFLoadBalancerRuleClassName=com.test.eurekaRibbon.rule.MyRule

 

二是直接获取LoadBalancer,可使用Ribbon中的Bean工厂获取

@RestController
public class HelloController {
	
	@Autowired
	private SpringClientFactory factory;
	
	@RequestMapping("/testBalancer")
	public String testBalancer() {
		
	    ILoadBalancer lb =  factory.getLoadBalancer("SERVICE-HI"); //指定服务名
		List<Server> allServers = lb.getAllServers();
		List<Server> upServers = lb.getReachableServers();
	
		return "hello";
	}
	

}

即可获取服务在注册中心的服务列表。

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐