为什么要扩展流媒体服务器（kurento）
我们从openvidu官网上拉下代码,启动后，默认是一个openvidu服务器对一个kurento服务器。分别去处理信令和流媒体。根据官方文档所指的，一个session,参与者有7个的会话条件下，不同配置的服务器能够承受的压力如下图：
是的，当我们使用4c8g的服务器的时候，理论上只能处理7个session,也就是只有7个房间同时存在！这在生产上是根本不能够承受的。因为信令服务器压力本身不大，所以我们必须扩展媒体服务器（kurento），让其承受能力增加。
怎么扩展kurento
openvidu 分为CE版本和Pro版本，我们使用的是CE版本，也就是源代码里面不会将kurento的代码给我们，那我们该怎么扩展源代码，让其可以负载多个kurento呢？首先可以看一下openviduPro-kurento的架构
是的，如上图，一个openvidu服务器扩展了3台kurento服务器，由于瓶颈在kurento上，不考虑openviduserver的话，理论上我们的可承受能力会乘以3！接下来我们分析一下openvidu的源码。
KMS_URIS=["ws://116.196.10.***:8888/kurento"]io.openvidu.server.config.OpenviduConfig:
...
public List checkKmsUris() {
String property = "KMS_URIS";
return asKmsUris(property, getValue(property));
}
...
从这里可以看到，代码的入口在这里。但是如果我们在KMS_URIS里面添加多个kurento地址，他还是只会取地址里面的第一条kurento地址，也就是没有处理负载操作。原因在下面：
public class FixedOneKmsManager extends KmsManager {
@Override
public List initializeKurentoClients(List kmsProperties, boolean disconnectUponFailure) throws Exception {
KmsProperties firstProps = kmsProperties.get(0);
KurentoClient kClient = null;
Kms kms = new Kms(firstProps, loadManager);
try {
kClient = KurentoClient.create(firstProps.getUri(), this.generateKurentoConnectionListener(kms.getId()));
this.addKms(kms);
kms.setKurentoClient(kClient);
// TODO: This should be done in KurentoClient connected event
kms.setKurentoClientConnected(true);
kms.setTimeOfKurentoClientConnection(System.currentTimeMillis());
} catch (KurentoException e) {
log.error("KMS in {} is not reachable by OpenVidu Server", firstProps.getUri());
if (kClient != null) {
kClient.destroy();
}
throw new Exception();
}
return Arrays.asList(kms);
}
@Override
@PostConstruct
protected void postConstructInitKurentoClients() {
try {
List kmsProps = new ArrayList<>();
for (String kmsUri : this.openviduConfig.getKmsUris()) {
String kmsId = KmsManager.generateKmsId();
kmsProps.add(new KmsProperties(kmsId, kmsUri));
}
this.initializeKurentoClients(kmsProps, true);
} catch (Exception e) {
// Some KMS wasn't reachable
log.error("Shutting down OpenVidu Server");
System.exit(1);
}
}
} initializeKurentoClients 方法里面清楚的写了，只会取KMS_URIS的第0条数据。那如果我们想处理多条，这里就需要返回一个集合，很简单，我们定义一个自己的KmsManager类,重写initializeKurentoClients方法。如下:
public class FixedJDKmsManager extends KmsManager {
@Override
public List initializeKurentoClients(List kmsProperties, boolean disconnectUponFailure) throws Exception {
ArrayList results = new ArrayList<>();
for(KmsProperties kmsp:kmsProperties){
KurentoClient kClient = null;
Kms kms = new Kms(kmsp, loadManager);
try {
kClient = KurentoClient.create(kmsp.getUri(), this.generateKurentoConnectionListener(kms.getId()));
this.addKms(kms);
kms.setKurentoClient(kClient);
// TODO: This should be done in KurentoClient connected event
kms.setKurentoClientConnected(true);
kms.setTimeOfKurentoClientConnection(System.currentTimeMillis());
results.add(kms);
} catch (KurentoException e) {
log.error("KMS in {} is not reachable by OpenVidu Server", kmsp.getUri());
if (kClient != null) {
kClient.destroy();
}
throw new Exception();
}
}
return results;
}
@Override
@PostConstruct
protected void postConstructInitKurentoClients() {
try {
List kmsProps = new ArrayList<>();
for (String kmsUri : this.openviduConfig.getKmsUris()) {
String kmsId = KmsManager.generateKmsId();
kmsProps.add(new KmsProperties(kmsId, kmsUri));
}
this.initializeKurentoClients(kmsProps, true);
} catch (Exception e) {
// Some KMS wasn't reachable
log.error("Shutting down OpenVidu Server");
System.exit(1);
}
}
} 配置使用自己定义的kmsManager
怎么实现加权轮询负载均衡
和kms_urls同理，我们可以在配置文件添加kms_weights:
KMS_WEIGHT=[1]并且，人为的控制kms_urls的size与kms_weight同等
然后在openviduConfig里面，将其与kms_urls类似的道理初始化。核心代码如下：
public class OpenviduConfig{
...
private List kmsWeights;
public List getKmsWeights() {
return kmsWeights;
}
public void setKmsWeights(List weights) {
this.kmsWeights = weights;
log.info("=====kms权重被重置为:{}", this.kmsUrisList);
}
public List initWeight() {
String property = "KMS_WEIGHT";
return asKmsWeights(property, getValue(property));
}
protected void checkConfigurationProperties(boolean loadDotenv) {
...
kmsUrisList = checkKmsUris();
kmsWeights = initWeight();
...
}
...
}
到此，我们将自己想要的加权参数放入容器了，但是openvidu还没有选择kurento，接下来需要找到openvidu寻找对应kurento的地方，根据权值去取对应的url：
熟读源码后可以发现，在KurentoSessionManager.joinRoom方法里面，调用getLessLoadedConnectedAndRunningKms方法，这个方法就是选择kms的方法。同样的道理，我们可以重写一个方法去按照自己的逻辑选择kms我写的如下：
public synchronized Kms getLoadBalanceConnectedAndRunningKms() throws NoSuchElementException {
List kmsLoads = getKmsLoads().stream().filter(kmsLoad -> kmsLoad.kms.isKurentoClientConnected()
&& mediaNodeStatusManager.isRunning(kmsLoad.kms.getId()) ).collect(Collectors.toList());
if (kmsLoads.isEmpty()) {
throw new NoSuchElementException();
} else {
//todo 这里编写kms的负载均衡
this.openviduConfig.getKmsWeights();
// Kms kms = kmsLoads.get(Integer.parseInt(kmsNode)).kms;
//第一次:初始化数据
KmsWeightRobin.initKmsLoads(kmsLoads,this.openviduConfig.getKmsWeights());
KmsLoad kmsWeightRobin = KmsWeightRobin.getKmsWeightRobin();
log.info("=========>>>>>>加权轮询后,选择kms:{}",kmsWeightRobin.kms.getUri());
return kmsWeightRobin.getKms();
}
}
public class KmsWeightRobin {
// static Map ipMap=new HashMap<>();
static Map kmsMap=new HashMap<>();
static List kmsLoads = null;
static List weights = null;
public static synchronized void initMap() {
kmsMap.clear();
for(int i = 0;i ipServerMap=new ConcurrentHashMap<>();
ipServerMap.putAll(kmsMap);
Set ipSet=ipServerMap.keySet();
Iterator ipIterator=ipSet.iterator();
//定义一个list放所有server
ArrayList ipArrayList=new ArrayList();
//循环set，根据set中的可以去得知map中的value，给list中添加对应数字的server数量
while (ipIterator.hasNext()){
KmsManager.KmsLoad serverName=ipIterator.next();
Integer weight=ipServerMap.get(serverName);
for (int i = 0;i < weight ;i++){
ipArrayList.add(serverName);
}
}
KmsManager.KmsLoad serverName=null;
if (pos>=ipArrayList.size()){
pos=0;
}
serverName=ipArrayList.get(pos);
//轮询+1
pos ++;
return serverName;
}
// public static Boolean initMapFlag = false;
public static void initKmsLoads(List kmsLoads,List weights){
if(ObjectUtils.isEmpty(KmsWeightRobin.kmsLoads)){
KmsWeightRobin.kmsLoads = kmsLoads;
}
if(ObjectUtils.isEmpty(KmsWeightRobin.weights)){
KmsWeightRobin.weights = weights;
}
// if(!initMapFlag){
initMap();
// initMapFlag = true;
// }
}
public static void updateKmsLoads(List kmsLoads,List weights){
return;
}
} 这样，在选择kms的时候，就可以让容器按照weight去选择了。