代码整理,简化字段构造
大石头 authored at 2021-04-16 23:19:46
4.25 KiB
X
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Threading;
using NewLife;
using NewLife.Log;
using NewLife.Threading;
using XCode.DataAccessLayer;

namespace XCode.Cache
{
    /// <summary>缓存基类</summary>
    public abstract class CacheBase<TEntity> : CacheBase where TEntity : Entity<TEntity>, new()
    {
        #region 属性
        /// <summary>连接名</summary>
        public String ConnName { get; set; }

        /// <summary>表名</summary>
        public String TableName { get; set; }
        #endregion

        /// <summary>调用委托方法前设置连接名和表名,调用后还原</summary>
        internal TResult Invoke<T, TResult>(Func<T, TResult> callback, T arg)
        {
            var cn = Entity<TEntity>.Meta.ConnName;
            var tn = Entity<TEntity>.Meta.TableName;

            if (cn != ConnName) Entity<TEntity>.Meta.ConnName = ConnName;
            if (tn != TableName) Entity<TEntity>.Meta.TableName = TableName;

            try
            {
                return callback(arg);
            }
            // 屏蔽对象销毁异常
            catch (ObjectDisposedException) { return default; }
            // 屏蔽线程取消异常
            catch (ThreadAbortException) { return default; }
            catch (Exception ex)
            {
                // 无效操作,句柄未初始化,不用出现
                if (ex is InvalidOperationException && ex.Message.Contains("句柄未初始化")) return default;
                DAL.WriteLog(ex.ToString());
                throw;
            }
            finally
            {
                if (cn != ConnName) Entity<TEntity>.Meta.ConnName = cn;
                if (tn != TableName) Entity<TEntity>.Meta.TableName = tn;
            }
        }
    }

    /// <summary>缓存基类</summary>
    public abstract class CacheBase : DisposeBase
    {
        #region 设置
        /// <summary>是否调试缓存模块</summary>
        public static Boolean Debug { get; set; }

        /// <summary>显示统计信息的周期。默认60*60s,DAL.Debug=true时10*60s,Debug=true时60s</summary>
        public static Int32 Period { get; set; }

        /// <summary>日志前缀</summary>
        protected String LogPrefix { get; set; }
        #endregion

        static CacheBase()
        {
#if DEBUG
            Debug = true;
#endif
        }

        internal void WriteLog(String format, params Object[] args)
        {
            if (Debug) XTrace.WriteLine(LogPrefix + format, args);
        }

        /// <summary>检查并显示统计信息</summary>
        /// <param name="total"></param>
        /// <param name="show"></param>
        internal static void CheckShowStatics(ref Int32 total, Action show)
        {
            Interlocked.Increment(ref total);

            NextShow = true;

            // 加入列表
            if (total < 10)
            {
                var key = show?.Target?.GetType().FullName;
                if (key != null && !_dic.ContainsKey(key))
                {
                    _dic.TryAdd(key, show);
                }
            }

            // 启动定时器
            if (_timer == null)
            {
                lock (typeof(CacheBase))
                {
                    if (_timer == null)
                    {
                        var ms = Period * 1000;
                        if (ms == 0)
                        {
                            ms = 60 * 60 * 1000;
                            if (DAL.Debug) ms = 10 * 60 * 1000;
                            if (Debug) ms = 1 * 60 * 1000;
                        }
                        _timer = new TimerX(Check, null, 10000, ms);
                    }
                }
            }
        }

        private static TimerX _timer;
        private static readonly ConcurrentDictionary<String, Action> _dic = new();
        private static Boolean NextShow;

        private static void Check(Object state)
        {
            if (!NextShow) return;

            NextShow = false;

            foreach (var item in _dic)
            {
                item.Value();
            }
        }
    }
}