开源 - Ideal库 - Excel资助类,TableHelper实现(三)

打印 上一主题 下一主题

主题 751|帖子 751|积分 2253

书接上回,我们今天继续讲解实现对象集合与DataTable的相互转换。

01、把表格转换为对象集合

该方法是将表格的列名称作为类的属性名,将表格的行数据转为类的对象。从而实现表格转换为对象集合。同时我们约定如果类的属性设置了DescriptionAttribute特性,则特性值和表格列名一一对应,如果没有设置特性则取属性名称和列名一一对应。
同时我们须要约束类只能为结构体或类,而不能是罗列、基础范例、以及集合范例、委托、接口等。
类的范例校验乐成后,我们还须要校验表格是否能转换为对象,即判断表格列名和类的属性名称或者Description特性值是否存在同等,如果没有一个表格列名和类的属性能对应上,则报表格列名无法映射至对象属性,无法完成转换异常。
当这些校验乐成后,开始循环处理处罚表格行记录,把每一行都转换为一个对象。
我们可以通过反射动态实例化对象,再通过反射对对象的属性动态赋值。因为我们的对象即支持类也支持结构体,因此这里面就会遇到一个技术题目,正常的property.SetValue方法并没有办法给结构体动态赋值。
这是因为结构体是值范例,而property.SetValue方法的参数都是object,因此这里面就涉及到装箱拆箱,因此SetValue是设置了装箱以后的对象,而并不能改变原对象。
而解决办法就是先把结构体赋值给object变量,然后对object变量进行SetValue设置值,最后再把object变量转为结构体。
下面我们一起看看具体实现代码:
  1. //把表格转换为对象集合
  2. //如果设置DescriptionAttribute,则将特性值作为表格的列名称
  3. //否则将属性名作为表格的列名称
  4. public static IEnumerable<T> ToModels<T>(DataTable dataTable)
  5. {
  6.     //T必须是结构体或类,并且不能是集合类型
  7.     AssertTypeValid<T>();
  8.     if (0 == dataTable.Rows.Count)
  9.     {
  10.         return [];
  11.     }
  12.     //获取T所有可写入属性
  13.     var properties = typeof(T).GetProperties().Where(u => u.CanWrite);
  14.     //校验表格是否能转换为对象
  15.     var isCanParse = IsCanMapDataTableToModel(dataTable, properties);
  16.     if (!isCanParse)
  17.     {
  18.         throw new NotSupportedException("The column name of the table cannot be mapped to an object property, and the conversion cannot be completed.");
  19.     }
  20.     var models = new List<T>();
  21.     foreach (DataRow dr in dataTable.Rows)
  22.     {
  23.         //通过反射实例化T
  24.         var model = Activator.CreateInstance<T>();
  25.         //把行数据映射到对象上
  26.         if (typeof(T).IsClass)
  27.         {
  28.             //处理T为类的情况
  29.             MapRowToModel<T>(dr, model, properties);
  30.         }
  31.         else
  32.         {
  33.             //处理T为结构体的情况
  34.             object boxed = model!;
  35.             MapRowToModel<object>(dr, boxed, properties);
  36.             model = (T)boxed;
  37.         }
  38.         models.Add(model);
  39.     }
  40.     return models;
  41. }
  42. //校验表格是否能转换为对象
  43. private static bool IsCanMapDataTableToModel(DataTable dataTable, IEnumerable<PropertyInfo> properties)
  44. {
  45.     var isCanParse = false;
  46.     foreach (var property in properties)
  47.     {
  48.         //根据属性获取列名
  49.         var columnName = GetColumnName(property);
  50.         if (!dataTable.Columns.Contains(columnName))
  51.         {
  52.             continue;
  53.         }
  54.         isCanParse = true;
  55.     }
  56.     return isCanParse;
  57. }
  58. //把行数据映射到对象上
  59. private static void MapRowToModel<T>(DataRow dataRow, T model, IEnumerable<PropertyInfo> properties)
  60. {
  61.     foreach (var property in properties)
  62.     {
  63.         //根据属性获取列名
  64.         var columnName = GetColumnName(property);
  65.         if (!dataRow.Table.Columns.Contains(columnName))
  66.         {
  67.             continue;
  68.         }
  69.         //获取单元格值
  70.         var value = dataRow[columnName];
  71.         if (value != DBNull.Value)
  72.         {
  73.             //给对象属性赋值
  74.             property.SetValue(model, Convert.ChangeType(value, property.PropertyType));
  75.         }
  76.     }
  77. }
复制代码
我们做个简朴的单元测试:
  1. [Fact]
  2. public void ToModels()
  3. {
  4.     //验证正常情况
  5.     var table = TableHelper.Create<Student<double>>();
  6.     var row1 = table.NewRow();
  7.     row1[0] = "Id-11";
  8.     row1[1] = "名称-12";
  9.     row1[2] = 33.13;
  10.     table.Rows.Add(row1);
  11.     var row2 = table.NewRow();
  12.     row2[0] = "Id-21";
  13.     row2[1] = "名称-22";
  14.     row2[2] = 33.23;
  15.     table.Rows.Add(row2);
  16.     var students = TableHelper.ToModels<Student<double>>(table);
  17.     Assert.Equal(2, students.Count());
  18.     Assert.Equal("Id-11", students.ElementAt(0).Id);
  19.     Assert.Equal("名称-12", students.ElementAt(0).Name);
  20.     Assert.Equal(33.13, students.ElementAt(0).Age);
  21.     Assert.Equal("Id-21", students.ElementAt(1).Id);
  22.     Assert.Equal("名称-22", students.ElementAt(1).Name);
  23.     Assert.Equal(33.23, students.ElementAt(1).Age);
  24. }
复制代码
02、把对象集合转换为表格

该方法首先会调用根据对象创建表格方法得到一个空白表格,然后通过反射获取对象的全部属性,然后循环处理处罚对象集合,把一个对象的全部属性值一个一个添加行的全部列中,如许就完成了一个对象映射成表的一行记录,直至全部对象转换完成即可得到一个表格。
代码如下:
  1. //把对象集合转为表格
  2. //如果设置DescriptionAttribute,则将特性值作为表格的列名称
  3. //否则将属性名作为表格的列名称
  4. public static DataTable ToDataTable<T>(IEnumerable<T> models, string? tableName = null)
  5. {
  6.     //创建表格
  7.     var dataTable = Create<T>(tableName);
  8.     if (models == null || !models.Any())
  9.     {
  10.         return dataTable;
  11.     }
  12.     //获取所有属性
  13.     var properties = typeof(T).GetProperties().Where(u => u.CanRead);
  14.     foreach (var model in models)
  15.     {
  16.         //创建行
  17.         var dataRow = dataTable.NewRow();
  18.         foreach (var property in properties)
  19.         {
  20.             //根据属性获取列名
  21.             var columnName = GetColumnName(property);
  22.             //填充行数据
  23.             dataRow[columnName] = property.GetValue(model);
  24.         }
  25.         dataTable.Rows.Add(dataRow);
  26.     }
  27.     return dataTable;
  28. }
复制代码
进行如下单元测试:
  1. [Fact]
  2. public void ToDataTable()
  3. {
  4.     //验证正常情况
  5.     var students = new List<Student<double>>();
  6.     var student1 = new Student<double>
  7.     {
  8.         Id = "Id-11",
  9.         Name = "名称-12",
  10.         Age = 33.13
  11.     };
  12.     students.Add(student1);
  13.     var student2 = new Student<double>
  14.     {
  15.         Id = "Id-21",
  16.         Name = "名称-22",
  17.         Age = 33.23
  18.     };
  19.     students.Add(student2);
  20.     var table = TableHelper.ToDataTable<Student<double>>(students, "学生表");
  21.     Assert.Equal("学生表", table.TableName);
  22.     Assert.Equal(2, table.Rows.Count);
  23.     Assert.Equal("Id-11", table.Rows[0][0]);
  24.     Assert.Equal("名称-12", table.Rows[0][1]);
  25.     Assert.Equal("33.13", table.Rows[0][2].ToString());
  26.     Assert.Equal("Id-21", table.Rows[1][0]);
  27.     Assert.Equal("名称-22", table.Rows[1][1]);
  28.     Assert.Equal("33.23", table.Rows[1][2].ToString());
  29. }
复制代码
03、把一维数组作为一列转换为表格

该方法比较简朴就是把一个一维数组作为一列数据创建一张表格,同时可以选择是否填写表名和列名。具体代码如下:
  1. //把一维数组作为一列转换为表格
  2. public static DataTable ToDataTableWithColumnArray<TColumn>(TColumn[] array, string? tableName = null, string? columnName = null)
  3. {
  4.     var dataTable = new DataTable(tableName);
  5.     //创建列
  6.     dataTable.Columns.Add(columnName, typeof(TColumn));
  7.     //添加行数据
  8.     foreach (var item in array)
  9.     {
  10.         var dataRow = dataTable.NewRow();
  11.         dataRow[0] = item;
  12.         dataTable.Rows.Add(dataRow);
  13.     }
  14.     return dataTable;
  15. }
复制代码
单元测试如下:
  1. [Fact]
  2. public void ToDataTableWithColumnArray()
  3. {
  4.     //验证正常情况
  5.     var columns = new string[] { "A", "B" };
  6.     var table = TableHelper.ToDataTableWithColumnArray<string>(columns, "学生表");
  7.     Assert.Equal("学生表", table.TableName);
  8.     Assert.Equal("Column1", table.Columns[0].ColumnName);
  9.     Assert.Equal(2, table.Rows.Count);
  10.     Assert.Equal("A", table.Rows[0][0]);
  11.     Assert.Equal("B", table.Rows[1][0]);
  12.     table = TableHelper.ToDataTableWithColumnArray<string>(columns, "学生表", "列");
  13.     Assert.Equal("列", table.Columns[0].ColumnName);
  14. }
复制代码
04、把一维数组作为一行转换为表格

该方法也比较简朴就是把一个一维数组作为一行数据创建一张表格,同时可以选择是否填写表名。具体代码如下:
  1. //把一维数组作为一行转换为表格
  2. public static DataTable ToDataTableWithRowArray<TRow>(TRow[] array, string? tableName = null)
  3. {
  4.     var dataTable = new DataTable(tableName);
  5.     //创建列
  6.     for (var i = 0; i < array.Length; i++)
  7.     {
  8.         dataTable.Columns.Add(null, typeof(TRow));
  9.     }
  10.     //添加行数据
  11.     var dataRow = dataTable.NewRow();
  12.     for (var i = 0; i < array.Length; i++)
  13.     {
  14.         dataRow[i] = array[i];
  15.     }
  16.     dataTable.Rows.Add(dataRow);
  17.     return dataTable;
  18. }
复制代码
05、行列转置

该方法是指把DataTable中的行和列互换,就是行的数据变成列,列的数据变成行。如下图示例:

这个示例转换,第一个表格中列名并没有作为数据进行转换,因此我们会提供一个可选项参数用来指示要不要把类目作为数据进行转换。
整个方法实现逻辑也很简朴,就是以原表格行数为列数创建一个新表格,然后在循环处理处罚原表格列,并把原表格一列数据填充至新表格的一行数据中,直至原表格全部列处理处罚完成则完成行列转置。具体代码如下:
  1. //行列转置
  2. public static DataTable Transpose(DataTable dataTable, bool isColumnNameAsData = true)
  3. {
  4.     var transposed = new DataTable(dataTable.TableName);
  5.     //如果列名作为数据,则需要多加一列
  6.     if (isColumnNameAsData)
  7.     {
  8.         transposed.Columns.Add();
  9.     }
  10.     //转置后,行数即为新的列数
  11.     for (int i = 0; i < dataTable.Rows.Count; i++)
  12.     {
  13.         transposed.Columns.Add();
  14.     }
  15.     //以列为单位,一次处理一列数据
  16.     for (var column = 0; column < dataTable.Columns.Count; column++)
  17.     {
  18.         //创建新行
  19.         var newRow = transposed.NewRow();
  20.         //如果列名作为数据,则先把列名加入第一列
  21.         if (isColumnNameAsData)
  22.         {
  23.             newRow[0] = dataTable.Columns[column].ColumnName;
  24.         }
  25.         //把一列数据转为一行数据
  26.         for (var row = 0; row < dataTable.Rows.Count; row++)
  27.         {
  28.             //如果列名作为数据,则行数据从第二列开始填充
  29.             var rowIndex = isColumnNameAsData ? row + 1 : row;
  30.             newRow[rowIndex] = dataTable.Rows[row][column];
  31.         }
  32.         transposed.Rows.Add(newRow);
  33.     }
  34.     return transposed;
  35. }
复制代码
下面进行简朴的单元测试:
  1. [Fact]
  2. public void Transpose_ColumnNameAsData()
  3. {
  4.     DataTable originalTable = new DataTable("测试");
  5.     originalTable.Columns.Add("A", typeof(string));
  6.     originalTable.Columns.Add("B", typeof(int));
  7.     originalTable.Columns.Add("C", typeof(int));
  8.     originalTable.Rows.Add("D", 1, 2);
  9.     //列名作为数据的情况
  10.     var table = TableHelper.Transpose(originalTable, true);
  11.     Assert.Equal(originalTable.TableName, table.TableName);
  12.     Assert.Equal("Column1", table.Columns[0].ColumnName);
  13.     Assert.Equal("Column2", table.Columns[1].ColumnName);
  14.     Assert.Equal(3, table.Rows.Count);
  15.     Assert.Equal("A", table.Rows[0][0]);
  16.     Assert.Equal("D", table.Rows[0][1]);
  17.     Assert.Equal("B", table.Rows[1][0]);
  18.     Assert.Equal("1", table.Rows[1][1].ToString());
  19.     Assert.Equal("C", table.Rows[2][0]);
  20.     Assert.Equal("2", table.Rows[2][1].ToString());
  21. }
复制代码
:测试方法代码以及示例源码都已经上传至代码库,有兴趣的可以看看。https://gitee.com/hugogoos/Ideal

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

美丽的神话

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表