Monitor的擴展支持string的超時鎖

對Monitor的使用可以防止lock的時間過長并且可以設定其對應的超時時間達到對預期代碼的一個控制,合理的使用timeout可以有助于程式的健壯性，但是對于不同的并發程式可能某些時候我們需要的粒度是不一樣的,從而產生的一個問題是需要更細粒度的鎖來保證,又因為默認的字串無法共享導致的無法通過string來進行鎖粒度的細分所以可能需要自己重寫一個鎖來保證到達更細粒度的控制,可能這時候有人要說不是有string.Intern可以進行字串保留,達到類似的效果,對于這個我只想說內部確定的字串確實可以達到這個效果,但是因為字串保留機制就導致了gc不會對此進行回收,會導致如果外部輸入的string是不可控的情況下就可以給程式造成諸如oom的問題,對此我拋磚引玉希望各位博友可以提出寶貴的意見或者建議,或者有現成的更好的解決方案也請分享一下,下面貼代碼:

    public class MonitorStr
    {
        private MonitorStr() { }
        private static ConcurrentDictionary<string, MonitorStrEntry> _lockDics = new ConcurrentDictionary<string, MonitorStrEntry>();
        private const int _concurrentCount = 31;
        private static object[] _lockers = new object[_concurrentCount];

        static MonitorStr()
        {
            for (int i = 0; i < _concurrentCount; i++)
            {
                _lockers[i] = new object();
            }
        }
        private static int GetIndex(string key)
        {
            return Math.Abs(key.GetHashCode() % _concurrentCount);
        }
        public static bool TryEnter(string key, int timeoutMillis)
        {
            if (string.IsNullOrWhiteSpace(key))
                throw new ArgumentNullException(nameof(key));
            MonitorStrEntry entry = null;
            var locker = _lockers[GetIndex(key)];
            lock (locker)
            {
                if (!_lockDics.TryGetValue(key, out entry))
                {
                    entry = new MonitorStrEntry();
                    _lockDics[key] = entry;
                }
                entry.Increment();
            }

            var acquired = Monitor.TryEnter(entry, timeoutMillis);
            if (!acquired)
                entry.Decrement();
            return acquired;
        }

        public static void Exit(string key)
        {
            var entry = _lockDics[key];
            Monitor.Exit(entry);
            if (entry.Decrement() == 0)
            {
                var locker = _lockers[GetIndex(key)];
                lock (locker)
                {
                    if (entry.CanRemove())
                    {
                        Console.WriteLine(key + "remove");
                        _lockDics.TryRemove(key, out var v);
                    }
                }
            }
        }
        class MonitorStrEntry
        {
            private int _lockCount;

            public int Increment()
            {
                Interlocked.Increment(ref _lockCount);
                return _lockCount;
            }

            public int Decrement()
            {
                Interlocked.Decrement(ref _lockCount);
                return _lockCount;
            }
            public bool CanRemove()
            {
                return _lockCount == 0;
            }
        }

    }

這個代碼的原理就是利用陣列物件鎖和傳入的string key進行對不同key之間的粒度的區分,因為不同的key之間的hashcode不一致所以對取到的鎖物件locker也不一樣,達到降低鎖并發的級別,字典存盤的entry內部維護一個鎖的加鎖次數達,利用cas保證并發多執行緒安全且高效，

如何使用

            var key = "testKey";
            var timeoutMillis = 3000;
            var acquired = false;
            try
            {

                 acquired = MonitorStr.TryEnter(key, timeoutMillis);
                if (acquired)
                {
                    //Do Something
                }
                else
                {
                    throw new XXXTimeOutException();
                }
            }
            finally
            {
                MonitorStr.Exit(key);
            }

哪個場景下可以使用呢?

            //使用場景,諸如多執行緒查資料庫環境的情況下
            var userId = "xxxx";
            var user = Cache.Query(userId);
            if (user == null)
            {
                var acquired = false;
                try
                {

                    acquired = MonitorStr.TryEnter(key, timeoutMillis);
                    if (acquired)
                    {
                        //Do Something
                        user = Cache.Query(userId);
                        if (user == null)
                        {
                            user = DB.Query(userId);
                        }
                    }
                    else
                    {
                        throw new XXXTimeOutException();
                    }
                }
                finally
                {
                    MonitorStr.Exit(key);
                }
            }

謝謝

標籤：.NET Core

上一篇：沒有提示錯誤，但是資料庫卻沒有更新

下一篇：Monitor的擴展支持string的超時鎖