關(guān)于.NET Attribute在數(shù)據(jù)校驗(yàn)中的應(yīng)用教程

更新時(shí)間：2020年05月13日 08:45:27 作者：hexuwsbg

這篇文章主要給大家介紹了關(guān)于.NET Attribute在數(shù)據(jù)校驗(yàn)中的應(yīng)用的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或者使用.NET具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

前言

Attribute（特性）的概念不在此贅述了，相信有點(diǎn).NET基礎(chǔ)的開發(fā)人員都明白，用過(guò)Attribute的人也不在少數(shù)，畢竟很多框架都提供自定義的屬性，類似于Newtonsoft.JSON中JsonProperty、JsonIgnore等

自定義特性

.NET 框架允許創(chuàng)建自定義特性，用于存儲(chǔ)聲明性的信息，且可在運(yùn)行時(shí)被檢索。該信息根據(jù)設(shè)計(jì)標(biāo)準(zhǔn)和應(yīng)用程序需要，可與任何目標(biāo)元素相關(guān)。

創(chuàng)建并使用自定義特性包含四個(gè)步驟：

聲明自定義特性
構(gòu)建自定義特性
在目標(biāo)程序元素上應(yīng)用自定義特性
通過(guò)反射訪問特性

聲明自定義特性

一個(gè)新的自定義特性必須派生自System.Attribute類，例如：

public class FieldDescriptionAttribute : Attribute
{
  public string Description { get; private set; }

  public FieldDescriptionAttribute(string description)
  {
    Description = description;
  }
}

public class UserEntity
{
  [FieldDescription("用戶名稱")]
  public string Name { get; set; }
}

該如何拿到我們標(biāo)注的信息呢？這時(shí)候需要使用反射獲取

   var type = typeof(UserEntity);
   var properties = type.GetProperties();
   foreach (var item in properties)
   {
     if(item.IsDefined(typeof(FieldDescriptionAttribute), true))
     {
       var attribute = item.GetCustomAttribute(typeof(FieldDescriptionAttribute)) as FieldDescriptionAttribute;
       Console.WriteLine(attribute.Description);
     }
   }

執(zhí)行結(jié)果如下：

從執(zhí)行結(jié)果上看，我們拿到了我們想要的數(shù)據(jù)，那么這個(gè)特性在實(shí)際使用過(guò)程中，到底有什么用途呢？

Attribute特性妙用

在實(shí)際開發(fā)過(guò)程中，我們的系統(tǒng)總會(huì)提供各種各樣的對(duì)外接口，其中參數(shù)的校驗(yàn)是必不可少的一個(gè)環(huán)節(jié)。然而沒有特性時(shí)，校驗(yàn)的代碼是這樣的：

 public class UserEntity
 {
   /// <summary>
   /// 姓名
   /// </summary>
   [FieldDescription("用戶名稱")]
   public string Name { get; set; }

   /// <summary>
   /// 年齡
   /// </summary>
   public int Age { get; set; }

   /// <summary>
   /// 地址
   /// </summary>
   public string Address { get; set; }
 }

   UserEntity user = new UserEntity();

   if (string.IsNullOrWhiteSpace(user.Name))
   {
     throw new Exception("姓名不能為空");
   }
   if (user.Age <= 0)
   {
     throw new Exception("年齡不合法");
   }
   if (string.IsNullOrWhiteSpace(user.Address))
   {
     throw new Exception("地址不能為空");
   }

字段多了之后這種代碼就看著非常繁瑣，并且看上去不直觀。對(duì)于這種繁瑣又惡心的代碼，有什么方法可以優(yōu)化呢？
使用特性后的驗(yàn)證寫法如下：

首先定義一個(gè)基礎(chǔ)的校驗(yàn)屬性，提供基礎(chǔ)的校驗(yàn)方法

  public abstract class AbstractCustomAttribute : Attribute
  {
    /// <summary>
    /// 校驗(yàn)后的錯(cuò)誤信息
    /// </summary>
    public string ErrorMessage { get; set; }

    /// <summary>
    /// 數(shù)據(jù)校驗(yàn)
    /// </summary>
    /// <param name="value"></param>
    public abstract void Validate(object value);
  }

然后可以定義常用的一些對(duì)應(yīng)的校驗(yàn)Attribute，例如RequiredAttribute、StringLengthAttribute

    /// <summary>
    /// 非空校驗(yàn)
    /// </summary>
    [AttributeUsage(AttributeTargets.Property)]
    public class RequiredAttribute : AbstractCustomAttribute
    {
      public override void Validate(object value)
      {
        if (value == null || string.IsNullOrWhiteSpace(value.ToString()))
        {
          throw new Exception(string.IsNullOrWhiteSpace(ErrorMessage) ? "字段不能為空" : ErrorMessage);
        }
      }
    }

    /// <summary>
    /// 自定義驗(yàn)證，驗(yàn)證字符長(zhǎng)度
    /// </summary>
    [AttributeUsage(AttributeTargets.Property)]
    public class StringLengthAttribute : AbstractCustomAttribute
    {
      private int _maxLength;
      private int _minLength;

      public StringLengthAttribute(int minLength, int maxLength)
      {
        this._maxLength = maxLength;
        this._minLength = minLength;
      }

      public override void Validate(object value)
      {
        if (value != null && value.ToString().Length >= _minLength && value.ToString().Length <= _maxLength)
        {
          return;
        }

        throw new Exception(string.IsNullOrWhiteSpace(ErrorMessage) ? $"字段長(zhǎng)度必須在{_minLength}與{_maxLength}之間" : ErrorMessage);
      }
    }

添加一個(gè)用于校驗(yàn)的ValidateExtensions

public static class ValidateExtensions
 {
   /// <summary>
   /// 校驗(yàn)
   /// </summary>
   /// <typeparam name="T"></typeparam>
   /// <returns></returns>
   public static void Validate<T>(this T entity) where T : class
   {
     Type type = entity.GetType();

     foreach (var item in type.GetProperties())
     {
       //需要對(duì)Property的字段類型做區(qū)分處理針對(duì)Object List 數(shù)組需要做區(qū)分處理
       if (item.PropertyType.IsPrimitive || item.PropertyType.IsEnum || item.PropertyType.IsValueType || item.PropertyType == typeof(string))
       {
         //如果是基元類型、枚舉類型、值類型或者字符串 直接進(jìn)行校驗(yàn)
         CheckProperty(entity, item);
       }
       else
       {
         //如果是引用類型
         var value = item.GetValue(entity, null);
         CheckProperty(entity, item);
         if (value != null)
         {
           if ((item.PropertyType.IsGenericType && Array.Exists(item.PropertyType.GetInterfaces(), t => t.GetGenericTypeDefinition() == typeof(IList<>))) || item.PropertyType.IsArray)
           {
             //判斷IEnumerable
             var enumeratorMI = item.PropertyType.GetMethod("GetEnumerator");
             var enumerator = enumeratorMI.Invoke(value, null);
             var moveNextMI = enumerator.GetType().GetMethod("MoveNext");
             var currentMI = enumerator.GetType().GetProperty("Current");
             int index = 0;
             while (Convert.ToBoolean(moveNextMI.Invoke(enumerator, null)))
             {
               var currentElement = currentMI.GetValue(enumerator, null);
               if (currentElement != null)
               {
                 currentElement.Validate();
               }
               index++;
             }
           }
           else
           {
             value.Validate();
           }
         }
       }
     }
   }

   private static void CheckProperty(object entity, PropertyInfo property)
   {
     if (property.IsDefined(typeof(AbstractCustomAttribute), true))//此處是重點(diǎn)
     {
       //此處是重點(diǎn)
       foreach (AbstractCustomAttribute attribute in property.GetCustomAttributes(typeof(AbstractCustomAttribute), true))
       {
         if (attribute == null)
         {
           throw new Exception("AbstractCustomAttribute not instantiate");
         }

         attribute.Validate(property.GetValue(entity, null));
       }
     }
   }
 }

新的實(shí)體類

 public class UserEntity
 {
   /// <summary>
   /// 姓名
   /// </summary>
   [Required]
   public string Name { get; set; }

   /// <summary>
   /// 年齡
   /// </summary>
   public int Age { get; set; }

   /// <summary>
   /// 地址
   /// </summary>
   [Required]
   public string Address { get; set; }

   [StringLength(11, 11)]
   public string PhoneNum { get; set; }
 }

調(diào)用方式

UserEntity user = new UserEntity();
user.Validate();

上面的校驗(yàn)邏輯寫的比較復(fù)雜，主要是考慮到對(duì)象中包含復(fù)雜對(duì)象的情況，如果都是簡(jiǎn)單對(duì)象，可以不用考慮，只需針對(duì)單個(gè)屬性做字段校驗(yàn)

現(xiàn)有的方式是在校驗(yàn)不通過(guò)的時(shí)候拋出異常，此處大家也可以自定義異常來(lái)表示校驗(yàn)的問題，也可以返回自定義的校驗(yàn)結(jié)果實(shí)體來(lái)記錄當(dāng)前是哪個(gè)字段出的問題，留待大家自己實(shí)現(xiàn)

如果您有更好的建議和想法歡迎提出，共同進(jìn)步

總結(jié)

到此這篇關(guān)于.NET Attribute在數(shù)據(jù)校驗(yàn)中的應(yīng)用的文章就介紹到這了,更多相關(guān).NET Attribute在數(shù)據(jù)校驗(yàn)的應(yīng)用內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: