把List<T>变成DataTable

最新推荐文章于 2023-01-05 22:59:46 发布

转载最新推荐文章于 2023-01-05 22:59:46 发布 · 77 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/vwxyzh/archive/2009/11/11/1601186.html

文章标签：

#测试 #数据库

本文介绍了一种将List<T>转换为DataTable的方法，适用于单元测试中构造数据集而不影响实际数据库。此方法通过反射获取泛型T的所有属性，并创建相应的DataTable结构。

最近在做一些单元测试，因为数据库里面的数据比较乱，不适合直接做单元测试的数据源，又不想自己再往里面插数据，把已经很混乱得数据库搞得更混乱，就直接写了从List<T>直接转换成对应的DataTable的类，然后在把这个DataTable再注入到单元测试里面。于是就有了这么一段代码：

Code
        public static DataTable ToDataTable<T>(this IEnumerable<T> list)
            where T : class
        {
            var props = GetProperties<T>().ToList();
            DataTable dt = new DataTable();
            foreach (var p in props)
            {
                Type t = p.Value.PropertyType;
                if (t.IsGenericType && t.GetGenericTypeDefinition() == typeof(Nullable<>))
                {
                    t = t.GetGenericArguments()[0];
                }
                dt.Columns.Add(p.Key, t);
            }
            foreach (var item in list)
            {
                var row = dt.NewRow();
                foreach (var p in props)
                {
                    row[p.Key] = p.Value.GetValue(item, null) ?? DBNull.Value;
                }
                dt.Rows.Add(row);
            }
            return dt;
        }

这里有一个GetProperies的方法，根据大家不同的T实体的习惯，需要对应的实现方式，因为我的实体本身是通过自定义的DbColumnAttribute来指定的，所以实现如下：

Code
        private static IEnumerable<KeyValuePair<string, PropertyInfo>> GetProperties<T>()
        {
            foreach (var prop in typeof(T).GetProperties(BindingFlags.Instance | BindingFlags.Public))
            {
                if (prop.GetIndexParameters().Length > 0)
                    continue;
                var setMethod = prop.GetSetMethod(false);
                if (setMethod == null)
                    continue;
                var attr = Attribute.GetCustomAttribute(prop, typeof(DbColumnAttribute), true);
                if (attr == null)
                    continue;
                yield return new KeyValuePair<string, PropertyInfo>(((DbColumnAttribute)attr).ColumnName ?? prop.Name, prop);
            }

转载于:https://www.cnblogs.com/vwxyzh/archive/2009/11/11/1601186.html