/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.
 */

using System;
using Apache.NMS.ActiveMQ.Commands;
using Apache.NMS.Util;

namespace Apache.NMS.ActiveMQ.State
{
	public class SessionState
	{
	    readonly SessionInfo info;

		private readonly AtomicDictionary<ProducerId, ProducerState> producers = 
			new AtomicDictionary<ProducerId, ProducerState>();
		private readonly AtomicDictionary<ConsumerId, ConsumerState> consumers = 
			new AtomicDictionary<ConsumerId, ConsumerState>();
		private readonly Atomic<bool> isShutdown = new Atomic<bool>(false);

		public SessionState(SessionInfo info)
		{
			this.info = info;
		}

		public override String ToString()
		{
			return info.ToString();
		}

		public void AddProducer(ProducerInfo info)
		{
			CheckShutdown();
			ProducerState producerState = new ProducerState(info);

			if(producers.ContainsKey(info.ProducerId))
			{
				producers[info.ProducerId] = producerState;
			}
			else
			{
				producers.Add(info.ProducerId, producerState);
			}
		}

		public ProducerState RemoveProducer(ProducerId id)
		{
			CheckShutdown();
			ProducerState ret = null;

			if(producers.TryGetValue(id, out ret))
			{
				producers.Remove(id);
				if(null != ret && ret.TransactionState != null)
				{
					ret.TransactionState.AddProducer(ret);
				}
			}

			return ret;
		}

		public void AddConsumer(ConsumerInfo info)
		{
			CheckShutdown();
			ConsumerState consumerState = new ConsumerState(info);

			if(consumers.ContainsKey(info.ConsumerId))
			{
				consumers.Add(info.ConsumerId, consumerState);
			}
			else
			{
				consumers.Add(info.ConsumerId, consumerState);
			}
		}

		public ConsumerState RemoveConsumer(ConsumerId id)
		{
			CheckShutdown();
			ConsumerState ret = null;

			if(consumers.TryGetValue(id, out ret))
			{
				consumers.Remove(id);
			}

			return ret;
		}

		public SessionInfo Info
		{
            get { return info; }
		}

		public AtomicCollection<ConsumerId> ConsumerIds
		{
			get { return consumers.Keys; }
		}

		public AtomicCollection<ProducerId> ProducerIds
		{
			get { return producers.Keys; }
		}

		public AtomicCollection<ProducerState> ProducerStates
		{
			get { return producers.Values; }
		}

		public ProducerState getProducerState(ProducerId producerId)
		{
			return producers[producerId];
		}

		public ProducerState this[ProducerId producerId]
		{
			get { return producers[producerId]; }
		}

		public AtomicCollection<ConsumerState> ConsumerStates
		{
			get { return consumers.Values; }
		}

		public ConsumerState getConsumerState(ConsumerId consumerId)
		{
			return consumers[consumerId];
		}

		public ConsumerState this[ConsumerId consumerId]
		{
            get { return consumers[consumerId]; }
		}

		private void CheckShutdown()
		{
			if(isShutdown.Value)
			{
				throw new ApplicationException("Disposed");
			}
		}

		public void Shutdown()
		{
			isShutdown.Value = true;
			producers.Clear();
			consumers.Clear();
		}
	}
}