ZhiYeJianKang_PeiXun/cyqdata-master/Tool/MDictionary.cs

using CYQ.Data.Table;
using System;
using System.Collections;
using System.Collections.Generic;
using System.Data.Common;
using System.Runtime.Serialization;
using System.Text;
using System.Threading;

namespace CYQ.Data.Tool
{
    /// <summary>
    /// 线程安全字典类
    /// </summary>
    /// <typeparam name="K">key</typeparam>
    /// <typeparam name="V">value</typeparam>
    public partial class MDictionary<K, V> : Dictionary<K, V>
    {
        private ReaderWriterLockSlim _lock = new ReaderWriterLockSlim();
        public MDictionary()
            : base()
        {

        }
        public MDictionary(IEqualityComparer<K> comparer)
            : base(comparer)
        {

        }
        public MDictionary(int capacity)
            : base(capacity)
        {

        }
        public MDictionary(int capacity, IEqualityComparer<K> comparer)
            : base(capacity, comparer)
        {

        }
        /// <summary>
        /// 添加值（带锁）【不存则，则添加；存在，则忽略】
        /// </summary>
        public new void Add(K key, V value)
        {
            _lock.TryEnterWriteLock(Timeout.Infinite);
            try
            {
                if (!ContainsKey(key))
                {
                    base.Add(key, value);
                }
            }
            finally
            {
                _lock.ExitWriteLock();
            }
        }
        /// <summary>
        /// 移除（带锁）
        /// </summary>
        /// <param name="key"></param>
        /// <returns></returns>
        public new bool Remove(K key)
        {
            _lock.TryEnterWriteLock(Timeout.Infinite);
            try
            {
                return base.Remove(key);
            }
            finally
            {
                _lock.ExitWriteLock();
            }

        }
        /// <summary>
        /// 索引取值（带锁）
        /// </summary>
        /// <param name="key"></param>
        /// <returns></returns>
        public new V this[K key]
        {
            get
            {
                _lock.TryEnterReadLock(Timeout.Infinite);
                try
                {
                    if (base.ContainsKey(key))
                    {
                        return base[key];
                    }
                    return default(V);
                }
                finally
                {
                    _lock.ExitReadLock();
                }
            }
            set
            {
                _lock.TryEnterWriteLock(Timeout.Infinite);
                try
                {
                    if (base.ContainsKey(key))
                    {
                        base[key] = value;
                    }
                }
                finally
                {
                    _lock.ExitWriteLock();
                }
            }
        }
        /// <summary>
        /// 通过index索引取值（带锁）
        /// </summary>
        /// <returns></returns>
        public V this[int index]
        {
            get
            {
                _lock.TryEnterReadLock(Timeout.Infinite);
                try
                {
                    if (index >= 0 && index < this.Count)
                    {
                        int i = 0;
                        foreach (V value in this.Values)
                        {
                            if (i == index)
                            {
                                return value;
                            }
                            i++;
                        }
                    }
                    else // 兼容存档hash int
                    {
                        K key = ConvertTool.ChangeType<K>(index);
                        if (base.ContainsKey(key))
                        {
                            return base[key];
                        }
                    }
                    return default(V);
                }
                finally
                {
                    _lock.ExitReadLock();
                }
            }
            set
            {
                _lock.TryEnterWriteLock(Timeout.Infinite);
                try
                {
                    if (index >= 0 && index < this.Count)
                    {
                        int i = 0;
                        foreach (K key in this.Keys)
                        {
                            if (i == index)
                            {
                                this[key] = value;
                                break;
                            }
                            i++;
                        }
                    }
                    else
                    {
                        K key = ConvertTool.ChangeType<K>(index);
                        base[key] = value;
                    }
                }
                finally
                {
                    _lock.ExitWriteLock();
                }
            }
        }

        /// <summary>
        /// 当Key为int时，通过此方法取值（带锁）
        /// </summary>
        /// <param name="key"></param>
        /// <returns></returns>
        public V Get(K key)
        {
            _lock.TryEnterReadLock(Timeout.Infinite);
            try
            {
                if (base.ContainsKey(key))
                {
                    return base[key];
                }
                return default(V);
            }
            finally
            {
                _lock.ExitReadLock();
            }
        }
        /// <summary>
        /// 检查值存在【则更新】，不存在则添加（带锁）
        /// </summary>
        public void Set(K key, V value)
        {
            _lock.TryEnterWriteLock(Timeout.Infinite);
            try
            {
                if (base.ContainsKey(key))
                {
                    base[key] = value;
                }
                else
                {
                    base.Add(key, value);
                }
            }
            finally
            {
                _lock.ExitWriteLock();
            }
        }
        /// <summary>
        /// 清空数据（带锁）
        /// </summary>
        public new void Clear()
        {
            _lock.TryEnterWriteLock(Timeout.Infinite);
            try
            {
                if (base.Count > 0)
                {
                    base.Clear();
                }
            }
            finally
            {
                _lock.ExitWriteLock();
            }

        }

        public new bool ContainsKey(K key)
        {
            if (key == null || base.Count == 0) { return false; }
            return base.ContainsKey(key);
        }
        /// <summary>
        /// 获取键列表（带锁）
        /// </summary>
        public List<K> GetKeys()
        {
            _lock.TryEnterReadLock(Timeout.Infinite);
            try
            {
                List<K> keys = new List<K>(base.Keys.Count);
                foreach (K item in base.Keys)
                {
                    keys.Add(item);
                }
                return keys;
            }
            finally
            {
                _lock.ExitReadLock();
            }
        }

        /// <summary>
        /// 获取值列表（带锁）
        /// </summary>
        public List<V> GetValues()
        {
            _lock.TryEnterReadLock(Timeout.Infinite);
            try
            {
                List<V> values = new List<V>(base.Values.Count);
                foreach (V item in base.Values)
                {
                    values.Add(item);
                }
                return values;
            }
            finally
            {
                _lock.ExitReadLock();
            }
        }
    }

    [Serializable]
    public partial class MDictionary<K, V>
    {
        protected MDictionary(SerializationInfo info, StreamingContext context)
            : base(info, context)
        {

        }
    }
}