org.kurento.room.kms.KmsManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kurento-room-server Show documentation
Kurento Room Server
There is a newer version: 6.6.0
/*
 * (C) Copyright 2015 Kurento (http://kurento.org/)
 * 
 * All rights reserved. This program and the accompanying materials are made
 * available under the terms of the GNU Lesser General Public License (LGPL)
 * version 2.1 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl-2.1.html
 * 
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package org.kurento.room.kms;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;

import org.kurento.client.KurentoClient;
import org.kurento.room.api.KurentoClientProvider;
import org.kurento.room.api.KurentoClientSessionInfo;
import org.kurento.room.exception.RoomException;
import org.kurento.room.exception.RoomException.Code;
import org.kurento.room.internal.DefaultKurentoClientSessionInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class KmsManager implements KurentoClientProvider {

	public static class KmsLoad implements Comparable {

		private Kms kms;
		private double load;

		public KmsLoad(Kms kms, double load) {
			this.kms = kms;
			this.load = load;
		}

		public Kms getKms() {
			return kms;
		}

		public double getLoad() {
			return load;
		}

		@Override
		public int compareTo(KmsLoad o) {
			return Double.compare(this.load, o.load);
		}
	}

	private final Logger log = LoggerFactory.getLogger(KmsManager.class);

	private List kmss = new ArrayList();
	private Iterator usageIterator = null;

	@Override
	public KurentoClient getKurentoClient(KurentoClientSessionInfo sessionInfo)
			throws RoomException {
		if (!(sessionInfo instanceof DefaultKurentoClientSessionInfo))
			throw new RoomException(Code.GENERIC_ERROR_CODE,
					"Unkown session info bean type (expected "
							+ DefaultKurentoClientSessionInfo.class.getName()
							+ ")");
		return getKms((DefaultKurentoClientSessionInfo) sessionInfo)
				.getKurentoClient();
	}

	/**
	 * Returns a {@link Kms} using a round-robin strategy.
	 * 
	 * @param sessionInfo session's id
	 */
	public synchronized Kms getKms(DefaultKurentoClientSessionInfo sessionInfo) {
		if (usageIterator == null || !usageIterator.hasNext())
			usageIterator = kmss.iterator();
		return usageIterator.next();
	}

	public synchronized void addKms(Kms kms) {
		this.kmss.add(kms);
	}

	public synchronized Kms getLessLoadedKms() {
		return Collections.min(getKmsLoads()).kms;
	}

	public synchronized Kms getNextLessLoadedKms() {
		List sortedLoads = getKmssSortedByLoad();
		if (sortedLoads.size() > 1)
			return sortedLoads.get(1).kms;
		else
			return sortedLoads.get(0).kms;
	}

	public synchronized List getKmssSortedByLoad() {
		List kmsLoads = getKmsLoads();
		Collections.sort(kmsLoads);
		return kmsLoads;
	}

	private List getKmsLoads() {
		ArrayList kmsLoads = new ArrayList<>();
		for (Kms kms : kmss) {
			double load = kms.getLoad();
			kmsLoads.add(new KmsLoad(kms, load));
			log.trace("Calc load {} for kms: {}", load, kms.getUri());
		}
		return kmsLoads;
	}
}