【转】接口重构一些思路-dubbo负载均衡扩展接口重构实例

转于自己在公司的Blog:

http://pt.alibaba-inc.com/wp/experience_1003/loadbalance_refactor.html

项目中的一个重构的过程及理由,用于知会团队成员,在这里备一个。

RPC远程调用框架中有很多可选的负载均衡策略,

比如:随机,轮循,最少连接等等,

这个时候就需要一个SPI扩展点,为后续增加新的策略提供可能,

重构前:

原接口形式,如下:

1.public class LoadBalance<T> {  
2.  
3.    // 给定资源和权重,返回选中资源的下标号  
4.    int select(T[] resources, int[] weights);  
5.  
6.}

问题一:

返回下标号,使接口输入输出不一致,并且限制了资源的包装,如:

Java代码 复制代码 收藏代码
1.public class LoadBalance {  
2.  
3.    // 给定资源和权重,返回选中资源  
4.    <T> T select(T[] resources, int[] weights);  
5.  
6.}

问题二:

作为扩展接口,即然使用泛型,表示策略实现不限制资源类型,接口本身定义泛型没有意义,泛型声明应该定义在方法上,如:

Java代码 复制代码 收藏代码
1.public class LoadBalance {  
2.  
3.    // 给定资源和权重,返回选中资源  
4.    <T> T select(T[] resources, int[] weights);  
5.  
6.}

问题三:

作为扩展接口,权重信息的传递过于特殊化,

比如现在最小连接数策略要用到当前活跃连接数,

需增加新的参数actives,如:

Java代码 复制代码 收藏代码
1.public class LoadBalance {  
2.  
3.    // 给定资源和权重,返回选中资源  
4.    <T> T select(T[] resources, int[] weights, int[] actives);  
5.  
6.}

问题四:

但像上面这样,你并不清楚后续还有什么参数要加入,接口的契约不容扩展,

另一种办法是,在最小连接数策略实现中将T强制转型成RpcInvoker接口,然后调用getActive(),

但即然active数通过get方获取,为什么weight却通过另一个参数传入,明显的不一致,

而且这里的强制转型,也会导致策略的实现并不能像原来期望的通用,

作为一个框架的扩展点,通用意义并不大,越通用越难用,上面的泛型T有过度设计之嫌,

直接用RpcInvoker作为参数,更能保证契约的完备性,如:

Java代码 复制代码 收藏代码
1.public class LoadBalance {  
2.  
3.    // RpcInvoker接口中有getWeight(), getActive()等获取参数方法  
4.    // 给定资源和权重,返回选中资源  
5.    RpcInvoker select(RpcInvoker[] resources);  
6.  
7.}

如果有通用性需求,也可以考虑再抽取一个接口,如:

Java代码 复制代码 收藏代码
1.public class LoadBalance {  
2.  
3.    // 给定资源和权重,返回选中资源  
4.    Selectable select(Selectable[] resources);  
5.  
6.}

问题五:

有一种需求是基于客户端一致性的,要求执行所有RpcInvoker,而不是选其中一个RpcInvoker执行,

原有实现,是将其作为特例,写死在代码中的,基于上面的LoadBalance接口,

完全可以将传入的所有RpcInvoker[]包装成一个总的RpcInvoker,里面用for循环委派所有调用。

如果有w+r>n(写节点+读节点>总节点)的一致性需求,也可以用相应方法处理,只是增加一些配置项,如:

Java代码 复制代码 收藏代码
1.public class AllLoadBalance {  
2.  
3.    public RpcInvoker select(RpcInvoker[] resources) {  
4.        return new AllRpcInvoker(resources);  
5.    }  
6.  
7.}  
8.class AllRpcInvoker implements RpcInvoker {  
9.  
10.    private RpcInvoker[] invokers;  
11.      
12.    public AllRpcInvoker(RpcInvoker[] invokers) {  
13.        this.invokers = invokers;  
14.    }  
15.      
16.    // Delegate all methods to invokers  
17.      
18.}

问题六:

有的策略实现是带状态的,比如轮循策略需记录轮循序号,

也就是并不能单实例使用LoadBalance实例,

这对框架的维护非常不利,容易给后来的维护者埋下地雷,

而且,同一个resources集合,必需用同一个LoadBalance实例,

也就是LoadBalance实例的变更是跟随resources集合的变更,

即然如此,资源集合的设定,可以在select()之前确定,如:

1.public class LoadBalance {  
2.  
3.    // 初始化资源集合  
4.    void init(RpcInvoker[] resources);  
5.  
6.    // 返回选中资源  
7.    RpcInvoker select();  
8.  
9.}

这样的好处是,LoadBalance的实现可以在init()时做预处理及缓存,

比如随机策略,需要统计总权重,如果在init()方法中统计,

select()时可以减少一次for循环,

而且,可以通过重复调用init()方法,复用单一LoadBalance实例,

当然,LoadBalance的实现需确保线程安全性。

------------------------

Dubbo设计分享系列:

一些设计上的基本常识

谈谈泛化式扩展与组合式扩展

相关推荐