您的位置:首页 > 编程语言

使用反射、特性简化代码

2016-01-29 15:07 531 查看
假设现在有一个学生类(Student)

/// <summary>
/// 学生类
/// </summary>
public class Student
{
/// <summary>
/// 名字
/// </summary>
private string name;
public string Name
{
get { return name; }
set { name = value; }
}

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

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

/// <summary>
/// 性别
/// </summary>
public string Sex;
}


如果需要判断某些字段(属性)是否为空,是否大于0,便有以下代码:

public static string ValidateStudent(Student student)
{
StringBuilder validateMessage = new StringBuilder();
if (string.IsNullOrEmpty(student.Name))
{
validateMessage.Append("名字不能为空");
}

if (string.IsNullOrEmpty(student.Sex))
{
validateMessage.Append("性别不能为空");
}

if (student.Age <= 0)
{
validateMessage.Append("年龄必填大于0");
}

//...... 几百行

// 写到这里发现不对啊,如果必填项有20多个,难道我要一直这样写吗!
return validateMessage.ToString();
}


这样的代码,重用性不高,而且效率低。

我们可以用特性,反射,然后遍历属性并检查特性。

首先自定义一个【必填】特性类,继承自Attribute

/// <summary>
/// 【必填】特性,继承自Attribute
/// </summary>
public sealed class RequireAttribute : Attribute
{
private bool isRequire;
public bool IsRequire
{
get { return isRequire; }
}

/// <summary>
/// 构造函数
/// </summary>
/// <param name="isRequire"></param>
public RequireAttribute(bool isRequire)
{
this.isRequire = isRequire;
}
}


然后用这个自定义的特性标记学生类的成员属性:

/// <summary>
/// 学生类
/// </summary>
public class Student
{
/// <summary>
/// 名字
/// </summary>
private string name;
[Require(true)]
public string Name
{
get { return name; }
set { name = value; }
}

/// <summary>
/// 年龄
/// </summary>
[Require(true)]
public int Age { get; set; }

/// <summary>
/// 地址
/// </summary>
[Require(false)]
public string Address { get; set; }

/// <summary>
/// 性别
/// </summary>
[Require(true)]
public string Sex;
}


通过特性检查类的属性:

/// <summary>
/// 检查方法,支持泛型
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="instance"></param>
/// <returns></returns>
public static string CheckRequire<T>(T instance)
{
var validateMsg = new StringBuilder();

//获取T类的属性
Type t = typeof (T);
var propertyInfos = t.GetProperties();

//遍历属性
foreach (var propertyInfo in propertyInfos)
{
//检查属性是否标记了特性
RequireAttribute attribute = (RequireAttribute) Attribute.GetCustomAttribute(propertyInfo, typeof (RequireAttribute));

//没标记,直接跳过
if (attribute == null)
{
continue;
}

//获取属性的数据类型
var type = propertyInfo.PropertyType.ToString().ToLower();

//获取该属性的值
var value = propertyInfo.GetValue(instance);

if (type.Contains("system.string"))
{
if (string.IsNullOrEmpty((string) value) && attribute.IsRequire)
validateMsg.Append(propertyInfo.Name).Append("不能为空").Append(",");
}
else if (type.Contains("system.int"))
{
if ((int) value == 0 && attribute.IsRequire)
validateMsg.Append(propertyInfo.Name).Append("必须大于0").Append(",");
}
}

return validateMsg.ToString();
}


执行验证:

static void Main(string[] args)
{
var obj = new Student()
{
Name = ""
};
Console.WriteLine(CheckRequire(obj));
Console.Read();
}


结果输出:



有人会发现,Sex也标记了[Require(true)],为什么没有验证信息,这是因为,Sex没有实现属性{ get; set; },GetProperties是获取不到的。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: