using System;
using System.Messaging;
using Apache.NMS.MSMQ;
/*
 * 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.
 */

namespace Apache.NMS.MSMQ.Readers
{
    /// <summary>
    /// MSMQ message reader, returning messages matching the specified
    /// lookup identifier.
    /// </summary>
	public class ByLookupIdMessageReader : AbstractMessageReader
	{
        private Int64 lookupId;

        // The MSMQ documentation for MessageQueue.PeekByLookupId(*) and
        // MessageQueue.ReceiveByLookupId(*) specifies:
        // InvalidOperationException: The message with the specified lookupId could not be found.
 
        /// <summary>
        /// Constructor.
        /// </summary>
        /// <param name="messageQueue">The MSMQ message queue from which
        /// messages will be read.</param>
        /// <param name="messageConverter">A message converter for mapping
        /// MSMQ messages to NMS messages.</param>
        /// <param name="lookupId">The lookup identifier of the message
        /// to be read.</param>
        public ByLookupIdMessageReader(MessageQueue messageQueue,
            IMessageConverter messageConverter, Int64 lookupId)
            : base(messageQueue, messageConverter)
        {
            this.lookupId = lookupId;
        }

        /// <summary>
        /// Returns without removing (peeks) the first message in the queue
        /// referenced by this MessageQueue matching the selection criteria.
        /// The Peek method is synchronous, so it blocks the current thread
        /// until a message becomes available.
        /// </summary>
        /// <returns>Peeked message.</returns>
        public override IMessage Peek()
        {
            try
            {
                return Convert(messageQueue.PeekByLookupId(lookupId));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Returns without removing (peeks) the first message in the queue
        /// referenced by this MessageQueue matching the selection criteria.
        /// The Peek method is synchronous, so it blocks the current thread
        /// until a message becomes available or the specified time-out occurs.
        /// </summary>
        /// <param name="timeSpan">Reception time-out.</param>
        /// <returns>Peeked message.</returns>
        public override IMessage Peek(TimeSpan timeSpan)
        {
            try
            {
                // No time-out option for receiving messages by lookup identifiers:
                // either the message is present in the queue, or the method throws
                // an exception immediately if the message is not in the queue. 
                return Convert(messageQueue.PeekByLookupId(lookupId));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Receives the first message available in the queue referenced by
        /// the MessageQueue matching the selection criteria.
        /// This call is synchronous, and blocks the current thread of execution
        /// until a message is available.
        /// </summary>
        /// <returns>Received message.</returns>
        public override IMessage Receive()
        {
            try
            {
                return Convert(messageQueue.ReceiveByLookupId(lookupId));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Receives the first message available in the queue referenced by the
        /// MessageQueue matching the selection criteria, and waits until either
        /// a message is available in the queue, or the time-out expires.
        /// </summary>
        /// <param name="timeSpan">Reception time-out.</param>
        /// <returns>Received message.</returns>
        public override IMessage Receive(TimeSpan timeSpan)
        {
            try
            {
                // No time-out option for receiving messages by lookup identifiers:
                // either the message is present in the queue, or the method throws
                // an exception immediately if the message is not in the queue. 
                return Convert(messageQueue.ReceiveByLookupId(lookupId));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Receives the first message available in the transactional queue
        /// referenced by the MessageQueue matching the selection criteria.
        /// This call is synchronous, and blocks the current thread of execution
        /// until a message is available.
        /// </summary>
        /// <param name="transaction">Transaction.</param>
        /// <returns>Received message.</returns>
        public override IMessage Receive(MessageQueueTransaction transaction)
        {
            try
            {
                return Convert(messageQueue.ReceiveByLookupId(
                    MessageLookupAction.Current, lookupId, transaction));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Receives the first message available in the transactional queue
        /// referenced by the MessageQueue matching the selection criteria,
        /// and waits until either a message is available in the queue, or the
        /// time-out expires.
        /// </summary>
        /// <param name="timeSpan">Reception time-out.</param>
        /// <param name="transaction">Transaction.</param>
        /// <returns>Received message.</returns>
        public override IMessage Receive(TimeSpan timeSpan,
            MessageQueueTransaction transaction)
        {
            try
            {
                // No time-out option for receiving messages by lookup identifiers:
                // either the message is present in the queue, or the method throws
                // an exception immediately if the message is not in the queue. 
                return Convert(messageQueue.ReceiveByLookupId(
                    MessageLookupAction.Current, lookupId, transaction));
            }
            catch(InvalidOperationException)
            {
                return null;
            }
            catch(Exception ex)
            {
                throw ex;
            }
        }

        /// <summary>
        /// Checks if an MSMQ message matches the selection criteria.
        /// </summary>
        /// <param name="message">MSMQ message.</param>
        /// <return>true if the message matches the selection criteria.</return>
        public override bool Matches(Message message)
        {
            return message.LookupId == lookupId;
        }
	}
}