openvidu-server 搭建 kurento 负载均衡机制

为什么要扩展流媒体服务器（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<String> checkKmsUris() {

        String property = "KMS_URIS";

        return asKmsUris(property, getValue(property));

  }
...

从这里可以看到，代码的入口在这里。但是如果我们在 KMS_URIS 里面添加多个 kurento 地址，他还是只会取地址里面的第一条 kurento 地址，也就是没有处理负载操作。原因在下面：


public class FixedOneKmsManager extends KmsManager {

	@Override
	public List<Kms> initializeKurentoClients(List<KmsProperties> 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<KmsProperties> 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<Kms> initializeKurentoClients(List<KmsProperties> kmsProperties, boolean disconnectUponFailure) throws Exception {

		ArrayList<Kms> 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<KmsProperties> 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<Integer> kmsWeights;


    public List<Integer> getKmsWeights() {
        return kmsWeights;
    }

    public void setKmsWeights(List<Integer> weights) {
        this.kmsWeights = weights;
        log.info("=====kms权重被重置为:{}", this.kmsUrisList);
    }

    public List<Integer> 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<KmsLoad> 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<String,Integer> ipMap=new HashMap<>();

    static Map<KmsManager.KmsLoad,Integer> kmsMap=new HashMap<>();

    static List<KmsManager.KmsLoad> kmsLoads = null;

    static List<Integer> weights = null;

    public  static synchronized void initMap() {

        kmsMap.clear();

        for(int i = 0;i<kmsLoads.size();i++){
            kmsMap.put(kmsLoads.get(i),weights.get(i));
        }
    }



    static Integer pos=0;
    public static KmsManager.KmsLoad getKmsWeightRobin(){
        Map<KmsManager.KmsLoad,Integer> ipServerMap=new ConcurrentHashMap<>();
        ipServerMap.putAll(kmsMap);

        Set<KmsManager.KmsLoad> ipSet=ipServerMap.keySet();
        Iterator<KmsManager.KmsLoad> ipIterator=ipSet.iterator();

        //定义一个list放所有server
        ArrayList<KmsManager.KmsLoad> ipArrayList=new ArrayList<KmsManager.KmsLoad>();

        //循环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<KmsManager.KmsLoad> kmsLoads,List<Integer> 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<KmsManager.KmsLoad> kmsLoads,List<Integer> weights){


        return;
    }


}

这样，在选择 kms 的时候，就可以让容器按照 weight 去选择了。

待解决或者改进

目前的负载方案，使用了一个 openvidu 多个 kms.但是对于 openvidu 还是单点的，我曾将想过将 openvidu 做高可用（多点）。但是目前 openvidu 的很多信息都是存储在 jvm 的内存里面（比如 session 等信息）。当然也有内存共享等可以解决，但是目前还没有做 openvidu 的多点方案。

openvidu-server 搭建 kurento 负载均衡机制

为什么要扩展流媒体服务器（kurento）

怎么扩展 kurento

怎么实现加权轮询负载均衡

待解决或者改进

相关帖子

基于洋葱打造代理池或者增强你的上网匿名性

nginx，一个高效的负载均衡应该是什么样的？

负载均衡的多种算法总结

Nginx 负载均衡算法

Ribbon 负载均衡使用详解

负载均衡与环境搭建脚本很配噢

思源笔记使用经验分享（一）

欢迎来到这里！