最近项目需要自定义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";
}
}
即可获取服务在注册中心的服务列表。