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
}
}
|