opensim/OpenSim/Region/ClientStack/LindenUDP/LLUDPClientCollection.cs

/*
 * Copyright (c) Contributors, http://opensimulator.org/
 * See CONTRIBUTORS.TXT for a full list of copyright holders.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright
 *       notice, this list of conditions and the following disclaimer in the
 *       documentation and/or other materials provided with the distribution.
 *     * Neither the name of the OpenSimulator Project nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE DEVELOPERS ``AS IS'' AND ANY
 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE CONTRIBUTORS BE LIABLE FOR ANY
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

using System;
using System.Collections.Generic;
using System.Net;
using OpenSim.Framework;
using OpenMetaverse;

using ReaderWriterLockImpl = OpenMetaverse.ReaderWriterLockSlim;

namespace OpenSim.Region.ClientStack.LindenUDP
{
    public sealed class UDPClientCollection
    {
        Dictionary<UUID, LLUDPClient> Dictionary1;
        Dictionary<IPEndPoint, LLUDPClient> Dictionary2;
        LLUDPClient[] Array;
        ReaderWriterLockImpl rwLock = new ReaderWriterLockImpl();

        public UDPClientCollection()
        {
            Dictionary1 = new Dictionary<UUID, LLUDPClient>();
            Dictionary2 = new Dictionary<IPEndPoint, LLUDPClient>();
            Array = new LLUDPClient[0];
        }

        public UDPClientCollection(int capacity)
        {
            Dictionary1 = new Dictionary<UUID, LLUDPClient>(capacity);
            Dictionary2 = new Dictionary<IPEndPoint, LLUDPClient>(capacity);
            Array = new LLUDPClient[0];
        }

        public void Add(UUID key1, IPEndPoint key2, LLUDPClient value)
        {
            rwLock.EnterWriteLock();

            try
            {
                if (Dictionary1.ContainsKey(key1))
                {
                    if (!Dictionary2.ContainsKey(key2))
                        throw new ArgumentException("key1 exists in the dictionary but not key2");
                }
                else if (Dictionary2.ContainsKey(key2))
                {
                    if (!Dictionary1.ContainsKey(key1))
                        throw new ArgumentException("key2 exists in the dictionary but not key1");
                }

                Dictionary1[key1] = value;
                Dictionary2[key2] = value;

                LLUDPClient[] oldArray = Array;
                int oldLength = oldArray.Length;

                LLUDPClient[] newArray = new LLUDPClient[oldLength + 1];
                for (int i = 0; i < oldLength; i++)
                    newArray[i] = oldArray[i];
                newArray[oldLength] = value;

                Array = newArray;
            }
            finally { rwLock.ExitWriteLock(); }
        }

        public bool Remove(UUID key1, IPEndPoint key2)
        {
            rwLock.EnterWriteLock();

            try
            {
                LLUDPClient value;
                if (Dictionary1.TryGetValue(key1, out value))
                {
                    Dictionary1.Remove(key1);
                    Dictionary2.Remove(key2);

                    LLUDPClient[] oldArray = Array;
                    int oldLength = oldArray.Length;

                    LLUDPClient[] newArray = new LLUDPClient[oldLength - 1];
                    int j = 0;
                    for (int i = 0; i < oldLength; i++)
                    {
                        if (oldArray[i] != value)
                            newArray[j++] = oldArray[i];
                    }

                    Array = newArray;
                    return true;
                }
            }
            finally { rwLock.ExitWriteLock(); }

            return false;
        }

        public void Clear()
        {
            rwLock.EnterWriteLock();

            try
            {
                Dictionary1.Clear();
                Dictionary2.Clear();
                Array = new LLUDPClient[0];
            }
            finally { rwLock.ExitWriteLock(); }
        }

        public int Count
        {
            get { return Array.Length; }
        }

        public bool ContainsKey(UUID key)
        {
            return Dictionary1.ContainsKey(key);
        }

        public bool ContainsKey(IPEndPoint key)
        {
            return Dictionary2.ContainsKey(key);
        }

        public bool TryGetValue(UUID key, out LLUDPClient value)
        {
            bool success;
            bool doLock = !rwLock.IsUpgradeableReadLockHeld;
            if (doLock) rwLock.EnterReadLock();

            try { success = Dictionary1.TryGetValue(key, out value); }
            finally { if (doLock) rwLock.ExitReadLock(); }

            return success;
        }

        public bool TryGetValue(IPEndPoint key, out LLUDPClient value)
        {
            bool success;
            bool doLock = !rwLock.IsUpgradeableReadLockHeld;
            if (doLock) rwLock.EnterReadLock();

            try { success = Dictionary2.TryGetValue(key, out value); }
            finally { if (doLock) rwLock.ExitReadLock(); }

            return success;
        }

        public void ForEach(Action<LLUDPClient> action)
        {
            bool doLock = !rwLock.IsUpgradeableReadLockHeld;
            if (doLock) rwLock.EnterUpgradeableReadLock();

            try { Parallel.ForEach<LLUDPClient>(Array, action); }
            finally { if (doLock) rwLock.ExitUpgradeableReadLock(); }
        }
    }
}