110 lines
4.4 KiB
C#
110 lines
4.4 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Threading.Tasks;
|
||
|
||
namespace CloudBuilder.Core.Extensions
|
||
{
|
||
using System;
|
||
using System.Collections.Concurrent;
|
||
using System.Reflection;
|
||
|
||
public static class CustomAttributeExtensions
|
||
{
|
||
/// <summary>
|
||
/// Cache Data
|
||
/// </summary>
|
||
private static readonly ConcurrentDictionary<string, object> Cache = new ConcurrentDictionary<string, object>();
|
||
|
||
/// <summary>
|
||
/// 获取CustomAttribute Value
|
||
/// </summary>
|
||
/// <typeparam name="TAttribute">Attribute的子类型</typeparam>
|
||
/// <typeparam name="TReturn">TReturn的子类型</typeparam>
|
||
/// <param name="sourceType">头部标有CustomAttribute类的类型</param>
|
||
/// <param name="attributeValueAction">取Attribute具体哪个属性值的匿名函数</param>
|
||
/// <returns>返回Attribute的值，没有则返回null</returns>
|
||
public static TReturn GetCustomAttributeValue<TAttribute, TReturn>(this Type sourceType, Func<TAttribute, TReturn> attributeValueAction)
|
||
where TAttribute : Attribute
|
||
{
|
||
return _getAttributeValue(sourceType, attributeValueAction, null);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取CustomAttribute Value
|
||
/// </summary>
|
||
/// <typeparam name="TAttribute">Attribute的子类型</typeparam>
|
||
/// <typeparam name="TReturn">TReturn的子类型</typeparam>
|
||
/// <param name="sourceType">头部标有CustomAttribute类的类型</param>
|
||
/// <param name="attributeValueAction">取Attribute具体哪个属性值的匿名函数</param>
|
||
/// <param name="propertyName">field name或property name</param>
|
||
/// <returns>返回Attribute的值，没有则返回null</returns>
|
||
public static TReturn GetCustomAttributeValue<TAttribute, TReturn>(this Type sourceType, Func<TAttribute, TReturn> attributeValueAction, string propertyName)
|
||
where TAttribute : Attribute
|
||
{
|
||
return _getAttributeValue(sourceType, attributeValueAction, propertyName);
|
||
}
|
||
|
||
#region private methods
|
||
|
||
private static TReturn _getAttributeValue<TAttribute, TReturn>(Type sourceType, Func<TAttribute, TReturn> attributeFunc, string propertyName)
|
||
where TAttribute : Attribute
|
||
{
|
||
var cacheKey = BuildKey<TAttribute>(sourceType, propertyName);
|
||
var value = Cache.GetOrAdd(cacheKey, k => GetValue(sourceType, attributeFunc, propertyName));
|
||
if (value is TReturn) return (TReturn)Cache[cacheKey];
|
||
return default(TReturn);
|
||
}
|
||
|
||
private static string BuildKey<TAttribute>(Type type, string propertyName) where TAttribute : Attribute
|
||
{
|
||
var attributeName = typeof(TAttribute).FullName;
|
||
if (string.IsNullOrEmpty(propertyName))
|
||
{
|
||
return type.FullName + "." + attributeName;
|
||
}
|
||
|
||
return type.FullName + "." + propertyName + "." + attributeName;
|
||
}
|
||
|
||
private static TReturn GetValue<TAttribute, TReturn>(this Type type, Func<TAttribute, TReturn> attributeValueAction, string name)
|
||
where TAttribute : Attribute
|
||
{
|
||
TAttribute attribute = default(TAttribute);
|
||
if (string.IsNullOrEmpty(name))
|
||
{
|
||
attribute = type.GetCustomAttribute<TAttribute>(false);
|
||
}
|
||
else
|
||
{
|
||
var propertyInfo = type.GetProperty(name);
|
||
if (propertyInfo != null)
|
||
{
|
||
attribute = propertyInfo.GetCustomAttribute<TAttribute>(false);
|
||
}
|
||
else
|
||
{
|
||
var fieldInfo = type.GetField(name);
|
||
if (fieldInfo != null)
|
||
{
|
||
attribute = fieldInfo.GetCustomAttribute<TAttribute>(false);
|
||
}
|
||
else
|
||
{
|
||
var methodInfo = type.GetMethod(name);
|
||
if (methodInfo != null)
|
||
{
|
||
attribute = methodInfo.GetCustomAttribute<TAttribute>(false);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return attribute == null ? default(TReturn) : attributeValueAction(attribute);
|
||
}
|
||
|
||
#endregion
|
||
}
|
||
}
|