Spring Cloud 学习笔记三：搭建微服务工程之Ribbon 自定义负载均衡策略

目录

Ribbon 自定义负载均衡策略

---

Ribbon 自定义负载均衡策略

通过实现 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 中，可通过配置的方式使用自定义的负载策略，blog-user-service 是调用的服务名称。

blog-user-service:
  ribbon:
    NFLoadBalancerRuleClassName: cn.com.stary1993.blog.rule.MyRule

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