/*
* RED5 Open Source Flash Server - http://code.google.com/p/red5/
*
* Copyright 2006-2014 by respective authors (see below). All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.red5.server;
import java.lang.management.ManagementFactory;
import java.util.Collection;
import java.util.Collections;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
import java.util.concurrent.atomic.AtomicInteger;
import javax.management.MBeanServer;
import javax.management.ObjectName;
import javax.management.StandardMBean;
import org.apache.commons.lang3.StringUtils;
import org.red5.server.api.IClient;
import org.red5.server.api.IClientRegistry;
import org.red5.server.exception.ClientNotFoundException;
import org.red5.server.exception.ClientRejectedException;
import org.red5.server.jmx.mxbeans.ClientRegistryMXBean;
import org.springframework.jmx.export.annotation.ManagedResource;
/**
* Registry for clients, wherein clients are mapped by their id.
*
* @author The Red5 Project
*/
@ManagedResource(objectName="org.red5.server:type=ClientRegistry,name=default", description="ClientRegistry")
public class ClientRegistry implements IClientRegistry, ClientRegistryMXBean {
/**
* Clients map
*/
private ConcurrentMap<String, IClient> clients = new ConcurrentHashMap<String, IClient>(6, 0.9f, 2);
/**
* Next client id
*/
private AtomicInteger nextId = new AtomicInteger(0);
/**
* The identifier for this client registry
*/
private String name;
public ClientRegistry() {
}
//allows for setting a "name" to be used with jmx for lookup
public ClientRegistry(String name) {
this.name = name;
if (StringUtils.isNotBlank(this.name)) {
try {
MBeanServer mbeanServer = ManagementFactory.getPlatformMBeanServer();
ObjectName oName = new ObjectName("org.red5.server:type=ClientRegistry,name=" + name);
mbeanServer.registerMBean(new StandardMBean(this, ClientRegistryMXBean.class, true), oName);
} catch (Exception e) {
//log.warn("Error on jmx registration", e);
}
}
}
/**
* Add client to registry
* @param client Client to add
*/
public void addClient(IClient client) {
addClient(client.getId(), client);
}
/**
* Add the client to the registry
*/
private void addClient(String id, IClient client) {
//check to see if the id already exists first
if (!hasClient(id)) {
clients.put(id, client);
} else {
// DW the Client object is meant to be unifying connections from a remote user. But currently the only case we
// specify this currently is when we use a remoting session. So we actually just create an arbitrary id, which means
// RTMP connections from same user are not combined.
//get the next available client id
String newId = nextId();
//update the client
client.setId(newId);
//add the client to the list
addClient(newId, client);
}
}
public Client getClient(String id) throws ClientNotFoundException {
Client result = (Client) clients.get(id);
if (result == null) {
throw new ClientNotFoundException(id);
}
return result;
}
/**
* Returns a list of Clients.
*/
public ClientList<Client> getClientList() {
ClientList<Client> list = new ClientList<Client>();
for (IClient c : clients.values()) {
list.add((Client) c);
}
return list;
}
/**
* Check if client registry contains clients.
*
* @return <code>True</code> if clients exist, otherwise <code>False</code>
*/
protected boolean hasClients() {
return !clients.isEmpty();
}
/**
* Return collection of clients
* @return Collection of clients
*/
@SuppressWarnings("unchecked")
protected Collection<IClient> getClients() {
if (!hasClients()) {
// Avoid creating new Collection object if no clients exist.
return Collections.EMPTY_SET;
}
return Collections.unmodifiableCollection(clients.values());
}
/**
* Check whether registry has client with given id
*
* @param id Client id
* @return true if client with given id was register with this registry, false otherwise
*/
public boolean hasClient(String id) {
if (id == null) {
// null ids are not supported
return false;
}
return clients.containsKey(id);
}
/**
* Return client by id
*
* @param id Client id
* @return Client object associated with given id
* @throws ClientNotFoundException if we can't find client
*/
public IClient lookupClient(String id) throws ClientNotFoundException {
return getClient(id);
}
/**
* Return client from next id with given params
*
* @param params Client params
* @return Client object
* @throws ClientNotFoundException if client not found
* @throws ClientRejectedException if client rejected
*/
public IClient newClient(Object[] params) throws ClientNotFoundException, ClientRejectedException {
// derive client id from the connection params or use next
String id = nextId();
IClient client = new Client(id, this);
addClient(id, client);
return client;
}
/**
* Return next client id
* @return Next client id
*/
public String nextId() {
//when we reach max int, reset to zero
if (nextId.get() == Integer.MAX_VALUE) {
nextId.set(0);
}
return String.format("%s", nextId.getAndIncrement());
}
/**
* Return previous client id
* @return Previous client id
*/
public String previousId() {
return String.format("%s", nextId.get());
}
/**
* Removes client from registry
* @param client Client to remove
*/
protected void removeClient(IClient client) {
clients.remove(client.getId());
}
}