.net使用自定义类属性

.net中可以使用Type.GetCustomAttributes获取类上的自定义属性,可以使用PropertyInfo.GetCustomAttributes获取属性信息上的自定义属性。

 

下面以定义一个简单数据库表的映射实体类来说明相关的使用方法,基于自定义类属性和自定义类中的属性的自定义属性,可以方便的进行类标记和类中属性的标记

 

创建一个类的自定义属性,用于标识数据库中的表名称,需要继承自Attribute类:

[AttributeUsage(AttributeTargets.Class, Inherited = false, AllowMultiple = false)]
    public sealed class TableAttribute : Attribute
    {
        private readonly string _TableName = "";
        public TableAttribute(string tableName)
        {
            this._TableName = tableName;
        }
        public string TableName
        {
            get { return this._TableName; }
        }
    }

创建一个属性的自定义属性,用于标识数据库表中字段的名称,需要继承自Attribute类

[AttributeUsage(AttributeTargets.Property, Inherited = false, AllowMultiple = false)]
    public class FieldAttribute : Attribute
    {
        private readonly string _FieldName = "";    ///数据库的字段名称

        private System.Data.DbType _Type = System.Data.DbType.String;   ///数据库的字段类型

 

        public FieldAttribute(string fieldName)

       {

              this._FieldName=fieldName;

       }

 

        public FieldAttribute(string fieldName,System.Data.DbType type)

       {

              this._FieldName=fieldName;

              this._Type=type;

       }

 

       public string FieldName
        {
            get { return this._FieldName; }
        }

 

        public System.Data.DbType Type

        {

             get{return this._Type;}

        }

     }

 

创建一个数据实体基类:

public class BaseEntity
{
        public BaseEntity()
        {
        }

 

         /// <summary>
        /// 获取表名称
        /// </summary>
        /// <returns></returns>
        public string GetTableName()
        {
            Type type = this.GetType();
            object[] objs = type.GetCustomAttributes(typeof(TableAttribute), true);
            if (objs.Length <= 0)
            {
                throw new Exception("实体类没有标识TableAttribute属性");
            }
            else
            {
                object obj = objs[0];
                TableAttribute ta = (TableAttribute)obj;
                return ta.TableName;                            //获取表名称
            }
        }
        /// <summary>
        /// 获取数据实体类上的FieldAttribute
        /// </summary>
        /// <param name="propertyName"></param>
        /// <returns></returns>
        public FieldAttribute GetFieldAttribute(string propertyName)
        {
            PropertyInfo field = this.GetType().GetProperty(propertyName);
            if (field == null)
            {
                throw new Exception("属性名" + propertyName + "不存在");
            }
            object[] objs = field.GetCustomAttributes(typeof(FieldAttribute), true);
            if (objs.Length <= 0)
            {
                throw new Exception("类体属性名" + propertyName + "没有标识FieldAttribute属性");
            }
            else
            {
                object obj = objs[0];
                FieldAttribute fieldAttribute=(FieldAttribute)obj;
                fieldAttribute.FieldValue=field.GetValue(this,null);
                return fieldAttribute;
            }
        }

}

 

创建数据实体

[Table("Wincms_Dictionary")]            ///映射到数据库的Wincms_Dictionary表
public class Wincms_Dictionary : BaseEntity
{

         private int _DictionaryId;

 

         public Wincms_Dictionary()

         {

         }

 

        [Field("DictionaryId",DbType.Int32)]                ///映射到数据库的Wincms_Dictionary表中的字段
        public int DictionaryId
        {
            get { return this._DictionaryId; }
            set
            {
                this._DictionaryId = value;
            }
        }

 

}

 

///基于实体类获取实体对应的表名称和字段名称

public class Test

{

 

       public static void main(string[] args)

        {

               Wincms_Dictionary dict=new Wincms_Dictionary();

               Console.WriteLine("表名称:"+GetTableName(dict));

               Console.WriteLine("字段名称:"+GetFieldName(dict,"DictionaryId"));

               Console.Read();

        }

 

       ///获取实体表名称

       public  static string GetTableName(BaseEntity entity)

       {

                return entity.GetTableName();

       }

 

       ///获取实体字段名称

       public static string GetFieldName(BaseEntity entity,string propertyName)

       {

              FieldAttribute fieldAttribute=entity.GetFieldAttribute(propertyName);

              return fieldAttribute.FieldName;

       }

}

输出结果为:

表名称:Wincms_Dictionary

字段名称:DictionaryId

 

你可能感兴趣的:(.net)