91 lines
3.3 KiB
C#
91 lines
3.3 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.Linq;
|
||
using System.Text;
|
||
using System.Reflection;
|
||
|
||
namespace Song.ViewData.Attri
|
||
{
|
||
public class WeishaAttr : Attribute
|
||
{
|
||
//所有特性
|
||
public static Type[] Attrs = null;
|
||
/// <summary>
|
||
/// 忽略此方法，默认为false
|
||
/// 说明：则某个类设置了特性后，下面的所有方法都需要验证，除非设置[Admin(Ignore=true)]
|
||
/// </summary>
|
||
public bool Ignore { get; set; }
|
||
public WeishaAttr(bool ignore)
|
||
{
|
||
Ignore = ignore;
|
||
}
|
||
public WeishaAttr()
|
||
{
|
||
}
|
||
/// <summary>
|
||
/// 获取方法的某一类特性
|
||
/// </summary>
|
||
/// <param name="method">要验证的方法</param>
|
||
/// <returns></returns>
|
||
public static T GetAttr<T>(MemberInfo method) where T : WeishaAttr
|
||
{
|
||
T attr = null;
|
||
//先验证对象，如果对象需验证，则下面方法全部需要验证登录，除非方法设置了[Admin(Ignore = true)]
|
||
object[] attrsObj = method.DeclaringType.GetCustomAttributes(typeof(T), true);
|
||
if (attrsObj.Length > 0) attr = attrsObj[0] as T;
|
||
//再验证方法
|
||
object[] attrsMethod = method.GetCustomAttributes(typeof(T), true);
|
||
if (attrsMethod.Length > 0)
|
||
{
|
||
T admin = attrsMethod[0] as T;
|
||
if (attr == null) attr = admin;
|
||
if (attr != null && admin.Ignore) attr.Ignore = admin.Ignore;
|
||
}
|
||
return attr;
|
||
}
|
||
/// <summary>
|
||
/// 获取方法的特性
|
||
/// </summary>
|
||
/// <param name="method">要验证的方法</param>
|
||
/// <returns></returns>
|
||
public static List<T> GetAttrs<T>(MemberInfo method) where T : WeishaAttr
|
||
{
|
||
List<T> list = new List<T>();
|
||
//先验证对象，如果对象需验证，则下面方法全部需要验证登录，除非方法设置了[Admin(Ignore = true)]
|
||
object[] attrsObj = method.DeclaringType.GetCustomAttributes(typeof(T), true);
|
||
foreach (object o in attrsObj) list.Add(o as T);
|
||
//再验证方法
|
||
object[] attrsMethod = method.GetCustomAttributes(typeof(T), true);
|
||
foreach (object b in attrsMethod)
|
||
{
|
||
bool isExist = false;
|
||
foreach (T t in list)
|
||
{
|
||
if (t.GetType().FullName == b.GetType().FullName)
|
||
{
|
||
isExist = true;
|
||
T tm = b as T;
|
||
if (tm.Ignore) t.Ignore = tm.Ignore;
|
||
break;
|
||
}
|
||
}
|
||
if (!isExist) list.Add(b as T);
|
||
}
|
||
return list;
|
||
}
|
||
/// <summary>
|
||
/// 初始化，获取所有特性
|
||
/// </summary>
|
||
public static Type[] Initialization()
|
||
{
|
||
if (Attrs != null) return Attrs;
|
||
string assemblyName = "Song.ViewData";
|
||
Assembly assembly = Assembly.Load(assemblyName);
|
||
Attrs = assembly.GetExportedTypes()
|
||
.Where(t => t.FullName.StartsWith("Song.ViewData.Attri") && !t.IsAbstract)
|
||
.OrderBy(c => c.Name).ToArray();
|
||
return Attrs;
|
||
}
|
||
}
|
||
}
|