C# System.Attribute(验证类)

2021-09-10 22:53:35

本文以一个项目中通用的验证类来举例说明如何使用自定义Attribute来扩展元数据。

 在项目中，我们为了保证各个层次之间的松藕合，通常把在各个层次之间传递数据的封装在一个称为实体类的类中，比如ActionFrom

  1. using System;
  2. namespace AttributeTest
  3. {
  4. public class ActionForm
  5. {
  6. private string email = "";
  7. private string password = "";
  8. public string Email
  9. {
  10. get { return this.email; }
  11. set { this.email = value; }
  12. }
  13. public string Password
  14. {
  15. get { return this.password; }
  16. set { this.password = value; }
  17. }
  18. }
  19. }

现在，在使用这些实体类中的数据之前，我们需要对其中的数据进行验证。通常我们会写个静态类，用来提供各种不同的验证方法。比如需要验证Email，验证Password，比如：

  1. using System;
  2. using System.Reflection;
  3. using System.Text.RegularExpressions;
  4. namespace AttributeTest
  5. {
  6. public class Validator
  7. {
  8. public static bool ValidateEmail(string email)
  9. {
  10. //方法体
  11. }
  12. public static bool ValidatePassword(string passwd)
  13. {
  14. //方法体
  15. }
  16. }
  17. }

这样的硬编码混迹于各个层次之间，一旦实体类里某个属性发生变化，就不得不修改各个层次中的相关验证代码。于是，我们想到可以使用一个统一的验证方法用来验证所有的实体类中的属性。

  1. public static bool Validate(string propertyName, string propertyValue, Validator.ValidateType t) {...}

这里，Validator.ValidateType 是Validator中提供的一个枚举。

  1. public enum ValidateType
  2. {
  3. Email,
  4. Password,
  5. Number,
  6. Id
  7. }

这里这个验证方法，的第三个参数使得验证与实体类的耦合密度增加了。我们还是不得不在修改实体类的时候，修改验证方法的调用代码。

现在，我们需要自定义Attribute来扩展实体类的元数据。通过对实体类元数据的描述，我们可以去掉验证方法里的第三个参数

  1. using System;
  2. namespace AttributeTest
  3. {
  4. [System.AttributeUsage(AttributeTargets.Property)]
  5. public class ValidateAttribute : System.Attribute
  6. {
  7. public ValidateAttribute(ValidateType validateType)
  8. {
  9. this.validateType = validateType;
  10. }
  11. private ValidateType validateType;
  12. public ValidateType ValidateType
  13. {
  14. get { return this.validateType; }
  15. set { this.validateType = value; }
  16. }
  17. }
  18. public enum ValidateType
  19. {
  20. Email,
  21. Password,
  22. Number,
  23. Id
  24. }
  25. }

自定义Attribute（特性）必须继承于System.Attribute。还可以通过System.AttributeUsageAttribute特性，控制自定义特性的使用范围（构件），例如，字段、方法。[System.AttributeUsage(AttributeTargets.Property)]限制这个自定义特性只能使用在类的属性上。

现在，我们实现这个验证方法：

  1. using System;
  2. using System.Reflection;
  3. using System.Text.RegularExpressions;
  4. namespace AttributeTest
  5. {
  6. public class Validator
  7. {
  8. public static bool Validate(object validateObject, string validateProperty)
  9. {
  10. System.Type t = validateObject.GetType();
  11. PropertyInfo pi = t.GetProperty(validateProperty);
  12. string validateValue = pi.GetValue(validateObject, null) as string;
  13. if (pi.IsDefined(typeof(ValidateAttribute), true))
  14. {
  15. object[] atts = pi.GetCustomAttributes(true);
  16. ValidateAttribute vatt = atts[0] as ValidateAttribute;
  17. string strExpr = "";
  18. switch (vatt.ValidateType)
  19. {
  20. case ValidateType.Email:
  21. strExpr = @"^[\w-]+(\.[\w-]+)*@[\w-]+(\.[\w-]+)+{1}quot;;
  22. break;
  23. case ValidateType.Password:
  24. strExpr = @"\d{6}";
  25. break;
  26. case ValidateType.Number:
  27. strExpr = @"^\d*{1}quot;;
  28. break;
  29. case ValidateType.Id:
  30. strExpr = @"^\w*{1}quot;;
  31. break;
  32. default:
  33. return true;
  34. }
  35. Regex validateRegex = new Regex(strExpr);
  36. return validateRegex.IsMatch(validateValue);
  37. }
  38. return true;
  39. }
  40. }
  41. }

该方法需要两个参数，一个是需要验证的实体类的实例，还有一个是需要验证的属性名。当然，我们还需要在实体类上加上我们自定义的特性：

  1. using System;
  2. namespace AttributeTest
  3. {
  4. public class ActionForm
  5. {
  6. private string email = "";
  7. private string password = "";
  8. [Validate(ValidateType.Email)]
  9. public string Email
  10. {
  11. get { return this.email; }
  12. set { this.email = value; }
  13. }
  14. [Validate(ValidateType.Password)]
  15. public string Password
  16. {
  17. get { return this.password; }
  18. set { this.password = value; }
  19. }
  20. }
  21. }


我们通过自定义特性对实体类的元数据进行扩展，指定每个属性需要验证的类型。


现在我们可以这样使用这个验证类：




    

  1. ActionForm form = new ActionForm();
    2. 

  2. form.Email = justacoder@123.com;
    3. 

  3. form.Password = "123456";
    4. 

  4. bool isValidEmail = Validator.Validate(form, "Email");
    5. 

  5. bool isValidPassword = Validator.Validate(form, "Password");
    6. 

  6. Console.WriteLine("Email is {0}.", isValidEmail?"valid":"invalid");
    7. 

  7. Console.WriteLine("Password is {0}.", isValidPassword?"valid":"invalid");
    8. 

  8. Console.ReadLine();
    9. 





我们通过抛出自定义异常的方法，将验证扩大到实体类级别的验证：




    

  1. public static void ValidateProperty(object validateObject, string validateProperty)
    2. 

  2. {
    3. 

  3. System.Type t = validateObject.GetType();
    4. 

  4. PropertyInfo pi = t.GetProperty(validateProperty);
    5. 

  5. string validateValue = pi.GetValue(validateObject, null) as string;
    6. 

  6. if( pi.IsDefined(typeof(ValidateAttribute), true) )
    7. 

  7. {
    8. 

  8. object[] atts = pi.GetCustomAttributes(true);
    9. 

  9. ValidateAttribute vatt = atts[0] as ValidateAttribute;
    10. 

  10. string strExpr = "";
    11. 

  11. switch(vatt.ValidateType)
    12. 

  12. {
    13. 

  13. case ValidateType.Email:
    14. 

  14. strExpr = @"^[\w-]+(\.[\w-]+)*@[\w-]+(\.[\w-]+)+{1}quot;;
    15. 

  15. break;
    16. 

  16. case ValidateType.Password:
    17. 

  17. strExpr = @"\d{6}";
    18. 

  18. break;
    19. 

  19. case ValidateType.Number:
    20. 

  20. strExpr = @"^\d*{1}quot;;
    21. 

  21. break;
    22. 

  22. case ValidateType.Id:
    23. 

  23. strExpr = @"^\w*{1}quot;;
    24. 

  24. break;
    25. 

  25. default:
    26. 

  26. return;
    27. 

  27. }
    28. 

  28. Regex validateRegex = new Regex(strExpr);
    29. 

  29. if( !validateRegex.IsMatch(validateValue) )
    30. 

  30. {
    31. 

  31. throw new ApplicationException(validateProperty + " is invalid.");
    32. 

  32. }
    33. 

  33. }
    34. 

  34. }
    35. 

  35. public static void Validate(object validateObject)
    36. 

  36. {
    37. 

  37. System.Type t = validateObject.GetType();
    38. 

  38. PropertyInfo[] ps = t.GetProperties();
    39. 

  39. foreach(PropertyInfo pi in ps)
    40. 

  40. {
    41. 

  41. ValidateProperty(validateObject, pi.Name);
    42. 

  42. }
    43. 

  43. }
    44. 





现在验证，只需要这样：




    

  1. try
    2. 

  2. {
    3. 

  3. Validator.Validate(form);
    4. 

  4. }
    5. 

  5. catch(Exception ex)
    6. 



上一篇：【SSH系列】Hibernate映射 -- 继承映射

下一篇：mysql性能调优与架构设计笔记