C#教程之数据库查询 - DataTable转Entity类型数据

当使用Sql语句查询数据库，返回DataSet数据集。

DataSet转化为数据列表，可以通过映射方式直接返回Entity数据列表

新建一个特性类，用于数据库列表列名称映射

LinqToDB提供了一个ColumnAttribute，但是通过反射不方便获取ColumnAttribute
获取CustomAttributes时，虽然可以筛选到指定ColumnAttribute，但是CustomAttributeData不容易转换到ColumnAttribute

var customAttributeDatas = mi.CustomAttributes.Where(i=>i.AttributeType.Name==nameof(ColumnAttribute)).ToList();

以下是自定义的特性类

    [AttributeUsage(AttributeTargets.Property)]
    public sealed class ColumnFieldAttribute : Attribute
    {
        /// <summary>
        /// 表对应的字段名

        /// </summary>
        public string ColumnName { set; get; }

        public ColumnFieldAttribute(string columnName)
        {
            ColumnName = columnName;
        }
    }

通过反射，将DataTable转换为数据列表

    /// <summary>
    /// 将DataRow/DataTable转换成Entity类型
    /// </summary>
    public static class DataTableConverter<T> where T : new()
    {
        /// <summary>
        /// 将DataTable转换成Entity列表
        /// </summary>
        /// <param name="dt"></param>
        /// <returns></returns>
        public static List<T> ToList(DataTable dt)
        {
            List<T> list = new List<T>(dt.Rows.Count);
            foreach (DataRow dr in dt.Rows)
            {
                list.Add(ToEntity(dr));
            }
            return list;
        }
        /// <summary>
        /// 将DataRow行转换成Entity
        /// </summary>
        /// <param name="dr"></param>
        /// <returns></returns>
        private static T ToEntity(DataRow dr)
        {
            T entity = new T();
            Type info = typeof(T);
            var members = info.GetMembers();
            foreach (var memberInfo in members)
            {
                if (memberInfo.MemberType == MemberTypes.Property)
                {
                    //读取属性上的DataField特性
                    object[] attributes = memberInfo.GetCustomAttributes(typeof(ColumnFieldAttribute), true);
                    foreach (var attr in attributes)
                    {
                        var dataFieldAttr = attr as ColumnFieldAttribute;
                        if (dataFieldAttr != null)
                        {
                            var propInfo = info.GetProperty(memberInfo.Name);
                            if (dr.Table.Columns.Contains(dataFieldAttr.ColumnName))
                            {
                                //根据ColumnName，将dataRow中的相对字段赋值给Entity属性
                                propInfo.SetValue(entity,
                                    Convert.ChangeType(dr[dataFieldAttr.ColumnName], propInfo.PropertyType),
                                    null);
                            }
                        }
                    }
                }
            }
            return entity;
        }
    }