初始导入

zm

2020-05-18 a18bfacbf56b401f6e0fdae8710fbca4df8cff77

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
package com.codingapi.tx.motan.balance;
 
import com.weibo.api.motan.cluster.loadbalance.ConsistentHashLoadBalance;
import com.weibo.api.motan.common.MotanConstants;
import com.weibo.api.motan.core.extension.Activation;
import com.weibo.api.motan.core.extension.SpiMeta;
import com.weibo.api.motan.rpc.Referer;
import com.weibo.api.motan.rpc.Request;
 
import java.util.List;
 
/**
 * <p>一致性 Hash，相同参数的请求总是发到同一提供者</p>
 *
 * @author 张峰 zfvip_it@163.com
 *  2017/12/1 10:44
 */
@SpiMeta(name = "consistentLcn")
@Activation(key = {MotanConstants.NODE_TYPE_SERVICE, MotanConstants.NODE_TYPE_REFERER})
public class ConsistentHashLoadBalanceProxy extends ConsistentHashLoadBalance {
 
    private LCNBalanceProxy lcnBalanceProxy = new LCNBalanceProxy();
 
    @Override
    protected Referer doSelect(Request request) {
        return lcnBalanceProxy.proxy(getReferers(),super.doSelect(request));
    }
 
    @Override
    protected void doSelectToHolder(Request request, List refersHolder) {
        super.doSelectToHolder(request, refersHolder);
        refersHolder.set(0, lcnBalanceProxy.proxy(getReferers(),super.doSelect(request)));
    }
}