2025年3月25日 星期二 甲辰(龙)年 月廿四 设为首页 加入收藏
rss
您当前的位置:首页 > 计算机 > 编程开发 > Spring Cloud

Spring Cloud Ribbon自定义负载均衡策略

时间:03-13来源:作者:点击数:89

通过实现 IRule 接口可以自定义负载策略,主要的选择服务逻辑在 choose 方法中。我们这边只是演示怎么自定义负载策略,所以没写选择的逻辑,直接返回服务列表中第一个服务。具体代码如下所示。

  • public class MyRule implements IRule {
  • private ILoadBalancer lb;
  • @Override
  • public Server choose(Object key) {
  • List<Server> servers = lb.getAllServers();
  • for (Server server : servers) {
  • System.out.println(server.getHostPort());
  • }
  • return servers.get(0);
  • }
  • @Override
  • public void setLoadBalancer(ILoadBalancer lb) {
  • this.lb = lb;
  • }
  • @Override
  • public ILoadBalancer getLoadBalancer() {
  • return lb;
  • }
  • }

Spring Cloud 中,可通过配置的方式使用自定义的负载策略,ribbon-config-demo 是调用的服务名称。

ribbon-config-demo.ribbon.NFLoadBalancerRuleClassName=net.biancheng.ribbon_eureka_demo.rule.MyRule

重启服务,访问调用了其他服务的接口,可以看到控制台的输出信息中已经有了我们自定义策略中输出的服务信息,并且每次都是调用第一个服务。这跟我们的逻辑是相匹配的。

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门