com.generallycloud.baseio.container.rtp.server.RTPRoom Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of baseio-all Show documentation
The all in one project of baseio
There is a newer version: 3.2.9-BETA-2
/*
 * Copyright 2015-2017 GenerallyCloud.com
 *  
 * 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 com.generallycloud.baseio.container.rtp.server;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.locks.ReentrantLock;

import com.generallycloud.baseio.common.Logger;
import com.generallycloud.baseio.common.LoggerFactory;
import com.generallycloud.baseio.component.DatagramSession;
import com.generallycloud.baseio.component.Session;
import com.generallycloud.baseio.concurrent.FixedAtomicInteger;
import com.generallycloud.baseio.concurrent.ReentrantList;
import com.generallycloud.baseio.container.rtp.RTPContext;
import com.generallycloud.baseio.protocol.DatagramPacket;

//FIXME 是不是要限制最多room数
public class RTPRoom {

	private static final FixedAtomicInteger			ROOM_ID = new FixedAtomicInteger();
	private static final Logger		logger		= LoggerFactory.getLogger(RTPRoom.class);

	private RTPContext				context		;
	private ReentrantList	datagramChannelList	= new ReentrantList<>(new ArrayList<>());
	private RTPRoomFactory			roomFactory	;
	private Integer				roomID		;
	private boolean				closed		= false;

	public RTPRoom(RTPContext context, Session session) {
		this.roomID = genRoomID();
		this.roomFactory = context.getRTPRoomFactory();
		this.context = context;
//		this.join(session.getDatagramChannel()); //FIXME udp 
	}

	public void broadcast(DatagramSession session, DatagramPacket packet) {

		List datagramChannels = datagramChannelList.takeSnapshot();

		for (DatagramSession ch : datagramChannels) {

			if (session == ch) {
				continue;
			}

			try {
				ch.sendPacket(packet);
			} catch (Throwable e) {
				logger.debug(e);
			}
		}
	}

	private Integer genRoomID() {
		return ROOM_ID.getAndIncrement();
	}

	public Integer getRoomID() {
		return roomID;
	}

	public boolean join(DatagramSession session) {
		
		if (session == null) {
			return false;
		}

		ReentrantLock lock = datagramChannelList.getReentrantLock();

		lock.lock();

		if (closed) {

			lock.unlock();

			return false;
		}

		if (!datagramChannelList.add(session)) {

			lock.unlock();

			return false;
		}

		lock.unlock();

//		Session session = (Session) session.getSession();

		//FIXME RTP
//		RTPSessionAttachment attachment = (RTPSessionAttachment) session.getAttachment(context.getPluginIndex());

//		attachment.setRTPRoom(this);

		return true;
	}

	public void leave(DatagramSession channel) {

		ReentrantLock lock = datagramChannelList.getReentrantLock();

		lock.lock();

		datagramChannelList.remove(channel);

		List chs = datagramChannelList.takeSnapshot();

		for (DatagramSession ch : chs) {

			if (ch == channel) {
				continue;
			}

			//FIXME RTP
//			SocketSession session = (SocketSession) ch.getSession();
			
//			Authority authority = ApplicationContextUtil.getAuthority(session);
//
//			MapMessage message = new MapMessage("mmm", authority.getUuid());
//
//			message.setEventName("break");
//
//			message.put("userID", authority.getUserID());
//
//			MQContext mqContext = MQContext.getInstance();
//
//			mqContext.offerMessage(message);
		}

		if (datagramChannelList.size() == 0) {

			this.closed = true;

			roomFactory.removeRTPRoom(roomID);
		}

		lock.unlock();
	}

}