引入redis服务,支持自动化单元测试
大石头 编写于 2022-03-31 22:56:30
X
using System;
using System.Collections.Generic;
using System.ComponentModel;
using System.Linq;
using System.Threading.Tasks;
using XCode.Configuration;
using NewLife;

namespace XCode.Cache
{
    /// <summary>统计字段缓存</summary>
    /// <typeparam name="TEntity"></typeparam>
    [DisplayName("统计字段")]
    public class FieldCache<TEntity> : EntityCache<TEntity> where TEntity : Entity<TEntity>, new()
    {
        private readonly String _fieldName;
        private FieldItem _field;
        private FieldItem _Unique;

        /// <summary>最大行数。默认50</summary>
        public Int32 MaxRows { get; set; } = 50;

        /// <summary>数据源条件</summary>
        public WhereExpression Where { get; set; }

        /// <summary>获取显示名的委托</summary>
        public Func<TEntity, String> GetDisplay { get; set; }

        /// <summary>显示名格式化字符串,两个参数是名称和个数</summary>
        public String DisplayFormat { get; set; } = "{0} ({1:n0})";

        /// <summary>对指定字段使用实体缓存</summary>
        /// <param name="fieldName"></param>
        public FieldCache(String fieldName)
        {
            WaitFirst = false;
            //Expire = 10 * 60;
            FillListMethod = Search;
            _fieldName = fieldName;

            LogPrefix = $"FieldCache<{typeof(TEntity).Name}+{_fieldName}>";
        }

        private void Init()
        {
            if (_field == null && !_fieldName.IsNullOrEmpty()) _field = Entity<TEntity>.Meta.Table.FindByName(_fieldName);

            if (_Unique == null)
            {
                var tb = _field.Table;
                var id = tb.Identity;
                if (id == null && tb.PrimaryKeys.Length == 1) id = tb.PrimaryKeys[0];
                _Unique = id ?? throw new Exception($"{tb.TableName}缺少唯一主键,无法使用缓存");

                // 数据量较大时,扩大有效期
                var count = Entity<TEntity>.Meta.Count;
                if (count > 1_000_000)
                    Expire *= 60;
                else if (count > 10_000)
                    Expire *= 10;
                else
                    Expire *= 3;
            }
        }

        private IList<TEntity> Search() => Entity<TEntity>.FindAll(Where.GroupBy(_field), "group_count desc", _Unique.Count("group_count") & _field, 0, MaxRows);

        private IDictionary<String, String> GetAll()
        {
            //var id = _field.Table.Identity;
            var list = Entities.Take(MaxRows).ToList();

            var dic = new Dictionary<String, String>();
            foreach (var entity in list)
            {
                var k = entity[_field.Name] + "";
                var v = k;
                if (GetDisplay != null)
                {
                    v = GetDisplay(entity);
                    if (v.IsNullOrEmpty()) v = $"[{k}]";
                }

                dic[k] = String.Format(DisplayFormat, v, entity["group_count"]);
            }

            // 更新缓存
            if (dic.Count > 0)
            {
                var key = $"{typeof(TEntity).Name}_{_field?.Name}";
                var dc = DataCache.Current;

                dc.FieldCache[key] = dic;
                dc.SaveAsync();
            }

            _task = null;

            return dic;
        }

        private Task<IDictionary<String, String>> _task;
        /// <summary>获取所有类别名称</summary>
        /// <returns></returns>
        public IDictionary<String, String> FindAllName()
        {
            Init();

            var key = $"{typeof(TEntity).Name}_{_field?.Name}";
            var dc = DataCache.Current;

            if (_task == null || _task.IsCompleted) _task = Task.Run(GetAll);

            // 优先从缓存读取
            if (dc.FieldCache.TryGetValue(key, out var rs)) return rs;

            return _task.Result;
        }

        #region 辅助
        /// <summary>输出名称</summary>
        /// <returns></returns>
        public override String ToString()
        {
            var type = GetType();
            var name = type.GetDisplayName() ?? type.Name;
            return $"{name}<{typeof(TEntity).FullName}>[{_field.Name}]";
        }
        #endregion
    }
}